upgrade search celery app in project

Friskes · May 30, 2024 · d07370e · d07370e
1 parent 292fa46
commit d07370e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/celery_starter/management/commands/runcelery.py b/src/celery_starter/management/commands/runcelery.py
@@ -136,8 +136,13 @@ def get_celery_app(self) -> tuple[str, str | None]:
                 if cmd[i].startswith('--app'):
                     return cmd[i].split('=')[-1].strip(), self.args[0]
         else:
-            if os.environ.get('CELERY_APP'):
-                return os.environ['CELERY_APP'], None
+            celery_app = (
+                os.environ.get('CELERY_APP', None)
+                or getattr(settings, 'CELERY_APP', None)
+                or default_app.conf.get('CELERY_APP', 'app')
+            )
+            if celery_app:
+                return celery_app, None
 
             settings_filename = 'settings.py'
             for root, _dirs, files in os.walk(self.BASE_DIR):