feat(api): Move Celery Beat schedule configuration

Author: pycook

cmdb-api/api/tasks/cmdb.py

@@ -495,3 +495,22 @@ def counter_daily():
         raise
     finally:
         db.session.close()
+
+
+# Beat schedule for CMDB tasks
+from celery.schedules import crontab
+
+CMDB_BEAT_SCHEDULE = {
+    'cmdb-counter-main': {
+        'task': 'cmdb.counter_main',
+        'schedule': crontab(minute='*'),
+    },
+    'cmdb-counter-adc': {
+        'task': 'cmdb.counter_adc',
+        'schedule': crontab(minute='*/5'),
+    },
+    'cmdb-counter-daily': {
+        'task': 'cmdb.counter_daily',
+        'schedule': crontab(hour=0, minute=0),
+    },
+}

cmdb-api/celery_worker.py

@@ -2,25 +2,14 @@
 
 from api.app import create_app
 from api.extensions import celery
-from celery.schedules import crontab
 
 # celery -A celery_worker.celery worker -l DEBUG -E -Q xxxx
 
 app = create_app()
 app.app_context().push()
 
-# Beat schedule configuration
-celery.conf.beat_schedule = {
-    'cmdb-counter-main': {
-        'task': 'cmdb.counter_main',
-        'schedule': crontab(minute='*'),
-    },
-    'cmdb-counter-adc': {
-        'task': 'cmdb.counter_adc',
-        'schedule': crontab(minute='*/5'),
-    },
-    'cmdb-counter-daily': {
-        'task': 'cmdb.counter_daily',
-        'schedule': crontab(hour=0, minute=0),
-    },
-}
+# Load beat schedules from all modules
+from api.tasks.cmdb import CMDB_BEAT_SCHEDULE
+
+celery.conf.beat_schedule = celery.conf.get('beat_schedule', {})
+celery.conf.beat_schedule.update(CMDB_BEAT_SCHEDULE)