Forráskód Böngészése

自动回复抓取

luojunhui 1 hónapja
szülő
commit
b902805785

+ 2 - 0
applications/config/task_chinese_name.py

@@ -22,4 +22,6 @@ name_map = {
     "update_account_open_rate_avg": "更新账号平均打开率",
     "update_limited_account_info": "更新限流账号信息",
     "update_account_read_avg": "更新账号平均阅读率",
+    "get_follow_result": "获取自动关注回复",
+    "extract_reply_result": "解析自动回复结果",
 }

+ 13 - 3
applications/tasks/task_handler.py

@@ -277,11 +277,21 @@ class TaskHandler(TaskMapper):
 
     # 自动关注公众号账号
     async def _auto_follow_account_handler(self) -> int:
-        task = AutoReplyCardsMonitor(
-            pool=self.db_client, log_client=self.log_client
-        )
+        task = AutoReplyCardsMonitor(pool=self.db_client, log_client=self.log_client)
         await task.deal(task_name="follow_gzh_task")
         return self.TASK_SUCCESS_STATUS
 
+    # 获取自动关注回复
+    async def _get_follow_result_handler(self) -> int:
+        task = AutoReplyCardsMonitor(pool=self.db_client, log_client=self.log_client)
+        await task.deal(task_name="get_auto_reply_task")
+        return self.TASK_SUCCESS_STATUS
+
+    # 解析自动回复结果
+    async def _extract_reply_result_handler(self) -> int:
+        task = AutoReplyCardsMonitor(pool=self.db_client, log_client=self.log_client)
+        await task.deal(task_name="extract_task")
+        return self.TASK_SUCCESS_STATUS
+
 
 __all__ = ["TaskHandler"]

+ 4 - 0
applications/tasks/task_scheduler.py

@@ -211,6 +211,10 @@ class TaskScheduler(TaskHandler):
             "update_account_open_rate_avg": self._update_account_open_rate_avg_handler,
             # 自动关注公众号账号
             "auto_follow_account": self._auto_follow_account_handler,
+            # 获取自动关注回复
+            "get_follow_result": self._get_follow_result_handler,
+            # 解析自动回复结果
+            "extract_reply_result": self._extract_reply_result_handler,
         }
 
         if task_name not in handlers: