소스 검색

Raise ImproperlyConfigured if CELERY_RESULT_DBURI not set, instead of weird SQLAlchemy error.

Ask Solem 15 년 전
부모
커밋
d9f57e7153
1개의 변경된 파일6개의 추가작업 그리고 1개의 파일을 삭제
  1. 6 1
      celery/backends/database.py

+ 6 - 1
celery/backends/database.py

@@ -2,9 +2,10 @@ from datetime import datetime
 
 
 from celery import conf
+from celery.backends.base import BaseDictBackend
 from celery.db.models import Task, TaskSet
 from celery.db.session import ResultSession
-from celery.backends.base import BaseDictBackend
+from celery.exceptions import ImproperlyConfigured
 
 
 class DatabaseBackend(BaseDictBackend):
@@ -12,6 +13,10 @@ class DatabaseBackend(BaseDictBackend):
 
     def __init__(self, dburi=conf.RESULT_DBURI,
             engine_options=None, **kwargs):
+        if not dburi:
+            raise ImproperlyConfigured(
+                    "Missing connection string! Do you have "
+                    "CELERY_RESULT_DBURI set to a real value?")
         self.dburi = dburi
         self.engine_options = dict(engine_options or {},
                                    **conf.RESULT_ENGINE_OPTIONS or {})