Merge pull request #262 from BEMServer/celery_init_app

Store BEMServerCore instance as Celery app attribute

Author: lafrech

src/bemserver_core/__init__.py

@@ -66,7 +66,7 @@ def __init__(self):
 
         # Configure Celery
         celery_app.set_default()
-        celery_app.conf.update(self.config["CELERY_CONFIG"])
+        celery_app.init_app(self)
 
         # Load plugins
         plugins.init_core(self)

src/bemserver_core/celery.py

@@ -117,6 +117,11 @@ class BEMServerCoreCelery(Celery):
 
     SCHEDULED_TASKS_NAME_SUFFIX = "Scheduled"
 
+    def init_app(self, bsc):
+        """Init Celery app with BEMServerCore instance"""
+        self.bsc = bsc
+        self.conf.update(bsc.config["CELERY_CONFIG"])
+
     def register_task(self, task, **options):
         """Register task