Allow environment specification of Celery broker

weiwzhang · Jul 14, 2016 · 5386a4d · 5386a4d
1 parent 4692d10
commit 5386a4d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cesium_app/celery_app.py b/cesium_app/celery_app.py
@@ -1,8 +1,12 @@
 from celery import Celery
 from cesium.celery_app import celery_config
 
+import os
+CELERY_BROKER = os.environ.get(
+        'CELERY_BROKER', celery_config['CELERY_BROKER'])
+
 celery_config['CELERY_IMPORTS'].append('cesium_app.celery_tasks')
 #celery_config['CELERY_RESULT_DBURI'] = 'db+postgresql://cesium:cesium@localhost/cesium'
 #celery_config['CELERY_RESULT_BACKEND'] = 'db+postgresql://cesium:cesium@localhost/cesium'
-app = Celery('cesium_app', broker=celery_config['CELERY_BROKER'])
+app = Celery('cesium_app', broker=CELERY_BROKER)
 app.config_from_object(celery_config)