Bläddra i källkod

Merge branch 'feature/luojunhui/20260529-cooperate-auto-reply-improve' of Server/LongArticleTaskServer into master

luojunhui 1 vecka sedan
förälder
incheckning
3f2f7fd8b1

+ 1 - 3
app/domains/monitor_tasks/auto_reply_cards_monitor/_mapper.py

@@ -222,9 +222,7 @@ class AutoReplyCardsMonitorMapper(AutoReplyCardsMonitorConst):
             FROM cooperate_auto_reply_detail
             WHERE cover_status = %s;
         """
-        return await self.pool.async_fetch(
-            query=query, params=(self.FAIL_STATUS,)
-        )
+        return await self.pool.async_fetch(query=query, params=(self.FAIL_STATUS,))
 
     # 保存封面至封面表
     async def save_cover(self, cover_id, root_source_id, oss_path):

+ 15 - 5
app/domains/monitor_tasks/auto_reply_cards_monitor/entrance.py

@@ -457,7 +457,9 @@ class AutoReplyCardsMonitor(AutoReplyCardsMonitorConst):
                 )
 
     # 创建单个账号自动回复任务
-    async def create_auto_reply_single_account_task(self, gh_id, account_name, priority=0):
+    async def create_auto_reply_single_account_task(
+        self, gh_id, account_name, priority: int = 1
+    ):
         task_id = self.tool.generate_task_id(task_name="auto_reply", gh_id=gh_id)
         # 先插入 task, 再创建自动回复任务
         create_row = await self.mapper.create_auto_reply_task(task_id, gh_id)
@@ -605,8 +607,10 @@ class AutoReplyCardsMonitor(AutoReplyCardsMonitorConst):
                     )
                     continue
 
-                print(gh_id, account_name, uv, priority)
-                await self.create_auto_reply_single_account_task(gh_id, account_name, priority)
+                # print(gh_id, account_name, uv, priority)
+                await self.create_auto_reply_single_account_task(
+                    gh_id, account_name, priority
+                )
 
             except Exception as e:
                 await self.log_service.log(
@@ -889,7 +893,11 @@ class AutoReplyCardsMonitor(AutoReplyCardsMonitorConst):
                 continue
 
             try:
-                xml_obj = json.loads(task_result) if isinstance(task_result, str) else task_result
+                xml_obj = (
+                    json.loads(task_result)
+                    if isinstance(task_result, str)
+                    else task_result
+                )
                 cover_ok = await self._try_download_cover(task_id, position, xml_obj)
                 if cover_ok:
                     success_cnt += 1
@@ -938,7 +946,9 @@ class AutoReplyCardsMonitor(AutoReplyCardsMonitorConst):
                 all_cover_fails = []
 
                 for task in tqdm(task_list, desc="解析任务"):
-                    total, parse_fails, cover_fails = await self.extract_single_xml(task)
+                    total, parse_fails, cover_fails = await self.extract_single_xml(
+                        task
+                    )
                     batch_total += total
                     all_parse_fails.extend(parse_fails)
                     all_cover_fails.extend(cover_fails)