Merge branch '2024-11-19-luojunhui-newMethod' of Server/title_with_video into 2024-09-23newDbTasks

tasks/newContentIdTask.py

@@ -40,6 +40,7 @@ class NewContentIdTask(object):
         self.gh_id_dict = json.loads(self.config.get_config_value("testAccountLevel2"))
         self.account_map = json.loads(self.config.get_config_value("accountMap"))
         self.spider_coroutines = self.config.get_config_value("spiderCoroutines")
+        self.new_method_gh_id = json.loads(self.config.get_config_value("newMethodGhId"))
 
     async def get_tasks(self):
         """
@@ -733,6 +734,7 @@ class NewContentIdTask(object):
         trace_id = params['trace_id']
         process_times = params['process_times']
         content_id = params['content_id']
+        gh_id = params['gh_id']
         print(kimi_result)
         if kimi_result:
             # 等待 kimi 操作执行完成之后，开始执行 spider_task
@@ -760,23 +762,31 @@ class NewContentIdTask(object):
                         info="etl_success",
                         trace_id=trace_id
                     )
-                    try:
-                        await self.publish_task(params, kimi_result['kimi_title'])
+                    if gh_id in self.new_method_gh_id:
                         logging(
-                            code="3004",
-                            info="publish_success",
+                            code="3013",
+                            info="new_method_gh_id",
                             trace_id=trace_id
                         )
-                        await record_trace_id(
-                            trace_id=trace_id,
-                            status=self.RECORD_SUCCESS_TRACE_ID_CODE
-                        )
-                    except Exception as e:
-                        logging(
-                            code="6004",
-                            info="publish 失败--{}".format(e),
-                            trace_id=params['trace_id']
-                        )
+                        return
+                    else:
+                        try:
+                            await self.publish_task(params, kimi_result['kimi_title'])
+                            logging(
+                                code="3004",
+                                info="publish_success",
+                                trace_id=trace_id
+                            )
+                            await record_trace_id(
+                                trace_id=trace_id,
+                                status=self.RECORD_SUCCESS_TRACE_ID_CODE
+                            )
+                        except Exception as e:
+                            logging(
+                                code="6004",
+                                info="publish 失败--{}".format(e),
+                                trace_id=params['trace_id']
+                            )
                 else:
                     logging(
                         code="6003",