瀏覽代碼

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

罗俊辉 9 月之前
父節點
當前提交
c14f8c8ac5
共有 2 個文件被更改,包括 30 次插入30 次删除
  1. 28 28
      applications/schedule/search_schedule.py
  2. 2 2
      applications/static/config.py

+ 28 - 28
applications/schedule/search_schedule.py

@@ -329,34 +329,34 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
     # print(params['title'].split("@@")[-1])
     for i in ranked_list:
         print(i['title'], i['score'])
-    # 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
+    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