feat: menu celery task

fmelihh · Dec 7, 2024 · b0702f9 · b0702f9
1 parent dbdc9d6
commit b0702f9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/recommendation_engine/app/tasks/comments.py b/src/recommendation_engine/app/tasks/comments.py
@@ -1,4 +1,5 @@
 from celery.app.task import Task
+from celery.schedules import crontab
 
 from ..shared_kernel.scheduler.celery_app import celery_application
 from ..features.comments.services import (
@@ -45,3 +46,10 @@ def run(self, *args, **kwargs):
 
 
 celery_application.register_task(CommentTask)
+
+
+celery_application.conf.beat_schedule[str(CommentTask.__name__)] = {
+    "task": str(CommentTask.__module__),
+    "schedule": crontab(hour="12", minute="30"),
+    "options": {"queue": "periodic"},
+}
diff --git a/src/recommendation_engine/app/tasks/menu.py b/src/recommendation_engine/app/tasks/menu.py
@@ -1,4 +1,5 @@
 from celery.app.task import Task
+from celery.schedules import crontab
 
 from ..shared_kernel.domain_providers import Providers
 from ..features.restaurants.services import RestaurantService
@@ -42,3 +43,9 @@ def run(self, *args, **kwargs):
 
 
 celery_application.register_task(MenuTask)
+
+celery_application.conf.beat_schedule[str(MenuTask.__name__)] = {
+    "task": str(MenuTask.__module__),
+    "schedule": crontab(hour="12", minute="30"),
+    "options": {"queue": "periodic"},
+}