From 9a85cdf666b86623ae619d7f1ce42019973132b9 Mon Sep 17 00:00:00 2001 From: erikvw Date: Tue, 20 Aug 2024 20:33:12 -0500 Subject: [PATCH] conditionally import celery_app --- meta_edc/__init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/meta_edc/__init__.py b/meta_edc/__init__.py index 639a482..e8d1391 100644 --- a/meta_edc/__init__.py +++ b/meta_edc/__init__.py @@ -1,9 +1,13 @@ import os +import sys from importlib.metadata import PackageNotFoundError, version -from .celery.debug import app as celery_debug -from .celery.live import app as celery_live -from .celery.uat import app as celery_uat +if "meta_edc.celery.live:app" in sys.argv: + from .celery.live import app as celery_app +elif "meta_edc.celery.uat:app" in sys.argv: + from .celery.uat import app as celery_app +else: + from .celery.debug import app as celery_app try: __version__ = version(os.getcwd().split(os.sep)[-1]) @@ -11,4 +15,4 @@ __version__ = None -__all__ = ["celery_live", "celery_uat", "celery_debug", "__version__"] +__all__ = ["celery_app", "__version__"]