videos_filter.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633
  1. import time
  2. import json
  3. import traceback
  4. import ast
  5. from datetime import date, timedelta, datetime
  6. from region_rule_rank_h import region_code
  7. from utils import filter_video_status, send_msg_to_feishu, filter_video_status_app
  8. from db_helper import RedisHelper
  9. from config import set_config
  10. from log import Log
  11. config_, env = set_config()
  12. log_ = Log()
  13. def filter_position_videos():
  14. """按位置排序视频过滤"""
  15. log_.info("position videos filter start...")
  16. position_key_list = [config_.RECALL_POSITION1_KEY_NAME, config_.RECALL_POSITION2_KEY_NAME]
  17. redis_helper = RedisHelper()
  18. for key_name in position_key_list:
  19. position = key_name.split('.')[-1]
  20. log_.info("position = {}".format(position))
  21. # 获取数据
  22. position_videos = redis_helper.get_data_from_redis(key_name=key_name)
  23. if position_videos is None:
  24. log_.info('position {} videos is None!'.format(position))
  25. continue
  26. else:
  27. # 过滤
  28. position_video_ids = [int(video_id) for video_id in ast.literal_eval(position_videos)]
  29. filter_video_ids = filter_video_status(video_ids=position_video_ids)
  30. # 重新写入redis
  31. redis_helper.set_data_to_redis(key_name=key_name,
  32. value=str(filter_video_ids),
  33. expire_time=30 * 3600)
  34. log_.info('position {} videos filter end!'.format(position))
  35. log_.info("position videos filter end!")
  36. def filter_relevant_videos():
  37. """运营强插相关推荐视频过滤"""
  38. log_.info("relevant videos with op filter filter start...")
  39. # 读取需要过滤的头部视频id
  40. redis_helper = RedisHelper()
  41. head_videos = redis_helper.get_data_from_set(key_name=config_.RELEVANT_TOP_VIDEOS_KEY_NAME)
  42. if head_videos is None or len(head_videos) == 0:
  43. log_.info("relevant videos with op filter end! head_videos = {}".format(head_videos))
  44. return
  45. # 过滤
  46. remove_head_vids = []
  47. for head_vid in head_videos:
  48. key_name = '{}{}'.format(config_.RELEVANT_VIDEOS_WITH_OP_KEY_NAME, head_vid)
  49. # 头部视频 对应的key不存在时,将head_vid移除对应redis
  50. if not redis_helper.key_exists(key_name=key_name):
  51. remove_head_vids.append(head_vid)
  52. log_.info('head_vid = {} relevant redis key not exist!'.format(head_vid))
  53. continue
  54. # 获取头部视频对应的相关视频
  55. relevant_videos = redis_helper.get_data_from_redis(key_name=key_name)
  56. # 该视频没有指定的相关性视频,将head_vid移除对应redis
  57. if relevant_videos is None:
  58. remove_head_vids.append(head_vid)
  59. log_.info('head_vid = {} not have relevant videos!'.format(head_vid))
  60. continue
  61. # 过滤
  62. relevant_videos = json.loads(relevant_videos)
  63. relevant_video_ids = [int(item['recommend_vid']) for item in relevant_videos]
  64. filtered_videos = filter_video_status(video_ids=relevant_video_ids)
  65. # 保留可推荐 且生效中 的视频
  66. relevant_videos_new = [
  67. item for item in relevant_videos
  68. if int(item['recommend_vid']) in filtered_videos and int(item['finish_time']) > int(time.time())
  69. ]
  70. # 过滤后没有符合的视频,将head_vid移除对应redis,删除对应的相关推荐的key
  71. if len(relevant_videos_new) == 0:
  72. remove_head_vids.append(head_vid)
  73. redis_helper.del_keys(key_name=key_name)
  74. log_.info('head_vid = {} filtered finished! new relevant videos count = {}'.format(
  75. head_vid, len(relevant_videos_new)))
  76. continue
  77. # 重新写入redis
  78. # 以最晚结束的视频的结束时间 - 当前时间 + 5s 作为key的过期时间
  79. finish_time_list = [item['finish_time'] for item in relevant_videos_new]
  80. expire_time = max(finish_time_list) - int(time.time()) + 5
  81. if expire_time <= 0:
  82. log_.info('head_vid = {} expire_time <= 0!'.format(head_vid))
  83. continue
  84. # 存入redis
  85. redis_helper.set_data_to_redis(key_name=key_name,
  86. value=json.dumps(relevant_videos_new),
  87. expire_time=expire_time)
  88. log_.info('head_vid = {} filtered finished! new relevant videos count = {}'.format(
  89. head_vid, len(relevant_videos_new)))
  90. # 将需要移除的头部视频id进行移除
  91. if len(remove_head_vids) == 0:
  92. log_.info('head videos remove finished! remove_head_vids = {}'.format(remove_head_vids))
  93. log_.info("relevant videos with op filter end!")
  94. return
  95. redis_helper.remove_value_from_set(key_name=config_.RELEVANT_TOP_VIDEOS_KEY_NAME, values=tuple(remove_head_vids))
  96. log_.info('head videos remove finished! remove_head_vids = {}'.format(remove_head_vids))
  97. log_.info("relevant videos with op filter end!")
  98. def filter_rov_pool(app_type=None):
  99. """ROV召回池视频过滤"""
  100. log_.info("rov recall pool filter start ...")
  101. # 拼接redis-key
  102. if app_type is None:
  103. key_name, _ = get_pool_redis_key(pool_type='rov')
  104. else:
  105. log_.info("appType = {}".format(app_type))
  106. key_name, _ = get_pool_redis_key(pool_type='rov', app_type=app_type)
  107. # 获取视频
  108. redis_helper = RedisHelper()
  109. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  110. if data is None:
  111. log_.info("data is None")
  112. log_.info("rov recall pool filter end!")
  113. return
  114. # 过滤
  115. video_ids = [int(video_id) for video_id in data]
  116. if app_type == config_.APP_TYPE['APP']:
  117. filtered_result = filter_video_status_app(video_ids=video_ids)
  118. else:
  119. filtered_result = filter_video_status(video_ids=video_ids)
  120. # 求差集,获取需要过滤掉的视频,并从redis中移除
  121. filter_videos = set(video_ids) - set(filtered_result)
  122. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  123. len(filtered_result),
  124. len(filter_videos)))
  125. if len(filter_videos) == 0:
  126. log_.info("rov recall pool filter end!")
  127. return
  128. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  129. log_.info("rov recall pool filter end!")
  130. def filter_flow_pool():
  131. """流量池视频过滤"""
  132. log_.info("flow pool filter start ...")
  133. app_type_list = [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]
  134. for _, app_type in config_.APP_TYPE.items():
  135. log_.info('app_type {} videos filter start...'.format(app_type))
  136. if app_type in app_type_list:
  137. filter_flow_pool_18_19(app_type=app_type)
  138. else:
  139. # 拼接redis-key
  140. key_name = get_pool_redis_key(pool_type='flow', app_type=app_type)
  141. # 获取视频
  142. redis_helper = RedisHelper()
  143. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  144. if data is None:
  145. log_.info("data is None")
  146. log_.info("app_type {} videos filter end!".format(app_type))
  147. continue
  148. # videoId与flowPool做mapping
  149. video_ids = []
  150. mapping = {}
  151. for video in data:
  152. video_id, flow_pool = video.split('-')
  153. video_id = int(video_id)
  154. if video_id not in video_ids:
  155. video_ids.append(video_id)
  156. mapping[video_id] = [flow_pool]
  157. else:
  158. mapping[video_id].append(flow_pool)
  159. # 过滤
  160. if len(video_ids) == 0:
  161. log_.info("data size = {}, video_ids size = {}, data = {}".format(len(data), len(video_ids), data))
  162. log_.info("app_type {} videos filter end!".format(app_type))
  163. continue
  164. if app_type == config_.APP_TYPE['APP']:
  165. filtered_result = filter_video_status_app(video_ids=video_ids)
  166. else:
  167. filtered_result = filter_video_status(video_ids=video_ids)
  168. # 求差集,获取需要过滤掉的视频,并从redis中移除
  169. filter_videos = set(video_ids) - set(filtered_result)
  170. log_.info("data size = {}, video_ids size = {}, filtered size = {}, filter sizer = {}".format(
  171. len(data), len(video_ids), len(filtered_result), len(filter_videos)))
  172. # 移除
  173. if len(filter_videos) == 0:
  174. log_.info("app_type {} videos filter end!".format(app_type))
  175. continue
  176. remove_videos = ['{}-{}'.format(video_id, flow_pool)
  177. for video_id in filter_videos
  178. for flow_pool in mapping[video_id]]
  179. redis_helper.remove_value_from_zset(key_name=key_name, value=remove_videos)
  180. log_.info("app_type {} videos filter end!".format(app_type))
  181. log_.info("flow pool filter end!")
  182. def filter_flow_pool_18_19(app_type):
  183. """流量池视频过滤"""
  184. log_.info('app_type {} videos filter start...'.format(app_type))
  185. # 拼接redis-key
  186. key_name = get_pool_redis_key(pool_type='flow', app_type=app_type)
  187. # 获取视频
  188. redis_helper = RedisHelper()
  189. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  190. if data is None:
  191. log_.info("data is None")
  192. log_.info("app_type {} videos filter end!".format(app_type))
  193. return
  194. video_ids = [int(video_id) for video_id in data]
  195. # 过滤
  196. if len(video_ids) == 0:
  197. log_.info("data size = {}, video_ids size = {}, data = {}".format(len(data), len(video_ids), data))
  198. log_.info("app_type {} videos filter end!".format(app_type))
  199. return
  200. filtered_result = filter_video_status(video_ids=video_ids)
  201. # 求差集,获取需要过滤掉的视频,并从redis中移除
  202. filter_videos = set(video_ids) - set(filtered_result)
  203. log_.info("data size = {}, video_ids size = {}, filtered size = {}, filter sizer = {}".format(
  204. len(data), len(video_ids), len(filtered_result), len(filter_videos)))
  205. # 移除
  206. if len(filter_videos) == 0:
  207. log_.info("app_type {} videos filter end!".format(app_type))
  208. return
  209. redis_helper.remove_value_from_zset(key_name=key_name, value=filter_videos)
  210. log_.info("app_type {} videos filter end!".format(app_type))
  211. log_.info("flow pool filter end!")
  212. def filter_bottom():
  213. """兜底视频过滤"""
  214. log_.info("bottom videos filter start ...")
  215. # 获取视频
  216. redis_helper = RedisHelper()
  217. data = redis_helper.get_data_zset_with_index(key_name=config_.BOTTOM_KEY_NAME, start=0, end=-1)
  218. if data is None:
  219. log_.info("data is None")
  220. log_.info("bottom videos filter end!")
  221. return
  222. # 过滤
  223. video_ids = [int(video_id) for video_id in data]
  224. filtered_result = filter_video_status(video_ids=video_ids)
  225. # 求差集,获取需要过滤掉的视频,并从redis中移除
  226. filter_videos = set(video_ids) - set(filtered_result)
  227. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  228. len(filtered_result),
  229. len(filter_videos)))
  230. if len(filter_videos) == 0:
  231. log_.info("bottom videos filter end!")
  232. return
  233. redis_helper.remove_value_from_zset(key_name=config_.BOTTOM_KEY_NAME, value=list(filter_videos))
  234. log_.info("bottom videos filter end!")
  235. def filter_rov_updated():
  236. """修改过ROV的视频过滤"""
  237. log_.info("update rov videos filter start ...")
  238. # 获取视频
  239. redis_helper = RedisHelper()
  240. data = redis_helper.get_data_zset_with_index(key_name=config_.UPDATE_ROV_KEY_NAME, start=0, end=-1)
  241. if data is None:
  242. log_.info("data is None")
  243. log_.info("update rov videos filter end!")
  244. return
  245. # 过滤
  246. video_ids = [int(video_id) for video_id in data]
  247. filtered_result = filter_video_status(video_ids=video_ids)
  248. # 求差集,获取需要过滤掉的视频,并从redis中移除
  249. filter_videos = set(video_ids) - set(filtered_result)
  250. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  251. len(filtered_result),
  252. len(filter_videos)))
  253. if len(filter_videos) == 0:
  254. log_.info("update rov videos filter end!")
  255. return
  256. redis_helper.remove_value_from_zset(key_name=config_.UPDATE_ROV_KEY_NAME, value=list(filter_videos))
  257. log_.info("update rov videos filter end!")
  258. def filter_rov_updated_app():
  259. """修改过ROV的视频过滤-app推荐状态过滤"""
  260. log_.info("update rov videos app filter start ...")
  261. # 获取视频
  262. redis_helper = RedisHelper()
  263. data = redis_helper.get_data_zset_with_index(key_name=config_.UPDATE_ROV_KEY_NAME_APP, start=0, end=-1)
  264. if data is None:
  265. log_.info("data is None")
  266. log_.info("update rov videos app filter end!")
  267. return
  268. # 过滤
  269. video_ids = [int(video_id) for video_id in data]
  270. filtered_result = filter_video_status_app(video_ids=video_ids)
  271. # 求差集,获取需要过滤掉的视频,并从redis中移除
  272. filter_videos = set(video_ids) - set(filtered_result)
  273. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  274. len(filtered_result),
  275. len(filter_videos)))
  276. if len(filter_videos) == 0:
  277. log_.info("update rov videos app filter end!")
  278. return
  279. redis_helper.remove_value_from_zset(key_name=config_.UPDATE_ROV_KEY_NAME_APP, value=list(filter_videos))
  280. log_.info("update rov videos app filter end!")
  281. def get_pool_redis_key(pool_type, app_type=None):
  282. """
  283. 拼接key
  284. :param pool_type: type-string {'rov': rov召回池, 'flow': 流量池}
  285. :param app_type: 产品标识
  286. :return: key_name
  287. """
  288. redis_helper = RedisHelper()
  289. if pool_type == 'rov':
  290. # appType = 6
  291. if app_type == config_.APP_TYPE['SHORT_VIDEO']:
  292. # 获取当前所在小时
  293. redis_date = datetime.now().hour
  294. # 判断热度列表是否更新,未更新则使用前一小时的热度列表
  295. key_name = '{}{}.{}'.format(config_.RECALL_KEY_NAME_PREFIX_APP_TYPE, app_type, redis_date)
  296. if redis_helper.key_exists(key_name):
  297. return key_name, redis_date
  298. else:
  299. if redis_date == 0:
  300. redis_date = 23
  301. else:
  302. redis_date = redis_date - 1
  303. key_name = '{}{}.{}'.format(config_.RECALL_KEY_NAME_PREFIX_APP_TYPE, app_type, redis_date)
  304. return key_name, redis_date
  305. # appType: [18, 19]
  306. elif app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
  307. key_name_prefix = f'{config_.RECALL_KEY_NAME_PREFIX_APP_TYPE}{app_type}.'
  308. now_date = datetime.today().strftime('%Y%m%d')
  309. now_h = datetime.now().hour
  310. key_name = f"{key_name_prefix}{now_date}.{now_h}"
  311. if redis_helper.key_exists(key_name):
  312. return key_name, now_h
  313. else:
  314. if now_h == 0:
  315. redis_h = 23
  316. redis_date = (datetime.today() - timedelta(days=1)).strftime('%Y%m%d')
  317. else:
  318. redis_h = now_h - 1
  319. redis_date = now_date
  320. key_name = f"{key_name_prefix}{redis_date}.{redis_h}"
  321. return key_name, redis_h
  322. else:
  323. # appType = 13 票圈视频app
  324. if app_type == config_.APP_TYPE['APP']:
  325. key_name_prefix = config_.RECALL_KEY_NAME_PREFIX_APP
  326. # # appType: [18, 19]
  327. # elif app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
  328. # key_name_prefix = f'{config_.RECALL_KEY_NAME_PREFIX_APP_TYPE}{app_type}.'
  329. # 其他
  330. else:
  331. key_name_prefix = config_.RECALL_KEY_NAME_PREFIX
  332. # 判断热度列表是否更新,未更新则使用前一天的热度列表
  333. key_name = key_name_prefix + time.strftime('%Y%m%d')
  334. if redis_helper.key_exists(key_name):
  335. redis_date = date.today().strftime('%Y%m%d')
  336. else:
  337. redis_date = (date.today() - timedelta(days=1)).strftime('%Y%m%d')
  338. key_name = key_name_prefix + redis_date
  339. return key_name, redis_date
  340. elif pool_type == 'flow':
  341. # 流量池
  342. return config_.FLOWPOOL_KEY_NAME_PREFIX + str(app_type)
  343. else:
  344. log_.error('pool type error')
  345. return None, None
  346. def filter_app_pool():
  347. """过滤票圈视频APP小时级数据"""
  348. log_.info("app pool filter start ...")
  349. redis_helper = RedisHelper()
  350. # 获取当前日期
  351. now_date = date.today().strftime('%Y%m%d')
  352. # 获取当前所在小时
  353. now_h = datetime.now().hour
  354. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  355. if now_h < 7:
  356. redis_date = (date.today() - timedelta(days=1)).strftime('%Y%m%d')
  357. redis_h = 21
  358. elif now_h > 21:
  359. redis_date = now_date
  360. redis_h = 21
  361. else:
  362. if now_h % 2 == 0:
  363. redis_date = now_date
  364. redis_h = now_h - 1
  365. else:
  366. redis_date = now_date
  367. redis_h = now_h
  368. log_.info(f'redis_date = {redis_date}, redis_h = {redis_h}.')
  369. # 拼接key
  370. key_name = f'{config_.APP_FINAL_RECALL_KEY_NAME_PREFIX}{redis_date}.{redis_h}'
  371. # 获取视频
  372. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  373. if data is None:
  374. log_.info("data is None")
  375. log_.info("app pool filter end!")
  376. return
  377. # 过滤
  378. video_ids = [int(video_id) for video_id in data]
  379. filtered_result = filter_video_status_app(video_ids=video_ids)
  380. # 求差集,获取需要过滤掉的视频,并从redis中移除
  381. filter_videos = set(video_ids) - set(filtered_result)
  382. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  383. len(filtered_result),
  384. len(filter_videos)))
  385. if len(filter_videos) == 0:
  386. log_.info("app pool filter end!")
  387. return
  388. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  389. log_.info("app pool filter end!")
  390. def filter_rov_h():
  391. """过滤小程序小时级数据"""
  392. rule_params = config_.RULE_PARAMS
  393. log_.info("rov_h pool filter start ...")
  394. redis_helper = RedisHelper()
  395. # 获取当前日期
  396. now_date = date.today().strftime('%Y%m%d')
  397. # 获取当前所在小时
  398. now_h = datetime.now().hour
  399. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  400. for key, value in rule_params.items():
  401. log_.info(f"rule = {key}, param = {value}")
  402. # 需过滤两个视频列表
  403. key_prefix_list = [config_.RECALL_KEY_NAME_PREFIX_BY_H, config_.RECALL_KEY_NAME_PREFIX_DUP_H]
  404. for i, key_prefix in enumerate(key_prefix_list):
  405. # 拼接key
  406. key_name = f"{key_prefix}{key}.{now_date}.{now_h}"
  407. log_.info(f"key_name: {key_name}")
  408. # 获取视频
  409. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  410. if data is None:
  411. log_.info("data is None")
  412. log_.info("filter end!")
  413. continue
  414. # 过滤
  415. video_ids = [int(video_id) for video_id in data]
  416. filtered_result = filter_video_status(video_ids=video_ids)
  417. # 求差集,获取需要过滤掉的视频,并从redis中移除
  418. filter_videos = set(video_ids) - set(filtered_result)
  419. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  420. len(filtered_result),
  421. len(filter_videos)))
  422. if len(filter_videos) == 0:
  423. log_.info("filter end!")
  424. continue
  425. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  426. if i == 0:
  427. # 将小时级的数据需要过滤的视频加入到线上过滤应用列表中
  428. redis_helper.add_data_with_set(key_name=f"{config_.H_VIDEO_FILER}{key}",
  429. values=filter_videos, expire_time=2*3600)
  430. log_.info("rov_h pool filter end!")
  431. def filter_rov_day():
  432. """过滤小程序天级数据"""
  433. rule_params = config_.RULE_PARAMS_DAY
  434. log_.info("rov_day pool filter start ...")
  435. redis_helper = RedisHelper()
  436. # 获取当前日期
  437. now_date = date.today().strftime('%Y%m%d')
  438. log_.info(f'now_date = {now_date}.')
  439. for key, value in rule_params.items():
  440. log_.info(f"rule = {key}, param = {value}")
  441. # 需过滤三个视频列表
  442. key_prefix_list = [
  443. config_.RECALL_KEY_NAME_PREFIX_BY_DAY,
  444. config_.RECALL_KEY_NAME_PREFIX_DUP_DAY_PRE,
  445. config_.RECALL_KEY_NAME_PREFIX_DUP_DAY_NOW
  446. ]
  447. for i, key_prefix in enumerate(key_prefix_list):
  448. # 拼接key
  449. key_name = f"{key_prefix}{key}.{now_date}"
  450. log_.info(f"key_name: {key_name}")
  451. # 获取视频
  452. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  453. if data is None:
  454. log_.info("data is None")
  455. log_.info("filter end!")
  456. continue
  457. # 过滤
  458. video_ids = [int(video_id) for video_id in data]
  459. filtered_result = filter_video_status(video_ids=video_ids)
  460. # 求差集,获取需要过滤掉的视频,并从redis中移除
  461. filter_videos = set(video_ids) - set(filtered_result)
  462. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  463. len(filtered_result),
  464. len(filter_videos)))
  465. if len(filter_videos) == 0:
  466. log_.info("filter end!")
  467. continue
  468. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  469. log_.info("rov_day pool filter end!")
  470. def filter_old_videos():
  471. """过滤老视频数据"""
  472. log_.info("old videos filter start ...")
  473. redis_helper = RedisHelper()
  474. # 获取当前日期
  475. now_date = date.today().strftime('%Y%m%d')
  476. log_.info(f'now_date = {now_date}.')
  477. # 拼接key
  478. key_name = f'{config_.RECALL_KEY_NAME_PREFIX_OLD_VIDEOS}{now_date}'
  479. # 获取视频
  480. data = redis_helper.get_data_from_set(key_name=key_name)
  481. if data is None:
  482. log_.info("data is None")
  483. log_.info("old videos filter end!")
  484. return
  485. # 过滤
  486. video_ids = [int(video_id) for video_id in data]
  487. filtered_result = filter_video_status(video_ids=video_ids)
  488. # 求差集,获取需要过滤掉的视频,并从redis中移除
  489. filter_videos = set(video_ids) - set(filtered_result)
  490. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  491. len(filtered_result),
  492. len(filter_videos)))
  493. if len(filter_videos) == 0:
  494. log_.info("old videos filter end!")
  495. return
  496. redis_helper.remove_value_from_set(key_name=key_name, values=filter_videos)
  497. log_.info("old videos filter end!")
  498. def filter_region_videos():
  499. """过滤地域分组规则视频"""
  500. region_code_list = [code for region, code in region_code.items()]
  501. rule_params = config_.RULE_PARAMS_REGION
  502. log_.info("region_h videos filter start ...")
  503. redis_helper = RedisHelper()
  504. # 获取当前日期
  505. now_date = date.today().strftime('%Y%m%d')
  506. # 获取当前所在小时
  507. now_h = datetime.now().hour
  508. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  509. for region in region_code_list:
  510. log_.info(f"region = {region}")
  511. for key, value in rule_params.items():
  512. log_.info(f"rule = {key}, param = {value}")
  513. # 需过滤两个视频列表
  514. key_prefix_list = [config_.RECALL_KEY_NAME_PREFIX_REGION_BY_H, config_.RECALL_KEY_NAME_PREFIX_DUP_REGION_H]
  515. for i, key_prefix in enumerate(key_prefix_list):
  516. # 拼接key
  517. key_name = f"{key_prefix}{region}.{key}.{now_date}.{now_h}"
  518. log_.info(f"key_name: {key_name}")
  519. # 获取视频
  520. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  521. if data is None:
  522. log_.info("data is None")
  523. log_.info("filter end!")
  524. continue
  525. # 过滤
  526. video_ids = [int(video_id) for video_id in data]
  527. filtered_result = filter_video_status(video_ids=video_ids)
  528. # 求差集,获取需要过滤掉的视频,并从redis中移除
  529. filter_videos = set(video_ids) - set(filtered_result)
  530. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  531. len(filtered_result),
  532. len(filter_videos)))
  533. if len(filter_videos) == 0:
  534. log_.info("filter end!")
  535. continue
  536. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  537. if i == 0:
  538. # 将小时级的数据需要过滤的视频加入到线上过滤应用列表中
  539. redis_helper.add_data_with_set(key_name=f"{config_.REGION_H_VIDEO_FILER}{region}.{key}",
  540. values=filter_videos, expire_time=2 * 3600)
  541. log_.info(f"region = {region} videos filter end!")
  542. log_.info("region_h videos filter end!")
  543. def main():
  544. try:
  545. # ROV召回池视频过滤
  546. filter_rov_pool()
  547. # appType = 6,ROV召回池视频过滤
  548. # filter_rov_pool(app_type=config_.APP_TYPE['SHORT_VIDEO'])
  549. # appType = 13,票圈视频APP视频过滤
  550. filter_rov_pool(app_type=config_.APP_TYPE['APP'])
  551. # appType = 18, ROV召回池视频过滤
  552. filter_rov_pool(app_type=config_.APP_TYPE['LAO_HAO_KAN_VIDEO'])
  553. # appType = 19, ROV召回池视频过滤
  554. filter_rov_pool(app_type=config_.APP_TYPE['ZUI_JING_QI'])
  555. # 流量池视频过滤
  556. filter_flow_pool()
  557. # 兜底视频过滤
  558. filter_bottom()
  559. # 修改过ROV的视频过滤
  560. filter_rov_updated()
  561. filter_rov_updated_app()
  562. # 运营强插相关推荐视频过滤
  563. filter_relevant_videos()
  564. # 按位置排序视频过滤
  565. filter_position_videos()
  566. # 过滤票圈视频APP小时级数据
  567. filter_app_pool()
  568. # 过滤小程序小时级数据
  569. filter_rov_h()
  570. # 过滤小程序天级数据
  571. filter_rov_day()
  572. # 过滤老视频数据
  573. filter_old_videos()
  574. # 过滤地域分组小时级视频
  575. filter_region_videos()
  576. except Exception as e:
  577. log_.error(traceback.format_exc())
  578. send_msg_to_feishu(
  579. webhook=config_.FEISHU_ROBOT['server_robot'].get('webhook'),
  580. key_word=config_.FEISHU_ROBOT['server_robot'].get('key_word'),
  581. msg_text='{} - 过滤失败 \n {}'.format(config_.ENV_TEXT, traceback.format_exc())
  582. )
  583. return
  584. if __name__ == '__main__':
  585. main()