luojunhui vor 1 Monat
Ursprung
Commit
f323ec3a17

+ 5 - 3
app/api/service/task_manager_service.py

@@ -1,6 +1,8 @@
 import json
 from typing import Optional
+
 from app.core.config import GlobalConfigSettings
+from app.core.config.settings import TaskChineseNameConfig
 
 
 class TaskConst:
@@ -15,7 +17,7 @@ class TaskConst:
 
 
 class TaskManagerUtils(TaskConst):
-    def __init__(self, config: GlobalConfigSettings):
+    def __init__(self, config: TaskChineseNameConfig):
         self.config = config
 
     def get_task_chinese_name(self, data):
@@ -23,7 +25,7 @@ class TaskManagerUtils(TaskConst):
         通过输入任务详情信息获取任务名称
         """
         task_name = data["task_name"]
-        task_name_chinese = self.config.task_chinese_name.get(task_name, task_name)
+        task_name_chinese = self.config.name_map.get(task_name, task_name)
 
         # account_method
         if task_name == "crawler_gzh_articles":
@@ -86,7 +88,7 @@ class TaskManagerUtils(TaskConst):
 
 class TaskManager(TaskManagerUtils):
     def __init__(self, pool, data, config: GlobalConfigSettings):
-        super().__init__(config)
+        super().__init__(config.task_chinese_name)
         self.pool = pool
         self.data = data
 

+ 1 - 1
app/core/config/settings/task_chinese_name.py

@@ -7,7 +7,7 @@ class TaskChineseNameConfig(BaseSettings):
     """冷启动配置"""
 
     # 分类映射
-    category_map: Dict[str, str] = Field(
+    name_map: Dict[str, str] = Field(
         default_factory=lambda: {
             "title_rewrite": "标题重写",
             "crawler_gzh_articles": "抓取公众号文章",

+ 1 - 1
app/jobs/task_config.py

@@ -162,7 +162,7 @@ TASK_CONFIGS = {
     ),
     # 合作账号任务
     "cooperate_accounts_monitor": TaskConfig(
-        timeout=3600,
+        timeout=7200,
         max_concurrent=2,
     ),
     "cooperate_accounts_detail": TaskConfig(