refactor!: use new filter to define celery command

eduNEXT · Nov 22, 2024 · 4718af6 · 4718af6
1 parent b3ea70e
commit 4718af6
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 28 deletions.
diff --git a/tutorcelery/__about__.py b/tutorcelery/__about__.py
@@ -1 +1 @@
-__version__ = "19.0.0"
+__version__ = "18.3.0"
diff --git a/tutorcelery/patches/k8s-override b/tutorcelery/patches/k8s-override
diff --git a/tutorcelery/plugin.py b/tutorcelery/plugin.py
@@ -6,7 +6,7 @@
 import click
 import importlib_resources
 import tutor
-from tutor import hooks
+from tutor import hooks, config
 
 from .__about__ import __version__
 from .hooks import CELERY_WORKERS_CONFIG, CELERY_WORKERS_ATTRS_TYPE
@@ -80,6 +80,18 @@ def is_celery_multiqueue(service: str) -> bool:
     return True
 
 
+@hooks.Actions.PROJECT_ROOT_READY.add()
+def run_this_on_start(config):
+    if is_celery_multiqueue("lms"):
+        hooks.Filters.LMS_WORKER_COMMAND.add_items([
+            "--queues=edx.lms.core.default"
+        ])
+    if is_celery_multiqueue("cms"):
+        hooks.Filters.CMS_WORKER_COMMAND.add_items([
+            "--queues=edx.cms.core.default"
+        ])
+
+
 hooks.Filters.CONFIG_DEFAULTS.add_items(
     [
         # Add your new settings that have default values here.