luojunhui 3 ヶ月 前
コミット
c83dd71322
1 ファイル変更12 行追加11 行削除
  1. 12 11
      tasks/crawler_accounts_by_association.py

+ 12 - 11
tasks/crawler_accounts_by_association.py

@@ -118,6 +118,15 @@ class CrawlerAccounts:
         affected_rows = self.db_client.save(query=update_query, params=(new_status, video_id_tuple, ori_status))
         return affected_rows
 
+    def cal_video_similarity_score(self):
+        # cal similarity score
+        video_list = self.get_video_list_without_score()
+        if video_list:
+            affected_rows = self.save_similarity_score_to_table(video_list)
+            print(affected_rows)
+        else:
+            print("no video recommendation")
+
 
 class ChannelsAccountCrawler(CrawlerAccounts):
     """
@@ -221,10 +230,7 @@ class ChannelsAccountCrawler(CrawlerAccounts):
                     }
                 )
 
-        # cal similarity score
-        video_list = self.get_video_list_without_score()
-        affected_rows = self.save_similarity_score_to_table(video_list)
-        print(affected_rows)
+        self.cal_video_similarity_score()
 
 
 class ToutiaoAccountCrawler(CrawlerAccounts):
@@ -311,10 +317,7 @@ class ToutiaoAccountCrawler(CrawlerAccounts):
                     },
                 )
 
-        # cal similarity score
-        video_list = self.get_video_list_without_score()
-        affected_rows = self.save_similarity_score_to_table(video_list)
-        print(affected_rows)
+        self.cal_video_similarity_score()
 
 
 class HaoKanAccountCrawler(CrawlerAccounts):
@@ -392,6 +395,4 @@ class HaoKanAccountCrawler(CrawlerAccounts):
                     }
                 )
 
-        video_list = self.get_video_list_without_score()
-        affected_rows = self.save_similarity_score_to_table(video_list)
-        print(affected_rows)
+        self.cal_video_similarity_score()