videos_filter.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746
  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. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  126. if len(filter_videos) == 0:
  127. log_.info("rov recall pool filter end!")
  128. return
  129. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  130. log_.info("rov recall pool filter end!")
  131. def filter_flow_pool():
  132. """流量池视频过滤"""
  133. log_.info("flow pool filter start ...")
  134. app_type_list = [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]
  135. for _, app_type in config_.APP_TYPE.items():
  136. log_.info('app_type {} videos filter start...'.format(app_type))
  137. if app_type in app_type_list:
  138. filter_flow_pool_18_19(app_type=app_type)
  139. else:
  140. # 拼接redis-key
  141. key_name = get_pool_redis_key(pool_type='flow', app_type=app_type)
  142. # 获取视频
  143. redis_helper = RedisHelper()
  144. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  145. if data is None:
  146. log_.info("data is None")
  147. log_.info("app_type {} videos filter end!".format(app_type))
  148. continue
  149. # videoId与flowPool做mapping
  150. video_ids = []
  151. mapping = {}
  152. for video in data:
  153. video_id, flow_pool = video.split('-')
  154. video_id = int(video_id)
  155. if video_id not in video_ids:
  156. video_ids.append(video_id)
  157. mapping[video_id] = [flow_pool]
  158. else:
  159. mapping[video_id].append(flow_pool)
  160. # 过滤
  161. if len(video_ids) == 0:
  162. log_.info("data size = {}, video_ids size = {}, data = {}".format(len(data), len(video_ids), data))
  163. log_.info("app_type {} videos filter end!".format(app_type))
  164. continue
  165. if app_type == config_.APP_TYPE['APP']:
  166. filtered_result = filter_video_status_app(video_ids=video_ids)
  167. else:
  168. filtered_result = filter_video_status(video_ids=video_ids)
  169. # 求差集,获取需要过滤掉的视频,并从redis中移除
  170. filter_videos = set(video_ids) - set(filtered_result)
  171. log_.info("data size = {}, video_ids size = {}, filtered size = {}, filter sizer = {}".format(
  172. len(data), len(video_ids), len(filtered_result), len(filter_videos)))
  173. # 移除
  174. if len(filter_videos) == 0:
  175. log_.info("app_type {} videos filter end!".format(app_type))
  176. continue
  177. remove_videos = ['{}-{}'.format(video_id, flow_pool)
  178. for video_id in filter_videos
  179. for flow_pool in mapping[video_id]]
  180. redis_helper.remove_value_from_zset(key_name=key_name, value=remove_videos)
  181. log_.info("app_type {} videos filter end!".format(app_type))
  182. log_.info("flow pool filter end!")
  183. def filter_flow_pool_18_19(app_type):
  184. """流量池视频过滤"""
  185. log_.info('app_type {} videos filter start...'.format(app_type))
  186. # 拼接redis-key
  187. key_name = get_pool_redis_key(pool_type='flow', app_type=app_type)
  188. # 获取视频
  189. redis_helper = RedisHelper()
  190. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  191. if data is None:
  192. log_.info("data is None")
  193. log_.info("app_type {} videos filter end!".format(app_type))
  194. return
  195. video_ids = [int(video_id) for video_id in data]
  196. # 过滤
  197. if len(video_ids) == 0:
  198. log_.info("data size = {}, video_ids size = {}, data = {}".format(len(data), len(video_ids), data))
  199. log_.info("app_type {} videos filter end!".format(app_type))
  200. return
  201. filtered_result = filter_video_status(video_ids=video_ids)
  202. # 求差集,获取需要过滤掉的视频,并从redis中移除
  203. filter_videos = set(video_ids) - set(filtered_result)
  204. log_.info("data size = {}, video_ids size = {}, filtered size = {}, filter sizer = {}".format(
  205. len(data), len(video_ids), len(filtered_result), len(filter_videos)))
  206. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  207. # 移除
  208. if len(filter_videos) == 0:
  209. log_.info("app_type {} videos filter end!".format(app_type))
  210. return
  211. redis_helper.remove_value_from_zset(key_name=key_name, value=filter_videos)
  212. log_.info("app_type {} videos filter end!".format(app_type))
  213. log_.info("flow pool filter end!")
  214. def filter_bottom():
  215. """兜底视频过滤"""
  216. log_.info("bottom videos filter start ...")
  217. # 获取视频
  218. redis_helper = RedisHelper()
  219. data = redis_helper.get_data_zset_with_index(key_name=config_.BOTTOM_KEY_NAME, start=0, end=-1)
  220. if data is None:
  221. log_.info("data is None")
  222. log_.info("bottom videos filter end!")
  223. return
  224. # 过滤
  225. video_ids = [int(video_id) for video_id in data]
  226. filtered_result = filter_video_status(video_ids=video_ids)
  227. # 求差集,获取需要过滤掉的视频,并从redis中移除
  228. filter_videos = set(video_ids) - set(filtered_result)
  229. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  230. len(filtered_result),
  231. len(filter_videos)))
  232. if len(filter_videos) == 0:
  233. log_.info("bottom videos filter end!")
  234. return
  235. redis_helper.remove_value_from_zset(key_name=config_.BOTTOM_KEY_NAME, value=list(filter_videos))
  236. log_.info("bottom videos filter end!")
  237. def filter_rov_updated():
  238. """修改过ROV的视频过滤"""
  239. log_.info("update rov videos filter start ...")
  240. # 获取视频
  241. redis_helper = RedisHelper()
  242. data = redis_helper.get_data_zset_with_index(key_name=config_.UPDATE_ROV_KEY_NAME, start=0, end=-1)
  243. if data is None:
  244. log_.info("data is None")
  245. log_.info("update rov videos filter end!")
  246. return
  247. # 过滤
  248. video_ids = [int(video_id) for video_id in data]
  249. filtered_result = filter_video_status(video_ids=video_ids)
  250. # 求差集,获取需要过滤掉的视频,并从redis中移除
  251. filter_videos = set(video_ids) - set(filtered_result)
  252. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  253. len(filtered_result),
  254. len(filter_videos)))
  255. log_.info({'key_name': config_.UPDATE_ROV_KEY_NAM, 'filter_videos': filter_videos})
  256. if len(filter_videos) == 0:
  257. log_.info("update rov videos filter end!")
  258. return
  259. redis_helper.remove_value_from_zset(key_name=config_.UPDATE_ROV_KEY_NAME, value=list(filter_videos))
  260. log_.info("update rov videos filter end!")
  261. def filter_rov_updated_app():
  262. """修改过ROV的视频过滤-app推荐状态过滤"""
  263. log_.info("update rov videos app filter start ...")
  264. # 获取视频
  265. redis_helper = RedisHelper()
  266. data = redis_helper.get_data_zset_with_index(key_name=config_.UPDATE_ROV_KEY_NAME_APP, start=0, end=-1)
  267. if data is None:
  268. log_.info("data is None")
  269. log_.info("update rov videos app filter end!")
  270. return
  271. # 过滤
  272. video_ids = [int(video_id) for video_id in data]
  273. filtered_result = filter_video_status_app(video_ids=video_ids)
  274. # 求差集,获取需要过滤掉的视频,并从redis中移除
  275. filter_videos = set(video_ids) - set(filtered_result)
  276. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  277. len(filtered_result),
  278. len(filter_videos)))
  279. if len(filter_videos) == 0:
  280. log_.info("update rov videos app filter end!")
  281. return
  282. redis_helper.remove_value_from_zset(key_name=config_.UPDATE_ROV_KEY_NAME_APP, value=list(filter_videos))
  283. log_.info("update rov videos app filter end!")
  284. def get_pool_redis_key(pool_type, app_type=None):
  285. """
  286. 拼接key
  287. :param pool_type: type-string {'rov': rov召回池, 'flow': 流量池}
  288. :param app_type: 产品标识
  289. :return: key_name
  290. """
  291. redis_helper = RedisHelper()
  292. if pool_type == 'rov':
  293. # appType = 6
  294. if app_type == config_.APP_TYPE['SHORT_VIDEO']:
  295. # 获取当前所在小时
  296. redis_date = datetime.now().hour
  297. # 判断热度列表是否更新,未更新则使用前一小时的热度列表
  298. key_name = '{}{}.{}'.format(config_.RECALL_KEY_NAME_PREFIX_APP_TYPE, app_type, redis_date)
  299. if redis_helper.key_exists(key_name):
  300. return key_name, redis_date
  301. else:
  302. if redis_date == 0:
  303. redis_date = 23
  304. else:
  305. redis_date = redis_date - 1
  306. key_name = '{}{}.{}'.format(config_.RECALL_KEY_NAME_PREFIX_APP_TYPE, app_type, redis_date)
  307. return key_name, redis_date
  308. # appType: [18, 19]
  309. elif app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
  310. key_name_prefix = f'{config_.RECALL_KEY_NAME_PREFIX_APP_TYPE}{app_type}.'
  311. now_date = datetime.today().strftime('%Y%m%d')
  312. now_h = datetime.now().hour
  313. key_name = f"{key_name_prefix}{now_date}.{now_h}"
  314. if redis_helper.key_exists(key_name):
  315. return key_name, now_h
  316. else:
  317. if now_h == 0:
  318. redis_h = 23
  319. redis_date = (datetime.today() - timedelta(days=1)).strftime('%Y%m%d')
  320. else:
  321. redis_h = now_h - 1
  322. redis_date = now_date
  323. key_name = f"{key_name_prefix}{redis_date}.{redis_h}"
  324. return key_name, redis_h
  325. else:
  326. # appType = 13 票圈视频app
  327. if app_type == config_.APP_TYPE['APP']:
  328. key_name_prefix = config_.RECALL_KEY_NAME_PREFIX_APP
  329. # # appType: [18, 19]
  330. # elif app_type in [config_.APP_TYPE['LAO_HAO_KAN_VIDEO'], config_.APP_TYPE['ZUI_JING_QI']]:
  331. # key_name_prefix = f'{config_.RECALL_KEY_NAME_PREFIX_APP_TYPE}{app_type}.'
  332. # 其他
  333. else:
  334. key_name_prefix = config_.RECALL_KEY_NAME_PREFIX
  335. # 判断热度列表是否更新,未更新则使用前一天的热度列表
  336. key_name = key_name_prefix + time.strftime('%Y%m%d')
  337. if redis_helper.key_exists(key_name):
  338. redis_date = date.today().strftime('%Y%m%d')
  339. else:
  340. redis_date = (date.today() - timedelta(days=1)).strftime('%Y%m%d')
  341. key_name = key_name_prefix + redis_date
  342. return key_name, redis_date
  343. elif pool_type == 'flow':
  344. # 流量池
  345. return config_.FLOWPOOL_KEY_NAME_PREFIX + str(app_type)
  346. else:
  347. log_.error('pool type error')
  348. return None, None
  349. def filter_app_pool():
  350. """过滤票圈视频APP小时级数据"""
  351. log_.info("app pool filter start ...")
  352. redis_helper = RedisHelper()
  353. # 获取当前日期
  354. now_date = date.today().strftime('%Y%m%d')
  355. # 获取当前所在小时
  356. now_h = datetime.now().hour
  357. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  358. if now_h < 7:
  359. redis_date = (date.today() - timedelta(days=1)).strftime('%Y%m%d')
  360. redis_h = 21
  361. elif now_h > 21:
  362. redis_date = now_date
  363. redis_h = 21
  364. else:
  365. if now_h % 2 == 0:
  366. redis_date = now_date
  367. redis_h = now_h - 1
  368. else:
  369. redis_date = now_date
  370. redis_h = now_h
  371. log_.info(f'redis_date = {redis_date}, redis_h = {redis_h}.')
  372. # 拼接key
  373. key_name = f'{config_.APP_FINAL_RECALL_KEY_NAME_PREFIX}{redis_date}.{redis_h}'
  374. # 获取视频
  375. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  376. if data is None:
  377. log_.info("data is None")
  378. log_.info("app pool filter end!")
  379. return
  380. # 过滤
  381. video_ids = [int(video_id) for video_id in data]
  382. filtered_result = filter_video_status_app(video_ids=video_ids)
  383. # 求差集,获取需要过滤掉的视频,并从redis中移除
  384. filter_videos = set(video_ids) - set(filtered_result)
  385. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  386. len(filtered_result),
  387. len(filter_videos)))
  388. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  389. if len(filter_videos) == 0:
  390. log_.info("app pool filter end!")
  391. return
  392. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  393. log_.info("app pool filter end!")
  394. def filter_rov_h():
  395. """过滤小程序小时级数据"""
  396. rule_params = config_.RULE_PARAMS
  397. log_.info("rov_h pool filter start ...")
  398. redis_helper = RedisHelper()
  399. # 获取当前日期
  400. now_date = date.today().strftime('%Y%m%d')
  401. # 获取当前所在小时
  402. now_h = datetime.now().hour
  403. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  404. for key, value in rule_params.items():
  405. log_.info(f"rule = {key}, param = {value}")
  406. # 需过滤两个视频列表
  407. key_prefix_list = [config_.RECALL_KEY_NAME_PREFIX_BY_H, config_.RECALL_KEY_NAME_PREFIX_DUP_H]
  408. for i, key_prefix in enumerate(key_prefix_list):
  409. # 拼接key
  410. key_name = f"{key_prefix}{key}.{now_date}.{now_h}"
  411. log_.info(f"key_name: {key_name}")
  412. # 获取视频
  413. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  414. if data is None:
  415. log_.info("data is None")
  416. log_.info("filter end!")
  417. continue
  418. # 过滤
  419. video_ids = [int(video_id) for video_id in data]
  420. filtered_result = filter_video_status(video_ids=video_ids)
  421. # 求差集,获取需要过滤掉的视频,并从redis中移除
  422. filter_videos = set(video_ids) - set(filtered_result)
  423. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  424. len(filtered_result),
  425. len(filter_videos)))
  426. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  427. if len(filter_videos) == 0:
  428. log_.info("filter end!")
  429. continue
  430. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  431. if i == 0:
  432. # 将小时级的数据需要过滤的视频加入到线上过滤应用列表中
  433. redis_helper.add_data_with_set(key_name=f"{config_.H_VIDEO_FILER}{key}",
  434. values=filter_videos, expire_time=2*3600)
  435. log_.info("rov_h pool filter end!")
  436. def filter_rov_day():
  437. """过滤小程序天级数据"""
  438. rule_params = config_.RULE_PARAMS_DAY
  439. log_.info("rov_day pool filter start ...")
  440. redis_helper = RedisHelper()
  441. # 获取当前日期
  442. now_date = date.today().strftime('%Y%m%d')
  443. log_.info(f'now_date = {now_date}.')
  444. for key, value in rule_params.items():
  445. log_.info(f"rule = {key}, param = {value}")
  446. # 需过滤三个视频列表
  447. key_prefix_list = [
  448. config_.RECALL_KEY_NAME_PREFIX_BY_DAY,
  449. config_.RECALL_KEY_NAME_PREFIX_DUP_DAY_PRE,
  450. config_.RECALL_KEY_NAME_PREFIX_DUP_DAY_NOW
  451. ]
  452. for i, key_prefix in enumerate(key_prefix_list):
  453. # 拼接key
  454. key_name = f"{key_prefix}{key}.{now_date}"
  455. log_.info(f"key_name: {key_name}")
  456. # 获取视频
  457. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  458. if data is None:
  459. log_.info("data is None")
  460. log_.info("filter end!")
  461. continue
  462. # 过滤
  463. video_ids = [int(video_id) for video_id in data]
  464. filtered_result = filter_video_status(video_ids=video_ids)
  465. # 求差集,获取需要过滤掉的视频,并从redis中移除
  466. filter_videos = set(video_ids) - set(filtered_result)
  467. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  468. len(filtered_result),
  469. len(filter_videos)))
  470. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  471. if len(filter_videos) == 0:
  472. log_.info("filter end!")
  473. continue
  474. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  475. log_.info("rov_day pool filter end!")
  476. def filter_old_videos():
  477. """过滤老视频数据"""
  478. log_.info("old videos filter start ...")
  479. redis_helper = RedisHelper()
  480. # 获取当前日期
  481. now_date = date.today().strftime('%Y%m%d')
  482. log_.info(f'now_date = {now_date}.')
  483. # 拼接key
  484. key_name = f'{config_.RECALL_KEY_NAME_PREFIX_OLD_VIDEOS}{now_date}'
  485. # 获取视频
  486. data = redis_helper.get_data_from_set(key_name=key_name)
  487. if data is None:
  488. log_.info("data is None")
  489. log_.info("old videos filter end!")
  490. return
  491. # 过滤
  492. video_ids = [int(video_id) for video_id in data]
  493. filtered_result = filter_video_status(video_ids=video_ids)
  494. # 求差集,获取需要过滤掉的视频,并从redis中移除
  495. filter_videos = set(video_ids) - set(filtered_result)
  496. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  497. len(filtered_result),
  498. len(filter_videos)))
  499. if len(filter_videos) == 0:
  500. log_.info("old videos filter end!")
  501. return
  502. redis_helper.remove_value_from_set(key_name=key_name, values=filter_videos)
  503. log_.info("old videos filter end!")
  504. def filter_region_videos():
  505. """过滤地域分组规则视频"""
  506. region_code_list = [code for region, code in region_code.items()]
  507. rule_params = config_.RULE_PARAMS_REGION
  508. log_.info("region_h videos filter start ...")
  509. redis_helper = RedisHelper()
  510. # 获取当前日期
  511. now_date = date.today().strftime('%Y%m%d')
  512. # 获取当前所在小时
  513. now_h = datetime.now().hour
  514. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  515. for region in region_code_list:
  516. log_.info(f"region = {region}")
  517. for key, value in rule_params.items():
  518. log_.info(f"rule = {key}, param = {value}")
  519. # 需过滤视频列表
  520. key_prefix_list = [
  521. config_.RECALL_KEY_NAME_PREFIX_REGION_BY_H,
  522. config_.RECALL_KEY_NAME_PREFIX_DUP1_REGION_DAY_H,
  523. config_.RECALL_KEY_NAME_PREFIX_DUP2_REGION_DAY_H,
  524. config_.RECALL_KEY_NAME_PREFIX_DUP_REGION_H
  525. ]
  526. for i, key_prefix in enumerate(key_prefix_list):
  527. # 拼接key
  528. key_name = f"{key_prefix}{region}.{key}.{now_date}.{now_h}"
  529. log_.info(f"key_name: {key_name}")
  530. # 获取视频
  531. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  532. if data is None:
  533. log_.info("data is None")
  534. log_.info("filter end!")
  535. continue
  536. # 过滤
  537. video_ids = [int(video_id) for video_id in data]
  538. filtered_result = filter_video_status(video_ids=video_ids)
  539. # 求差集,获取需要过滤掉的视频,并从redis中移除
  540. filter_videos = set(video_ids) - set(filtered_result)
  541. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  542. len(filtered_result),
  543. len(filter_videos)))
  544. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  545. if len(filter_videos) == 0:
  546. log_.info("filter end!")
  547. continue
  548. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  549. if i == 0:
  550. # 将小时级的数据需要过滤的视频加入到线上过滤应用列表中
  551. redis_helper.add_data_with_set(key_name=f"{config_.REGION_H_VIDEO_FILER}{region}.{key}",
  552. values=filter_videos, expire_time=2 * 3600)
  553. log_.info(f"region = {region} videos filter end!")
  554. log_.info("region_h videos filter end!")
  555. def filter_region_videos_by_day():
  556. """过滤地域分组天级规则视频"""
  557. region_code_list = [code for region, code in region_code.items()]
  558. rule_params = config_.RULE_PARAMS_REGION_DAY
  559. log_.info("region_day videos filter start ...")
  560. redis_helper = RedisHelper()
  561. # 获取当前日期
  562. now_date = date.today().strftime('%Y%m%d')
  563. log_.info(f'now_date = {now_date}.')
  564. for region in region_code_list:
  565. log_.info(f"region = {region}")
  566. for key, value in rule_params.items():
  567. log_.info(f"rule = {key}, param = {value}")
  568. # 需过滤视频列表
  569. key_prefix_list = [
  570. config_.RECALL_KEY_NAME_PREFIX_REGION_BY_DAY
  571. ]
  572. for i, key_prefix in enumerate(key_prefix_list):
  573. # 拼接key
  574. key_name = f"{key_prefix}{region}.{key}.{now_date}"
  575. log_.info(f"key_name: {key_name}")
  576. # 获取视频
  577. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  578. if data is None:
  579. log_.info("data is None")
  580. log_.info("filter end!")
  581. continue
  582. # 过滤
  583. video_ids = [int(video_id) for video_id in data]
  584. filtered_result = filter_video_status(video_ids=video_ids)
  585. # 求差集,获取需要过滤掉的视频,并从redis中移除
  586. filter_videos = set(video_ids) - set(filtered_result)
  587. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  588. len(filtered_result),
  589. len(filter_videos)))
  590. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  591. if len(filter_videos) == 0:
  592. log_.info("filter end!")
  593. continue
  594. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  595. log_.info(f"region = {region} videos filter end!")
  596. log_.info("region_day videos filter end!")
  597. def filter_rov_h_24h():
  598. """过滤小程序小时级更新24h数据"""
  599. rule_params = config_.RULE_PARAMS_24H
  600. log_.info("rov_h_by24h pool filter start ...")
  601. redis_helper = RedisHelper()
  602. # 获取当前日期
  603. now_date = date.today().strftime('%Y%m%d')
  604. # 获取当前所在小时
  605. now_h = datetime.now().hour
  606. log_.info(f'now_date = {now_date}, now_h = {now_h}.')
  607. for key, value in rule_params.items():
  608. log_.info(f"rule = {key}, param = {value}")
  609. # 需过滤两个视频列表
  610. key_prefix_list = [config_.RECALL_KEY_NAME_PREFIX_BY_24H, config_.RECALL_KEY_NAME_PREFIX_DUP_24H]
  611. for i, key_prefix in enumerate(key_prefix_list):
  612. # 拼接key
  613. key_name = f"{key_prefix}{key}.{now_date}.{now_h}"
  614. log_.info(f"key_name: {key_name}")
  615. # 获取视频
  616. data = redis_helper.get_data_zset_with_index(key_name=key_name, start=0, end=-1)
  617. if data is None:
  618. log_.info("data is None")
  619. log_.info("filter end!")
  620. continue
  621. # 过滤
  622. video_ids = [int(video_id) for video_id in data]
  623. filtered_result = filter_video_status(video_ids=video_ids)
  624. # 求差集,获取需要过滤掉的视频,并从redis中移除
  625. filter_videos = set(video_ids) - set(filtered_result)
  626. log_.info("video_ids size = {}, filtered size = {}, filter sizer = {}".format(len(video_ids),
  627. len(filtered_result),
  628. len(filter_videos)))
  629. log_.info({'key_name': key_name, 'filter_videos': filter_videos})
  630. if len(filter_videos) == 0:
  631. log_.info("filter end!")
  632. continue
  633. redis_helper.remove_value_from_zset(key_name=key_name, value=list(filter_videos))
  634. if i == 0:
  635. # 将小时级的数据需要过滤的视频加入到线上过滤应用列表中
  636. redis_helper.add_data_with_set(key_name=f"{config_.H_VIDEO_FILER_24H}{key}",
  637. values=filter_videos, expire_time=2*3600)
  638. log_.info("rov_h_by24h pool filter end!")
  639. def main():
  640. try:
  641. # ROV召回池视频过滤
  642. filter_rov_pool()
  643. # appType = 6,ROV召回池视频过滤
  644. # filter_rov_pool(app_type=config_.APP_TYPE['SHORT_VIDEO'])
  645. # appType = 13,票圈视频APP视频过滤
  646. filter_rov_pool(app_type=config_.APP_TYPE['APP'])
  647. # appType = 18, ROV召回池视频过滤
  648. filter_rov_pool(app_type=config_.APP_TYPE['LAO_HAO_KAN_VIDEO'])
  649. # appType = 19, ROV召回池视频过滤
  650. filter_rov_pool(app_type=config_.APP_TYPE['ZUI_JING_QI'])
  651. # 流量池视频过滤
  652. filter_flow_pool()
  653. # 兜底视频过滤
  654. filter_bottom()
  655. # 修改过ROV的视频过滤
  656. filter_rov_updated()
  657. filter_rov_updated_app()
  658. # 运营强插相关推荐视频过滤
  659. # filter_relevant_videos()
  660. # 按位置排序视频过滤
  661. # filter_position_videos()
  662. # 过滤票圈视频APP小时级数据
  663. filter_app_pool()
  664. # 过滤小程序小时级数据
  665. filter_rov_h()
  666. # 过滤小程序天级数据
  667. filter_rov_day()
  668. # 过滤老视频数据
  669. # filter_old_videos()
  670. # 过滤地域分组小时级视频
  671. filter_region_videos()
  672. # 过滤地域分组天级视频
  673. filter_region_videos_by_day()
  674. # 过滤小时级更新24h视频
  675. filter_rov_h_24h()
  676. except Exception as e:
  677. log_.error(traceback.format_exc())
  678. send_msg_to_feishu(
  679. webhook=config_.FEISHU_ROBOT['server_robot'].get('webhook'),
  680. key_word=config_.FEISHU_ROBOT['server_robot'].get('key_word'),
  681. msg_text='{} - 过滤失败 \n {}'.format(config_.ENV_TEXT, traceback.format_exc())
  682. )
  683. return
  684. if __name__ == '__main__':
  685. main()