Jelajahi Sumber

weixin账号抓取方法,增加category_list作为输入

luojunhui 8 bulan lalu
induk
melakukan
f6acc8a191
1 mengubah file dengan 4 tambahan dan 4 penghapusan
  1. 4 4
      coldStartTasks/publish/publishCategoryArticles.py

+ 4 - 4
coldStartTasks/publish/publishCategoryArticles.py

@@ -7,7 +7,7 @@ import json
 
 from pandas import DataFrame
 
-from applications import aiditApi, log, bot, DeNetMysql
+from applications import aiditApi, log, bot
 from config import apolloConfig
 
 apollo = apolloConfig()
@@ -223,17 +223,17 @@ class CategoryColdStartTask(object):
             article_id_list = filter_df['article_id'].values.tolist()
             self.change_article_status_while_publishing(article_id_list=article_id_list)
 
-    def do_job(self):
+    def do_job(self, category_list=None):
         """
         执行任务
         :return:
         """
-        category_list = self.category_map.keys()
+        if not category_list:
+            category_list = self.category_map.keys()
         log(
             task="category_publish_task",
             function="do_job",
             message="开始自动创建品类文章抓取计划",
-            status="success",
             data={
                 "category_list": list(category_list)
             }