给每个定时任务增加一个开关

.env.example

@@ -5,6 +5,10 @@ DATABASE_URL=postgresql+asyncpg://user:password@localhost:5432/ai_talk_callback_
 CELERY_BROKER_URL=redis://localhost:6379/0
 CELERY_RESULT_BACKEND=redis://localhost:6379/0
 
+# 任务配置
+ENABLED_PUSH_DATA_TO_DTC_TASK=false
+ENABLED_CALL_API_TASK=false
+
 # Flower监控配置
 FLOWER_ENABLED=true
 FLOWER_PORT=5555

app/celery_tasks.py

@@ -41,6 +41,7 @@ engine = create_engine(
 )
 
 @celery_app.task(bind=True, name='push_data_to_dtc')
+@conditional_task(enabled=settings.enabled_push_data_to_dtc_task)
 def push_data_to_dtc_task(self):
     """
     推送数据给DTC的Celery任务
@@ -221,7 +222,7 @@ def push_data_to_dtc_task(self):
 
 
 @celery_app.task(bind=True, name='call_api', max_retries=None)
-@conditional_task(enabled=False)
+@conditional_task(enabled=settings.enabled_call_api_task)
 def execute_call_api_task(self):
     """遍历API配置文件中的所有API信息并调用接口"""
     current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')

app/config.py

@@ -18,6 +18,10 @@ class Settings(BaseSettings):
     celery_timezone: str = "Asia/Shanghai"
     celery_enable_utc: bool = False
 
+    # 任务配置
+    enabled_push_data_to_dtc_task: bool = False  # 是否启用推送数据给DTC的任务
+    enabled_call_api_task: bool = False  # 是否启用调用API的任务
+
     # Flower监控配置
     flower_enabled: bool = True  # 是否启用Flower监控
     flower_url: str = "http://localhost:5555"  # Flower访问URL