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

2024-07-01
标题相似度-测试服务

罗俊辉 1 éve
szülő
commit
de95af8983
2 módosított fájl, 33 hozzáadás és 30 törlés
  1. 31 28
      applications/schedule/search_schedule.py
  2. 2 2
      applications/static/config.py

+ 31 - 28
applications/schedule/search_schedule.py

@@ -326,34 +326,37 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
     )
     # 按照标题相似度排序
     ranked_list = title_similarity_rank(content_title=content_title, recall_list=recall_list)
-    index = 0
-    for recall_obj in ranked_list:
-        if recall_obj:
-            platform = recall_obj['platform']
-            recall_video = recall_obj['result']
-            if recall_video:
-                index += 1
-                await video_sender(
-                    video_obj=recall_video,
-                    user=gh_id_dict.get(gh_id),
-                    trace_id=trace_id,
-                    platform=platform,
-                    index=index
-                )
-                logging(
-                    code="1007",
-                    info="成功请求etl",
-                    data=recall_video,
-                    trace_id=trace_id
-                )
-                if index >= 3:
-                    print("already downloaded 3 videos")
-                    logging(
-                        code="1008",
-                        info="成功下载三条视频",
-                        trace_id=trace_id
-                    )
-                    break
+    print(recall_list)
+    print("排序后")
+    print(ranked_list)
+    # index = 0
+    # for recall_obj in ranked_list:
+    #     if recall_obj:
+    #         platform = recall_obj['platform']
+    #         recall_video = recall_obj['result']
+    #         if recall_video:
+    #             index += 1
+    #             await video_sender(
+    #                 video_obj=recall_video,
+    #                 user=gh_id_dict.get(gh_id),
+    #                 trace_id=trace_id,
+    #                 platform=platform,
+    #                 index=index
+    #             )
+    #             logging(
+    #                 code="1007",
+    #                 info="成功请求etl",
+    #                 data=recall_video,
+    #                 trace_id=trace_id
+    #             )
+    #             if index >= 3:
+    #                 print("already downloaded 3 videos")
+    #                 logging(
+    #                     code="1008",
+    #                     info="成功下载三条视频",
+    #                     trace_id=trace_id
+    #                 )
+    #                 break
 
 
 async def re_search_videos(params, trace_id, gh_id):

+ 2 - 2
applications/static/config.py

@@ -457,10 +457,10 @@ ab_test_config = {
 }
 
 # prod
-db_article = "long_articles_video"
+# db_article = "long_articles_video"
 
 # dev
-# db_article = "long_articles_video_dev"
+db_article = "long_articles_video_dev"
 article_queue = "long_articles_queue"
 
 # spider coroutines