Plug custom Celery tasks via dynamic settings (getredash#3819)

* plug custom celery tasks via dynamic settings * an extra blank line
pushpay · Nov 14, 2019 · 1d62a4e · 1d62a4e
1 parent 08f2fc4
commit 1d62a4e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/redash/settings/dynamic_settings.py b/redash/settings/dynamic_settings.py
@@ -8,3 +8,13 @@ def query_time_limit(is_scheduled, user_id, org_id):
     adhoc_time_limit = int_or_none(os.environ.get('REDASH_ADHOC_QUERY_TIME_LIMIT', None))
 
     return scheduled_time_limit if is_scheduled else adhoc_time_limit
+
+
+# Provide any custom tasks you'd like to run periodically
+def custom_tasks():
+    return {
+        # 'key': {
+        #     'task': 'redash.task.example',
+        #     'schedule': timedelta(minutes=5)
+        # }
+    }
diff --git a/redash/worker.py b/redash/worker.py
@@ -46,6 +46,8 @@
         'schedule': timedelta(minutes=5)
     }
 
+celery_schedule.update(settings.dynamic_settings.custom_tasks())
+
 celery.conf.update(result_backend=settings.CELERY_RESULT_BACKEND,
                    beat_schedule=celery_schedule,
                    timezone='UTC',