video_rank.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582
  1. import json
  2. import random
  3. import numpy
  4. from log import Log
  5. from config import set_config
  6. from video_recall import PoolRecall
  7. from db_helper import RedisHelper
  8. from utils import FilterVideos, send_msg_to_feishu
  9. log_ = Log()
  10. config_ = set_config()
  11. def video_rank(data, size, top_K, flow_pool_P):
  12. """
  13. 视频分发排序
  14. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  15. :param size: 请求数
  16. :param top_K: 保证topK为召回池视频 type-int
  17. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  18. :return: rank_result
  19. """
  20. if not data['rov_pool_recall'] and not data['flow_pool_recall']:
  21. return []
  22. # 将各路召回的视频按照score从大到小排序
  23. # 最惊奇相关推荐相似视频
  24. # relevant_recall = [item for item in data['rov_pool_recall']
  25. # if item.get('pushFrom') == config_.PUSH_FROM['top_video_relevant_appType_19']]
  26. # relevant_recall_rank = sorted(relevant_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  27. # 最惊奇完整影视视频
  28. # whole_movies_recall = [item for item in data['rov_pool_recall']
  29. # if item.get('pushFrom') == config_.PUSH_FROM['whole_movies']]
  30. # whole_movies_recall_rank = sorted(whole_movies_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  31. # 最惊奇影视解说视频
  32. # talk_videos_recall = [item for item in data['rov_pool_recall']
  33. # if item.get('pushFrom') == config_.PUSH_FROM['talk_videos']]
  34. # talk_videos_recall_rank = sorted(talk_videos_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  35. # 小时级更新数据
  36. # h_recall = [item for item in data['rov_pool_recall'] if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_h']]
  37. # h_recall_rank = sorted(h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  38. # 相对30天天级规则更新数据
  39. day_30_recall = [item for item in data['rov_pool_recall']
  40. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_30day']]
  41. day_30_recall_rank = sorted(day_30_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  42. # 地域分组小时级规则更新数据
  43. region_h_recall = [item for item in data['rov_pool_recall']
  44. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_h']]
  45. region_h_recall_rank = sorted(region_h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  46. # 地域分组小时级更新24h规则更新数据
  47. region_24h_recall = [item for item in data['rov_pool_recall']
  48. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_24h']]
  49. region_24h_recall_rank = sorted(region_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  50. # 地域分组天级规则更新数据
  51. # region_day_recall = [item for item in data['rov_pool_recall']
  52. # if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_region_day']]
  53. # region_day_recall_rank = sorted(region_day_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  54. # 相对24h规则更新数据
  55. rule_24h_recall = [item for item in data['rov_pool_recall']
  56. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h']]
  57. rule_24h_recall_rank = sorted(rule_24h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  58. # 相对24h规则筛选后剩余更新数据
  59. rule_24h_dup_recall = [item for item in data['rov_pool_recall']
  60. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_24h_dup']]
  61. rule_24h_dup_recall_rank = sorted(rule_24h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  62. # 相对48h规则更新数据
  63. rule_48h_recall = [item for item in data['rov_pool_recall']
  64. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_48h']]
  65. rule_48h_recall_rank = sorted(rule_48h_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  66. # 相对48h规则筛选后剩余更新数据
  67. rule_48h_dup_recall = [item for item in data['rov_pool_recall']
  68. if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_48h_dup']]
  69. rule_48h_dup_recall_rank = sorted(rule_48h_dup_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  70. # 天级规则更新数据
  71. # day_recall = [item for item in data['rov_pool_recall'] if item.get('pushFrom') == config_.PUSH_FROM['rov_recall_day']]
  72. # day_recall_rank = sorted(day_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  73. # ROV召回池
  74. # rov_initial_recall = [
  75. # item for item in data['rov_pool_recall']
  76. # if item.get('pushFrom') not in
  77. # [config_.PUSH_FROM['top_video_relevant_appType_19'],
  78. # config_.PUSH_FROM['rov_recall_h'],
  79. # config_.PUSH_FROM['rov_recall_region_h'],
  80. # config_.PUSH_FROM['rov_recall_region_24h'],
  81. # config_.PUSH_FROM['rov_recall_region_day'],
  82. # config_.PUSH_FROM['rov_recall_24h'],
  83. # config_.PUSH_FROM['rov_recall_24h_dup'],
  84. # config_.PUSH_FROM['rov_recall_48h'],
  85. # config_.PUSH_FROM['rov_recall_48h_dup'],
  86. # config_.PUSH_FROM['rov_recall_day'],
  87. # config_.PUSH_FROM['whole_movies'],
  88. # config_.PUSH_FROM['talk_videos']]
  89. # ]
  90. # rov_initial_recall_rank = sorted(rov_initial_recall, key=lambda k: k.get('rovScore', 0), reverse=True)
  91. # rov_recall_rank = whole_movies_recall_rank + talk_videos_recall_rank + h_recall_rank + \
  92. # day_30_recall_rank + region_h_recall_rank + region_24h_recall_rank + \
  93. # region_day_recall_rank + rule_24h_recall_rank + rule_24h_dup_recall_rank + \
  94. # rule_48h_recall_rank + rule_48h_dup_recall_rank + \
  95. # day_recall_rank + rov_initial_recall_rank
  96. rov_recall_rank = day_30_recall_rank + \
  97. region_h_recall_rank + region_24h_recall_rank + \
  98. rule_24h_recall_rank + rule_24h_dup_recall_rank + \
  99. rule_48h_recall_rank + rule_48h_dup_recall_rank
  100. # 流量池
  101. flow_recall_rank = sorted(data['flow_pool_recall'], key=lambda k: k.get('rovScore', 0), reverse=True)
  102. # 对各路召回的视频进行去重
  103. rov_recall_rank, flow_recall_rank = remove_duplicate(rov_recall=rov_recall_rank, flow_recall=flow_recall_rank,
  104. top_K=top_K)
  105. # log_.info('remove_duplicate finished! rov_recall_rank = {}, flow_recall_rank = {}'.format(
  106. # rov_recall_rank, flow_recall_rank))
  107. # rank_result = relevant_recall_rank
  108. rank_result = []
  109. # 从ROV召回池中获取top k
  110. if len(rov_recall_rank) > 0:
  111. rank_result.extend(rov_recall_rank[:top_K])
  112. rov_recall_rank = rov_recall_rank[top_K:]
  113. else:
  114. rank_result.extend(flow_recall_rank[:top_K])
  115. flow_recall_rank = flow_recall_rank[top_K:]
  116. # 按概率 p 及score排序获取 size - k 个视频
  117. i = 0
  118. while i < size - top_K:
  119. # 随机生成[0, 1)浮点数
  120. rand = random.random()
  121. # log_.info('rand: {}'.format(rand))
  122. if rand < flow_pool_P:
  123. if flow_recall_rank:
  124. rank_result.append(flow_recall_rank[0])
  125. flow_recall_rank.remove(flow_recall_rank[0])
  126. else:
  127. rank_result.extend(rov_recall_rank[:size - top_K - i])
  128. return rank_result[:size]
  129. else:
  130. if rov_recall_rank:
  131. rank_result.append(rov_recall_rank[0])
  132. rov_recall_rank.remove(rov_recall_rank[0])
  133. else:
  134. rank_result.extend(flow_recall_rank[:size - top_K - i])
  135. return rank_result[:size]
  136. i += 1
  137. return rank_result[:size]
  138. def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
  139. """
  140. 视频分发排序
  141. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  142. :param size: 请求数
  143. :param top_K: 保证topK为召回池视频 type-int
  144. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  145. :return: rank_result
  146. """
  147. if not videoIds or len(videoIds)==0:
  148. return []
  149. redisObj = RedisHelper()
  150. video_scores = redisObj.get_batch_key(videoIds)
  151. video_items = []
  152. for i in range(len(video_scores)):
  153. try:
  154. video_score_str = json.load(video_scores[i])
  155. video_items.append((videoIds[i], video_score_str[0]))
  156. except Exception:
  157. video_items.append((videoIds[i], 0.0))
  158. sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)
  159. rov_recall_rank = sort_items
  160. fast_flow_recall_rank = []
  161. flow_recall_rank = []
  162. for item in sort_items:
  163. if item[0] in fast_flow_set:
  164. fast_flow_recall_rank.append(item)
  165. elif item[0] in flow_set:
  166. flow_recall_rank.append(item)
  167. # all flow result
  168. all_flow_recall_rank = fast_flow_recall_rank+flow_recall_rank
  169. rank_result = []
  170. rank_set = set('')
  171. # 从ROV召回池中获取top k
  172. if len(rov_recall_rank) > 0:
  173. rank_result.extend(rov_recall_rank[:top_K])
  174. rov_recall_rank = rov_recall_rank[top_K:]
  175. else:
  176. rank_result.extend(all_flow_recall_rank[:top_K])
  177. all_flow_recall_rank = all_flow_recall_rank[top_K:]
  178. for rank_item in rank_result:
  179. rank_set.add(rank_item[0])
  180. # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
  181. i = 0
  182. while i < size - top_K:
  183. # 随机生成[0, 1)浮点数
  184. rand = random.random()
  185. # log_.info('rand: {}'.format(rand))
  186. if rand < flow_pool_P:
  187. for flow_item in all_flow_recall_rank:
  188. if flow_item[0] in rank_set:
  189. continue
  190. else:
  191. rank_result.append(flow_item)
  192. rank_set.add(flow_item[0])
  193. else:
  194. for recall_item in rov_recall_rank:
  195. if recall_item[0] in rank_set:
  196. continue
  197. else:
  198. rank_result.append(recall_item)
  199. rank_set.add(recall_item[0])
  200. i += 1
  201. return rank_result[:size]
  202. def refactor_video_rank(rov_recall_rank, fast_flow_set, flow_set, size, top_K, flow_pool_P):
  203. """
  204. 视频分发排序
  205. :param data: 各路召回的视频 type-dict {'rov_pool_recall': [], 'flow_pool_recall': []}
  206. :param size: 请求数
  207. :param top_K: 保证topK为召回池视频 type-int
  208. :param flow_pool_P: size-top_K视频为流量池视频的概率 type-float
  209. :return: rank_result
  210. """
  211. if not rov_recall_rank or len(rov_recall_rank) == 0:
  212. return []
  213. fast_flow_recall_rank = []
  214. flow_recall_rank = []
  215. for item in rov_recall_rank:
  216. vid = item.get('videoId', 0)
  217. #print(item)
  218. if vid in fast_flow_set:
  219. fast_flow_recall_rank.append(item)
  220. elif vid in flow_set:
  221. flow_recall_rank.append(item)
  222. # all flow result
  223. all_flow_recall_rank = fast_flow_recall_rank + flow_recall_rank
  224. rank_result = []
  225. rank_set = set('')
  226. # 从ROV召回池中获取top k
  227. if len(rov_recall_rank) > 0:
  228. rank_result.extend(rov_recall_rank[:top_K])
  229. rov_recall_rank = rov_recall_rank[top_K:]
  230. else:
  231. rank_result.extend(all_flow_recall_rank[:top_K])
  232. all_flow_recall_rank = all_flow_recall_rank[top_K:]
  233. #已存放了多少VID
  234. for rank_item in rank_result:
  235. rank_set.add(rank_item.get('videoId', 0))
  236. # 按概率 p 及score排序获取 size - k 个视频, 第4个位置按概率取流量池
  237. i = 0
  238. while i < size - top_K:
  239. # 随机生成[0, 1)浮点数
  240. rand = random.random()
  241. # log_.info('rand: {}'.format(rand))
  242. if rand < flow_pool_P:
  243. for flow_item in all_flow_recall_rank:
  244. flow_vid = flow_item.get('videoId', 0)
  245. if flow_vid in rank_set:
  246. continue
  247. else:
  248. rank_result.append(flow_item)
  249. rank_set.add(flow_vid)
  250. else:
  251. for recall_item in rov_recall_rank:
  252. flow_vid = recall_item.get('videoId', 0)
  253. if flow_vid in rank_set:
  254. continue
  255. else:
  256. rank_result.append(recall_item)
  257. rank_set.add(flow_vid)
  258. i += 1
  259. return rank_result[:size]
  260. def remove_duplicate(rov_recall, flow_recall, top_K):
  261. """
  262. 对多路召回的视频去重
  263. 去重原则:
  264. 如果视频在ROV召回池topK,则保留ROV召回池,否则保留流量池
  265. :param rov_recall: ROV召回池-已排序
  266. :param flow_recall: 流量池-已排序
  267. :param top_K: 保证topK为召回池视频 type-int
  268. :return:
  269. """
  270. flow_recall_result = []
  271. rov_recall_remove = []
  272. flow_recall_video_ids = [item['videoId'] for item in flow_recall]
  273. # rov_recall topK
  274. for item in rov_recall[:top_K]:
  275. if item['videoId'] in flow_recall_video_ids:
  276. flow_recall_video_ids.remove(item['videoId'])
  277. # other
  278. for item in rov_recall[top_K:]:
  279. if item['videoId'] in flow_recall_video_ids:
  280. rov_recall_remove.append(item)
  281. # rov recall remove
  282. for item in rov_recall_remove:
  283. rov_recall.remove(item)
  284. # flow recall remove
  285. for item in flow_recall:
  286. if item['videoId'] in flow_recall_video_ids:
  287. flow_recall_result.append(item)
  288. return rov_recall, flow_recall_result
  289. def bottom_strategy(request_id, size, app_type, ab_code, params):
  290. """
  291. 兜底策略: 从ROV召回池中获取top1000,进行状态过滤后的视频
  292. :param request_id: request_id
  293. :param size: 需要获取的视频数
  294. :param app_type: 产品标识 type-int
  295. :param ab_code: abCode
  296. :param params:
  297. :return:
  298. """
  299. pool_recall = PoolRecall(request_id=request_id, app_type=app_type, ab_code=ab_code)
  300. key_name, _ = pool_recall.get_pool_redis_key(pool_type='rov')
  301. redis_helper = RedisHelper(params=params)
  302. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=1000)
  303. if not data:
  304. log_.info('{} —— ROV推荐进入了二次兜底, data = {}'.format(config_.ENV_TEXT, data))
  305. send_msg_to_feishu('{} —— ROV推荐进入了二次兜底,请查看是否有数据更新失败问题。'.format(config_.ENV_TEXT))
  306. # 二次兜底
  307. bottom_data = bottom_strategy_last(size=size, app_type=app_type, ab_code=ab_code, params=params)
  308. return bottom_data
  309. # 视频状态过滤采用离线定时过滤方案
  310. # 状态过滤
  311. # filter_videos = FilterVideos(app_type=app_type, video_ids=data)
  312. # filtered_data = filter_videos.filter_video_status(video_ids=data)
  313. if len(data) > size:
  314. random_data = numpy.random.choice(data, size, False)
  315. else:
  316. random_data = data
  317. bottom_data = [{'videoId': int(item), 'pushFrom': config_.PUSH_FROM['bottom'], 'abCode': ab_code}
  318. for item in random_data]
  319. return bottom_data
  320. def bottom_strategy_last(size, app_type, ab_code, params):
  321. """
  322. 兜底策略: 从兜底视频中随机获取视频,进行状态过滤后的视频
  323. :param size: 需要获取的视频数
  324. :param app_type: 产品标识 type-int
  325. :param ab_code: abCode
  326. :param params:
  327. :return:
  328. """
  329. redis_helper = RedisHelper(params=params)
  330. bottom_data = redis_helper.get_data_zset_with_index(key_name=config_.BOTTOM_KEY_NAME, start=0, end=-1)
  331. random_data = numpy.random.choice(bottom_data, size * 30, False)
  332. # 视频状态过滤采用离线定时过滤方案
  333. # 状态过滤
  334. # filter_videos = FilterVideos(app_type=app_type, video_ids=random_data)
  335. # filtered_data = filter_videos.filter_video_status(video_ids=random_data)
  336. bottom_data = [{'videoId': int(video_id), 'pushFrom': config_.PUSH_FROM['bottom_last'], 'abCode': ab_code}
  337. for video_id in random_data[:size]]
  338. return bottom_data
  339. def bottom_strategy2(size, app_type, mid, uid, ab_code, client_info, params):
  340. """
  341. 兜底策略: 从兜底视频中随机获取视频,进行过滤后的视频
  342. :param size: 需要获取的视频数
  343. :param app_type: 产品标识 type-int
  344. :param mid: mid
  345. :param uid: uid
  346. :param ab_code: abCode
  347. :param client_info: 地域信息
  348. :param params:
  349. :return:
  350. """
  351. # 获取存在城市分组数据的城市编码列表
  352. city_code_list = [code for _, code in config_.CITY_CODE.items()]
  353. # 获取provinceCode
  354. province_code = client_info.get('provinceCode', '-1')
  355. # 获取cityCode
  356. city_code = client_info.get('cityCode', '-1')
  357. if city_code in city_code_list:
  358. # 分城市数据存在时,获取城市分组数据
  359. region_code = city_code
  360. else:
  361. region_code = province_code
  362. if region_code == '':
  363. region_code = '-1'
  364. redis_helper = RedisHelper(params=params)
  365. bottom_data = redis_helper.get_data_from_set(key_name=config_.BOTTOM2_KEY_NAME)
  366. bottom_result = []
  367. if bottom_data is None:
  368. return bottom_result
  369. if len(bottom_data) > 0:
  370. try:
  371. random_data = numpy.random.choice(bottom_data, size * 5, False)
  372. except Exception as e:
  373. random_data = bottom_data
  374. video_ids = [int(item) for item in random_data]
  375. # 过滤
  376. filter_ = FilterVideos(request_id=params.request_id, app_type=app_type, mid=mid, uid=uid, video_ids=video_ids)
  377. filtered_data = filter_.filter_videos(pool_type='flow', region_code=region_code)
  378. if filtered_data:
  379. bottom_result = [{'videoId': int(video_id), 'pushFrom': config_.PUSH_FROM['bottom2'], 'abCode': ab_code}
  380. for video_id in filtered_data[:size]]
  381. return bottom_result
  382. def video_rank_by_w_h_rate(videos):
  383. """
  384. 视频宽高比实验(每组的前两个视频调整为横屏视频),根据视频宽高比信息对视频进行重排
  385. :param videos:
  386. :return:
  387. """
  388. redis_helper = RedisHelper()
  389. # ##### 判断前两个视频是否是置顶视频 或者 流量池视频
  390. top_2_push_from_flag = [False, False]
  391. for i, video in enumerate(videos[:2]):
  392. if video['pushFrom'] in [config_.PUSH_FROM['top'], config_.PUSH_FROM['flow_recall']]:
  393. top_2_push_from_flag[i] = True
  394. if top_2_push_from_flag[0] and top_2_push_from_flag[1]:
  395. return videos
  396. # ##### 判断前两个视频是否为横屏
  397. top_2_w_h_rate_flag = [False, False]
  398. for i, video in enumerate(videos[:2]):
  399. if video['pushFrom'] in [config_.PUSH_FROM['top'], config_.PUSH_FROM['flow_recall']]:
  400. # 视频来源为置顶 或 流量池时,不做判断
  401. top_2_w_h_rate_flag[i] = True
  402. elif video['pushFrom'] in [config_.PUSH_FROM['rov_recall'], config_.PUSH_FROM['bottom']]:
  403. # 视频来源为 rov召回池 或 一层兜底时,判断是否是横屏
  404. w_h_rate = redis_helper.get_score_with_value(
  405. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['rov_recall'], value=video['videoId'])
  406. if w_h_rate is not None:
  407. top_2_w_h_rate_flag[i] = True
  408. elif video['pushFrom'] == config_.PUSH_FROM['bottom_last']:
  409. # 视频来源为 二层兜底时,判断是否是横屏
  410. w_h_rate = redis_helper.get_score_with_value(
  411. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['bottom_last'], value=video['videoId'])
  412. if w_h_rate is not None:
  413. top_2_w_h_rate_flag[i] = True
  414. if top_2_w_h_rate_flag[0] and top_2_w_h_rate_flag[1]:
  415. return videos
  416. # ##### 前两个视频中有不符合前面两者条件的,对视频进行位置调整
  417. # 记录横屏视频位置
  418. horizontal_video_index = []
  419. # 记录流量池视频位置
  420. flow_video_index = []
  421. # 记录置顶视频位置
  422. top_video_index = []
  423. for i, video in enumerate(videos):
  424. # 视频来源为置顶
  425. if video['pushFrom'] == config_.PUSH_FROM['top']:
  426. top_video_index.append(i)
  427. # 视频来源为流量池
  428. elif video['pushFrom'] == config_.PUSH_FROM['flow_recall']:
  429. flow_video_index.append(i)
  430. # 视频来源为rov召回池 或 一层兜底
  431. elif video['pushFrom'] in [config_.PUSH_FROM['rov_recall'], config_.PUSH_FROM['bottom']]:
  432. w_h_rate = redis_helper.get_score_with_value(
  433. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['rov_recall'], value=video['videoId'])
  434. if w_h_rate is not None:
  435. horizontal_video_index.append(i)
  436. else:
  437. continue
  438. # 视频来源为 二层兜底
  439. elif video['pushFrom'] == config_.PUSH_FROM['bottom_last']:
  440. w_h_rate = redis_helper.get_score_with_value(
  441. key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['bottom_last'], value=video['videoId'])
  442. if w_h_rate is not None:
  443. horizontal_video_index.append(i)
  444. else:
  445. continue
  446. # 重新排序
  447. top2_index = []
  448. for i in range(2):
  449. if i in top_video_index:
  450. top2_index.append(i)
  451. elif i in flow_video_index:
  452. top2_index.append(i)
  453. flow_video_index.remove(i)
  454. elif i in horizontal_video_index:
  455. top2_index.append(i)
  456. horizontal_video_index.remove(i)
  457. elif len(horizontal_video_index) > 0:
  458. # 调整横屏视频到第一位
  459. top2_index.append(horizontal_video_index[0])
  460. # 从横屏位置记录中移除
  461. horizontal_video_index.pop(0)
  462. elif i == 0:
  463. return videos
  464. # 重排
  465. flow_result = [videos[i] for i in flow_video_index]
  466. other_result = [videos[i] for i in range(len(videos)) if i not in top2_index and i not in flow_video_index]
  467. top2_result = []
  468. for i, j in enumerate(top2_index):
  469. item = videos[j]
  470. if i != j:
  471. # 修改abCode
  472. item['abCode'] = config_.AB_CODE['w_h_rate']
  473. top2_result.append(item)
  474. new_rank_result = top2_result
  475. for i in range(len(top2_index), len(videos)):
  476. if i in flow_video_index:
  477. new_rank_result.append(flow_result[0])
  478. flow_result.pop(0)
  479. else:
  480. new_rank_result.append(other_result[0])
  481. other_result.pop(0)
  482. return new_rank_result
  483. def video_rank_with_old_video(rank_result, old_video_recall, size, top_K, old_video_index=2):
  484. """
  485. 视频分发排序 - 包含老视频, 老视频插入固定位置
  486. :param rank_result: 排序后的结果
  487. :param size: 请求数
  488. :param old_video_index: 老视频插入的位置索引,默认为2
  489. :return: new_rank_result
  490. """
  491. if not old_video_recall:
  492. return rank_result
  493. if not rank_result:
  494. return old_video_recall[:size]
  495. # 视频去重
  496. rank_video_ids = [item['videoId'] for item in rank_result]
  497. old_video_remove = []
  498. for old_video in old_video_recall:
  499. if old_video['videoId'] in rank_video_ids:
  500. old_video_remove.append(old_video)
  501. for item in old_video_remove:
  502. old_video_recall.remove(item)
  503. if not old_video_recall:
  504. return rank_result
  505. # 插入老视频
  506. # 随机获取一个视频
  507. ind = random.randint(0, len(old_video_recall) - 1)
  508. old_video = old_video_recall[ind]
  509. # 插入
  510. if len(rank_result) < top_K:
  511. new_rank_result = rank_result + [old_video]
  512. else:
  513. new_rank_result = rank_result[:old_video_index] + [old_video] + rank_result[old_video_index:]
  514. if len(new_rank_result) > size:
  515. # 判断后两位视频来源
  516. push_from_1 = new_rank_result[-1]['pushFrom']
  517. push_from_2 = new_rank_result[-2]['pushFrom']
  518. if push_from_2 == config_.PUSH_FROM['rov_recall'] and push_from_1 == config_.PUSH_FROM['flow_recall']:
  519. new_rank_result = new_rank_result[:-2] + new_rank_result[-1:]
  520. return new_rank_result[:size]
  521. if __name__ == '__main__':
  522. d_test = [{'videoId': 10028734, 'rovScore': 99.977, 'pushFrom': 'recall_pool', 'abCode': 10000},
  523. {'videoId': 1919925, 'rovScore': 99.974, 'pushFrom': 'recall_pool', 'abCode': 10000},
  524. {'videoId': 9968118, 'rovScore': 99.972, 'pushFrom': 'recall_pool', 'abCode': 10000},
  525. {'videoId': 9934863, 'rovScore': 99.971, 'pushFrom': 'recall_pool', 'abCode': 10000},
  526. {'videoId': 10219869, 'flowPool': '1#1#1#1640830818883', 'rovScore': 82.21929728934731, 'pushFrom': 'flow_pool', 'abCode': 10000},
  527. {'videoId': 10212814, 'flowPool': '1#1#1#1640759014984', 'rovScore': 81.26694187726412, 'pushFrom': 'flow_pool', 'abCode': 10000},
  528. {'videoId': 10219437, 'flowPool': '1#1#1#1640827620520', 'rovScore': 81.21634156641908, 'pushFrom': 'flow_pool', 'abCode': 10000},
  529. {'videoId': 1994050, 'rovScore': 99.97, 'pushFrom': 'recall_pool', 'abCode': 10000},
  530. {'videoId': 9894474, 'rovScore': 99.969, 'pushFrom': 'recall_pool', 'abCode': 10000},
  531. {'videoId': 10028081, 'rovScore': 99.966, 'pushFrom': 'recall_pool', 'abCode': 10000}]
  532. res = video_rank_by_w_h_rate(videos=d_test)
  533. for tmp in res:
  534. print(tmp)