Bläddra i källkod

Add the ability to specify __tablename__ for SQLALchemy tables.

Ryan Petrello 12 år sedan
förälder
incheckning
2bc216f6d9
2 ändrade filer med 6 tillägg och 0 borttagningar
  1. 2 0
      celery/app/defaults.py
  2. 4 0
      celery/backends/database/__init__.py

+ 2 - 0
celery/app/defaults.py

@@ -125,6 +125,8 @@ NAMESPACES = {
         'REDIS_MAX_CONNECTIONS': Option(type='int'),
         'RESULT_BACKEND': Option(type='string'),
         'RESULT_DB_SHORT_LIVED_SESSIONS': Option(False, type='bool'),
+        'RESULT_DB_TASK_TABLENAME': Option('celery_taskmeta'),
+        'RESULT_DB_TASKSET_TABLENAME': Option('celery_tasksetmeta'),
         'RESULT_DBURI': Option(),
         'RESULT_ENGINE_OPTIONS': Option(type='dict'),
         'RESULT_EXCHANGE': Option('celeryresults'),

+ 4 - 0
celery/backends/database/__init__.py

@@ -69,6 +69,10 @@ class DatabaseBackend(BaseBackend):
             'short_lived_sessions',
             conf.CELERY_RESULT_DB_SHORT_LIVED_SESSIONS,
         )
+
+        Task.__table__.name = conf.CELERY_RESULT_DB_TASK_TABLENAME
+        TaskSet.__table__.name = conf.CELERY_RESULT_DB_TASKSET_TABLENAME
+
         if not self.dburi:
             raise ImproperlyConfigured(
                 'Missing connection string! Do you have '