|
|
@@ -350,3 +350,73 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
|
|
|
info="视频搜索失败， 被敏感词过滤",
|
|
|
trace_id=trace_id
|
|
|
)
|
|
|
+
|
|
|
+
|
|
|
+async def re_search_videos(params, trace_id, gh_id, mysql_client):
|
|
|
+ """
|
|
|
+ 重新搜索接口
|
|
|
+ :param params:
|
|
|
+ :param trace_id:
|
|
|
+ :param gh_id:
|
|
|
+ :param mysql_client:
|
|
|
+ :return:
|
|
|
+ cls.ori_title = info["ori_title"]
|
|
|
+ cls.article_summary = info["content_title"]
|
|
|
+ cls.article_keys = info["content_keys"]
|
|
|
+ cls.trace_id = info["trace_id"]
|
|
|
+ """
|
|
|
+ obj = {
|
|
|
+ "ori_title": params['ori_title'],
|
|
|
+ "content_title": params['kimi_summary'],
|
|
|
+ "content_keys": params['kimi_keys'],
|
|
|
+ "trace_id": params['trace_id']
|
|
|
+ }
|
|
|
+ SearchAB = SearchABTest(info=obj, gh_id=gh_id)
|
|
|
+ recall_obj_1 = await SearchAB.ab_1()
|
|
|
+ # recall_obj_1 = await SearchAB.ab_0()
|
|
|
+ await asyncio.sleep(3)
|
|
|
+ recall_obj_2 = await SearchAB.ab_2()
|
|
|
+ await asyncio.sleep(3)
|
|
|
+ recall_obj_3 = await SearchAB.ab_3()
|
|
|
+ print("{}---视频搜索正常".format(trace_id))
|
|
|
+ recall_list = [recall_obj_1, recall_obj_2, recall_obj_3]
|
|
|
+ un_empty_list = [i for i in recall_list if i]
|
|
|
+ if len(un_empty_list) < 3:
|
|
|
+ await asyncio.sleep(3)
|
|
|
+ recall_obj_4 = await SearchAB.ab_4()
|
|
|
+ if recall_obj_4:
|
|
|
+ un_empty_list.append(recall_obj_4)
|
|
|
+
|
|
|
+ # 逐条下载，逐条写表
|
|
|
+ if un_empty_list:
|
|
|
+ for index, recall_obj in enumerate(un_empty_list, 1):
|
|
|
+ platform = recall_obj["platform"]
|
|
|
+ recall_video = recall_obj["result"]
|
|
|
+ if recall_video:
|
|
|
+ logging(
|
|
|
+ code="7002",
|
|
|
+ info="视频搜索成功, 搜索平台为--{}".format(platform),
|
|
|
+ trace_id=trace_id,
|
|
|
+ data=recall_video,
|
|
|
+ )
|
|
|
+ video_id = await video_sender(
|
|
|
+ video_obj=recall_video,
|
|
|
+ user=gh_id_dict.get(gh_id),
|
|
|
+ trace_id=trace_id,
|
|
|
+ platform=platform,
|
|
|
+ )
|
|
|
+ update_id_sql = f"""
|
|
|
+ UPDATE long_articles_video
|
|
|
+ SET
|
|
|
+ recall_video_id{index} = {video_id}
|
|
|
+ WHERE
|
|
|
+ trace_id = '{trace_id}'
|
|
|
+ """
|
|
|
+ await mysql_client.async_insert(update_id_sql)
|
|
|
+ else:
|
|
|
+ logging(
|
|
|
+ code="7003",
|
|
|
+ info="视频搜索失败， 被敏感词过滤",
|
|
|
+ trace_id=trace_id
|
|
|
+ )
|
|
|
+
|
罗俊辉