Browse Source

database backend: Always create tables if they don't exist.

Ask Solem 15 năm trước cách đây
mục cha
commit
5137a3e9d9
1 tập tin đã thay đổi với 3 bổ sung3 xóa
  1. 3 3
      celery/db/session.py

+ 3 - 3
celery/db/session.py

@@ -5,10 +5,11 @@ from sqlalchemy.orm import sessionmaker
 from sqlalchemy.ext.declarative import declarative_base
 
 from celery import conf
+from celery.utils.compat import defaultdict
 
 ResultModelBase = declarative_base()
 
-_SETUP = {"results": False}
+_SETUP = defaultdict(lambda: False)
 _ENGINES = {}
 
 
@@ -31,6 +32,5 @@ def setup_results(engine):
 
 def ResultSession(dburi=conf.RESULT_DBURI, **kwargs):
     engine, session = create_session(dburi, **kwargs)
-    if os.environ.get("CELERYINIT"):
-        setup_results(engine)
+    setup_results(engine)
     return session()