Squash #2041 - patch from init file

department-of-veterans-affairs · Nov 19, 2024 · 47000ee · 47000ee
1 parent 107d306
commit 47000ee
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/app/__init__.py b/app/__init__.py
@@ -7,6 +7,7 @@
 from flask_marshmallow import Marshmallow
 from flask_migrate import Migrate
 from time import monotonic
+from ddtrace import patch
 from notifications_utils.clients.zendesk.zendesk_client import ZendeskClient
 from notifications_utils.clients.statsd.statsd_client import StatsdClient
 from notifications_utils.clients.redis.redis_client import RedisClient
@@ -41,6 +42,7 @@
 
 load_dotenv()
 
+patch(celery=True)
 
 migrate = Migrate()
 ma = Marshmallow()

diff --git a/app/celery/celery.py b/app/celery/celery.py
@@ -1,7 +1,6 @@
 import time
 
-import celery
-from celery import Task
+from celery import Task, Celery
 from celery.signals import worker_process_shutdown, worker_shutting_down, worker_process_init
 from flask import current_app
 
@@ -83,7 +82,7 @@ def __call__(
     return NotifyTask
 
 
-class NotifyCelery(celery.Celery):
+class NotifyCelery(Celery):
     def init_app(
         self,
         app,