Browse Source

只针对level1的通过裂变率重拍

luojunhui 5 months ago
parent
commit
2ee0ac54bf
2 changed files with 13 additions and 6 deletions
  1. 2 1
      applications/match_algorithm/rank.py
  2. 11 5
      tasks/history_task.py

+ 2 - 1
applications/match_algorithm/rank.py

@@ -95,7 +95,7 @@ async def get_title_oss_fission_dict(db_client, config, content_id) -> dict[str:
     :return:
     """
     FISSION_DETAIL_TABLE = config.fission_detail_table
-    LONG_ARTICLES_TEXT_TABLE = config.long_article_text_table
+    LONG_ARTICLES_TEXT_TABLE = config.article_text_table
     two_days_ago_dt = (datetime.now() - timedelta(days=2)).strftime('%Y%m%d')
     sql = f"""
         SELECT
@@ -123,5 +123,6 @@ async def get_title_oss_fission_dict(db_client, config, content_id) -> dict[str:
                 continue
             else:
                 fission_info_dict[key] = value
+        return fission_info_dict
     else:
         return {}

+ 11 - 5
tasks/history_task.py

@@ -170,14 +170,20 @@ class historyContentIdTask(object):
                     config=self.config,
                     content_id=content_id
                 )
+                fission_list = [[i] + [fission_dict[i]['fission_0_on_read']] for i in fission_dict.keys()]
+                sorted_fission_list = sorted(fission_list, key=lambda x: x[1], reverse=True)
                 download_videos_with_fission_info = []
-                for video in download_videos:
-                    video["fission_0_rate"] = fission_dict.get(video['video_oss_path'], {}).get("fission_0_rate", 0)
-                    video["fission_0_on_read"] = fission_dict.get(video['video_oss_path'], {}).get("fission_0_on_read", 0)
+                for index, video in enumerate(download_videos[:3]):
+                    video['video_oss_path'] = sorted_fission_list[index][0]
+                    video["fission_0_on_read"] = sorted_fission_list[index][1]
                     download_videos_with_fission_info.append(video)
+                # for video in download_videos:
+                #     video["fission_0_rate"] = fission_dict.get(video['video_oss_path'], {}).get("fission_0_rate", 0)
+                #     video["fission_0_on_read"] = fission_dict.get(video['video_oss_path'], {}).get("fission_0_on_read", 0)
+                #     download_videos_with_fission_info.append(video)
                 # sorted_videos = sorted(download_videos_with_fission_info, key=lambda x: x['fission_0_rate'], reverse=True)
-                sorted_videos = sorted(download_videos_with_fission_info, key=lambda x: x['fission_0_on_read'], reverse=True)
-                video_list = sorted_videos[:3]
+                # sorted_videos = sorted(download_videos_with_fission_info, key=lambda x: x['fission_0_on_read'], reverse=True)
+                video_list = download_videos_with_fission_info
             case _:
                 print("未传流量池信息")
                 video_list = download_videos[:3]