utils.py 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919
  1. import traceback
  2. import requests
  3. import json
  4. import time
  5. import gevent
  6. import pandas as pd
  7. import random
  8. from datetime import datetime
  9. # from db_helper import HologresHelper, RedisHelper, MysqlHelper
  10. from db_helper import RedisHelper, MysqlHelper
  11. from config import set_config
  12. from log import Log
  13. from parameter_update import param_update_risk_rule
  14. from parameter_update import param_update_risk_videos
  15. from parameter_update import param_update_risk_filter_flag
  16. config_ = set_config()
  17. log_ = Log()
  18. def send_msg_to_feishu(msg_text):
  19. """发送消息到飞书"""
  20. # webhook地址
  21. webhook = 'https://open.feishu.cn/open-apis/bot/v2/hook/8de4de35-30ed-4692-8854-7a154e89b2f2'
  22. # 自定义关键词key_word
  23. key_word = '服务报警'
  24. headers = {'Content-Type': 'application/json'}
  25. payload_message = {
  26. "msg_type": "text",
  27. "content": {
  28. "text": '{}: {}'.format(key_word, msg_text)
  29. }
  30. }
  31. response = requests.request('POST', url=webhook, headers=headers, data=json.dumps(payload_message))
  32. # print(response.text)
  33. def request_post(request_url, request_data, timeout):
  34. """
  35. post 请求 HTTP接口
  36. :param request_url: 接口URL
  37. :param request_data: 请求参数
  38. :param timeout: 超时时间,单位为秒,type-float or tuple(connect_timeout, read_timeout)
  39. :return: res_data json格式
  40. """
  41. try:
  42. headers = {"Connection": "close"}
  43. #print(request_url)
  44. #print(headers)
  45. response = requests.post(url=request_url, json=request_data, timeout=timeout, headers=headers)
  46. #print("response:", response)
  47. if response.status_code == 200:
  48. res_data = json.loads(response.text)
  49. return res_data
  50. else:
  51. return None
  52. except Exception as e:
  53. #print(e)
  54. log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
  55. return None
  56. def request_post_data(request_url, request_data, timeout):
  57. """
  58. post 请求 HTTP接口
  59. :param request_url: 接口URL
  60. :param request_data: 请求参数
  61. :param timeout: 超时时间,单位为秒,type-float or tuple(connect_timeout, read_timeout)
  62. :return: res_data json格式
  63. """
  64. try:
  65. headers = {'content-type': 'application/json'}
  66. response = requests.post(url=request_url, data=request_data, timeout=timeout, headers=headers)
  67. #print("response:", response)
  68. if response.status_code == 200:
  69. res_data = json.loads(response.text)
  70. return res_data['outputs']
  71. else:
  72. return None
  73. except Exception as e:
  74. #print(e)
  75. log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
  76. return None
  77. def request_get(request_url, timeout):
  78. """
  79. get 请求 HTTP接口
  80. :param request_url: 接口URL
  81. :param timeout: 超时时间,单位为秒,type-float or tuple(connect_timeout, read_timeout)
  82. :return: res_data json格式
  83. """
  84. try:
  85. response = requests.get(url=request_url, timeout=timeout)
  86. if response.status_code == 200:
  87. res_data = json.loads(response.text)
  88. return res_data
  89. else:
  90. return None
  91. except Exception as e:
  92. log_.error('url: {}, exception: {}, traceback: {}'.format(request_url, e, traceback.format_exc()))
  93. return None
  94. def get_user_has30day_return(mid):
  95. """
  96. 获取用户近30天是否有回流
  97. :param mid: mid
  98. :return: data, type
  99. """
  100. if not mid:
  101. return None
  102. # 获取redis中存储的状态值
  103. user_key = f"{config_.KEY_NAME_PREFIX_USER_HAS30DAY_RETURN}{mid}"
  104. redis_helper = RedisHelper()
  105. data = redis_helper.get_data_from_redis(key_name=user_key)
  106. if data is not None:
  107. return int(data)
  108. else:
  109. request_url = f"{config_.GET_USER_30DayReturnCnt_URL}{mid}"
  110. result = request_get(request_url=request_url, timeout=0.1)
  111. if result is None:
  112. return None
  113. if result['code'] != 0:
  114. return None
  115. data = result['data']
  116. if data is True:
  117. redis_data = 1
  118. else:
  119. redis_data = 0
  120. redis_helper.set_data_to_redis(key_name=user_key, value=redis_data, expire_time=2 * 3600)
  121. return redis_data
  122. def get_videos_remain_view_count(app_type, videos):
  123. """
  124. 获取视频在流量池中的剩余可分发数
  125. :param app_type: 产品标识 type-int
  126. :param videos: 视频信息 (视频id, 流量池标记) type-list,[{'videoId': video_id, 'flowPool': flow_pool}, ...]
  127. :return: data type-list,[(video_id, flow_pool, view_count), ...]
  128. error_flag 错误标记,True为错误
  129. """
  130. error_flag = False
  131. if not videos:
  132. return [], error_flag
  133. request_data = {'appType': app_type, 'videos': videos}
  134. result = request_post(request_url=config_.GET_REMAIN_VIEW_COUNT_URL, request_data=request_data, timeout=(0.1, 1))
  135. if result is None:
  136. error_flag = True
  137. return [], error_flag
  138. if result['code'] != 0:
  139. log_.info('获取视频在流量池中的剩余可分发数失败')
  140. error_flag = True
  141. return [], error_flag
  142. data = [(item['videoId'], item['flowPool'], item['distributeCount']) for item in result['data']]
  143. return data, error_flag
  144. def get_videos_local_distribute_count(video_id, flow_pool):
  145. """
  146. 获取流量池视频本地分发数
  147. :param video_id: video_id
  148. :param flow_pool: 流量池标记
  149. :return: current_count 本地记录的分发数
  150. """
  151. # redis_h = datetime.now().hour
  152. # if datetime.now().minute >= 30:
  153. # redis_h += 0.5
  154. # key_name = config_.LOCAL_DISTRIBUTE_COUNT_PREFIX + str(redis_h)
  155. key_name = f'{config_.LOCAL_DISTRIBUTE_COUNT_PREFIX}{video_id}:{flow_pool}'
  156. redis_helper = RedisHelper()
  157. # video = '{}-{}'.format(video_id, flow_pool)
  158. # current_count = redis_helper.get_score_with_value(key_name=key_name, value=video)
  159. current_count = redis_helper.get_data_from_redis(key_name=key_name)
  160. if current_count is not None:
  161. return int(current_count)
  162. else:
  163. return None
  164. def update_video_w_h_rate(video_id, key_name):
  165. """
  166. 获取横屏视频的宽高比,并存入redis中 (width/height>1)
  167. :param video_id: videoId type-int
  168. :param key_name: redis key
  169. :return: None
  170. """
  171. # 获取数据
  172. sql = "SELECT id, width, height, rotate FROM longvideo.wx_video WHERE id = {};".format(video_id)
  173. mysql_helper = MysqlHelper()
  174. data = mysql_helper.get_data(sql=sql)
  175. if len(data) == 0:
  176. return
  177. # 更新到redis
  178. width, height, rotate = int(data[0][1]), int(data[0][2]), int(data[0][3])
  179. if width == 0 or height == 0:
  180. return
  181. if rotate in (90, 270):
  182. w_h_rate = height / width
  183. else:
  184. w_h_rate = width / height
  185. if w_h_rate > 1:
  186. info_data = {int(video_id): w_h_rate}
  187. else:
  188. return
  189. redis_helper = RedisHelper()
  190. # 写入新数据
  191. if len(info_data) > 0:
  192. redis_helper.add_data_with_zset(key_name=key_name, data=info_data)
  193. class FilterVideos(object):
  194. """视频过滤"""
  195. def __init__(self, request_id, app_type, video_ids, mid='', uid=''):
  196. """
  197. 初始化
  198. :param request_id: request_id
  199. :param app_type: 产品标识 type-int
  200. :param video_ids: 需过滤的视频列表 type-list
  201. :param mid: mid type-string
  202. :param uid: uid type-string
  203. """
  204. self.request_id = request_id
  205. self.app_type = app_type
  206. self.mid = mid
  207. self.uid = uid
  208. self.video_ids = video_ids
  209. def filter_video_status_h(self, video_ids, rule_key, data_key, ab_code, province_code, key_flag=''):
  210. """召回小时级更新的视频状态过滤"""
  211. # 根据Redis缓存中的数据过滤
  212. redis_helper = RedisHelper()
  213. # 获取不符合推荐状态的视频
  214. if ab_code in [code for _, code in config_.AB_CODE['region_rank_by_h'].items()]:
  215. if key_flag == 'region_24h':
  216. key_prefix = f"{config_.REGION_H_VIDEO_FILER_24H}{province_code}."
  217. elif key_flag == 'day_24h':
  218. key_prefix = f"{config_.H_VIDEO_FILER_24H}{province_code}."
  219. else:
  220. key_prefix = f"{config_.REGION_H_VIDEO_FILER}{province_code}."
  221. elif ab_code in [code for _, code in config_.AB_CODE['rank_by_24h'].items()]:
  222. key_prefix = config_.H_VIDEO_FILER_24H
  223. elif key_flag == '24h':
  224. key_prefix = config_.H_VIDEO_FILER_24H
  225. else:
  226. key_prefix = config_.H_VIDEO_FILER
  227. filter_videos_list = redis_helper.get_data_from_set(
  228. key_name=f"{key_prefix}{self.app_type}.{data_key}.{rule_key}"
  229. )
  230. if not filter_videos_list:
  231. return video_ids
  232. filter_videos = [int(video) for video in filter_videos_list]
  233. filtered_videos = [video_id for video_id in video_ids if video_id not in filter_videos]
  234. return filtered_videos
  235. def filter_videos_h(self, rule_key, data_key, ab_code, province_code, key_flag='', pool_type='rov'):
  236. """召回小时级更新的视频过滤"""
  237. # 预曝光过滤
  238. # st_pre = time.time()
  239. filtered_pre_result = self.filter_video_previewed(self.video_ids)
  240. # et_pre = time.time()
  241. # log_.info({
  242. # 'logTimestamp': int(time.time() * 1000),
  243. # 'request_id': self.request_id,
  244. # 'app_type': self.app_type,
  245. # 'mid': self.mid,
  246. # 'uid': self.uid,
  247. # 'operation': 'preview_filter',
  248. # 'request_videos': self.video_ids,
  249. # 'preview_filter_result': filtered_pre_result,
  250. # 'executeTime': (time.time() - st_pre) * 1000
  251. # })
  252. if not filtered_pre_result:
  253. return None
  254. # 视频状态过滤
  255. # st_status = time.time()
  256. filtered_status_result = self.filter_video_status_h(video_ids=filtered_pre_result, rule_key=rule_key,
  257. data_key=data_key, ab_code=ab_code,
  258. province_code=province_code, key_flag=key_flag)
  259. # et_status = time.time()
  260. # log_.info({
  261. # 'logTimestamp': int(time.time() * 1000),
  262. # 'request_id': self.request_id,
  263. # 'app_type': self.app_type,
  264. # 'mid': self.mid,
  265. # 'uid': self.uid,
  266. # 'operation': 'status_filter',
  267. # 'request_videos': filtered_pre_result,
  268. # 'status_filter_result': filtered_status_result,
  269. # 'executeTime': (time.time() - st_status) * 1000
  270. # })
  271. if not filtered_status_result:
  272. return None
  273. # 视频已曝光过滤
  274. st_viewed = time.time()
  275. filtered_viewed_result = self.filter_video_viewed(video_ids=filtered_status_result)
  276. # et_viewed = time.time()
  277. log_.info({
  278. 'logTimestamp': int(time.time() * 1000),
  279. 'pool_type': pool_type,
  280. 'request_id': self.request_id,
  281. 'app_type': self.app_type,
  282. 'mid': self.mid,
  283. 'uid': self.uid,
  284. 'operation': 'view_filter',
  285. 'request_videos': filtered_status_result,
  286. 'view_filter_result': filtered_viewed_result,
  287. 'executeTime': (time.time() - st_viewed) * 1000
  288. })
  289. if not filtered_viewed_result:
  290. return None
  291. else:
  292. return [int(video_id) for video_id in filtered_viewed_result]
  293. def filter_videos(self, pool_type='rov', region_code=None, shield_config=None):
  294. """视频过滤"""
  295. # todo: 添加app和region的风险过滤。
  296. st_viewed = time.time()
  297. videos_filtered = self.filter_videos_with_risk_video(self.video_ids, self.app_type, region_code)
  298. # log_.info({
  299. # 'logTimestamp': int(time.time() * 1000),
  300. # 'pool_type': "zhangbo-filter-pool_type",
  301. # 'request_id': self.request_id,
  302. # 'app_type': self.app_type,
  303. # 'mid': "zhangbo-filter_videos",
  304. # 'uid': self.uid,
  305. # 'operation': 'shield_filter',
  306. # 'request_videos': self.video_ids,
  307. # 'shield_filter_result': videos_filtered,
  308. # 'executeTime': (time.time() - st_viewed) * 1000
  309. # })
  310. # 预曝光过滤
  311. st_pre = time.time()
  312. filtered_pre_result = self.filter_video_previewed(videos_filtered)
  313. # print("filtered_pre:", (time.time()-st_pre)*1000)
  314. # et_pre = time.time()
  315. # log_.info({
  316. # 'logTimestamp': int(time.time() * 1000),
  317. # 'request_id': self.request_id,
  318. # 'app_type': self.app_type,
  319. # 'mid': self.mid,
  320. # 'uid': self.uid,
  321. # 'operation': 'preview_filter',
  322. # 'request_videos': self.video_ids,
  323. # 'preview_filter_result': filtered_pre_result,
  324. # 'executeTime': (time.time() - st_pre) * 1000
  325. # })
  326. if not filtered_pre_result:
  327. return None
  328. # 视频状态过滤采用离线定时过滤方案
  329. # 视频状态过滤
  330. # st_status = time.time()
  331. # filtered_status_result = self.filter_video_status(video_ids=filtered_pre_result)
  332. # et_status = time.time()
  333. # log_.info('filter by video status: result = {}, execute time = {}ms'.format(
  334. # filtered_status_result, (et_status - st_status) * 1000))
  335. # if not filtered_status_result:
  336. # return None
  337. # 视频已曝光过滤
  338. st_viewed = time.time()
  339. filtered_viewed_result = self.filter_video_viewed(video_ids=filtered_pre_result)
  340. # print("filtered_pre:", (time.time() - st_viewed) * 1000)
  341. # et_viewed = time.time()
  342. # log_.info({
  343. # 'logTimestamp': int(time.time() * 1000),
  344. # 'pool_type': pool_type,
  345. # 'request_id': self.request_id,
  346. # 'app_type': self.app_type,
  347. # 'mid': self.mid,
  348. # 'uid': self.uid,
  349. # 'operation': 'view_filter',
  350. # 'request_videos': filtered_pre_result,
  351. # 'view_filter_result': filtered_viewed_result,
  352. # 'executeTime': (time.time() - st_viewed) * 1000
  353. # })
  354. if not filtered_viewed_result:
  355. return None
  356. filtered_viewed_videos = [int(video_id) for video_id in filtered_viewed_result]
  357. return filtered_viewed_videos
  358. # if pool_type == 'flow' or pool_type=='normal':
  359. # # 流量池视频需过滤屏蔽视频
  360. # if region_code is None or shield_config is None:
  361. # return filtered_viewed_videos
  362. # else:
  363. # shield_key_name_list = shield_config.get(region_code, None)
  364. # if shield_key_name_list is not None:
  365. # filtered_shield_video_ids = self.filter_shield_video(
  366. # video_ids=filtered_viewed_videos, shield_key_name_list=shield_key_name_list
  367. # )
  368. # log_.info({
  369. # 'logTimestamp': int(time.time() * 1000),
  370. # 'pool_type': pool_type,
  371. # 'request_id': self.request_id,
  372. # 'app_type': self.app_type,
  373. # 'mid': self.mid,
  374. # 'uid': self.uid,
  375. # 'operation': 'shield_filter',
  376. # 'request_videos': filtered_viewed_videos,
  377. # 'shield_filter_result': filtered_shield_video_ids,
  378. # 'executeTime': (time.time() - st_viewed) * 1000
  379. # })
  380. # # print("filtered_pre flow:", (time.time() - st_viewed) * 1000)
  381. # return filtered_shield_video_ids
  382. # else:
  383. # return filtered_viewed_videos
  384. # else:
  385. # return filtered_viewed_videos
  386. def filter_video_previewed(self, video_ids):
  387. """
  388. 预曝光过滤
  389. :param video_ids: 需过滤的视频列表 type-list
  390. :return: filtered_videos 过滤后的列表 type-list
  391. """
  392. pre_time = time.time()
  393. if not self.mid or self.mid == 'null':
  394. # mid为空时,不做预曝光过滤
  395. return video_ids
  396. # 根据Redis缓存中的数据过滤
  397. redis_helper = RedisHelper()
  398. # key拼接
  399. key_name = f"{config_.PREVIEW_KEY_PREFIX}{self.app_type}:{self.mid}"
  400. #print("key_name:", key_name)
  401. pe_videos_list = redis_helper.get_data_from_set(key_name)
  402. #print("pe_videos_list:", pe_videos_list)
  403. # log_.info('****app_type = {}, mid = {}, uid = {}, pe_videos_list = {}'.format(
  404. # self.app_type, self.mid, self.uid, pe_videos_list))
  405. # log_.info('****app_type = {}, mid = {}, uid = {}, video_ids = {}'.format(
  406. # self.app_type, self.mid, self.uid, video_ids))
  407. if not pe_videos_list:
  408. return video_ids
  409. pe_videos = [int(video) for video in pe_videos_list]
  410. #print("pe_videos:", len(pe_videos))
  411. filtered_videos = [video_id for video_id in video_ids if video_id not in pe_videos]
  412. #print(f"pre res: {filtered_videos}\nexecute_time: {(time.time() - pre_time) * 1000}")
  413. return filtered_videos
  414. # def filter_video_status(self, video_ids):
  415. # """
  416. # 对视频状态进行过滤
  417. # :param video_ids: 视频id列表 type-list
  418. # :return: filtered_videos
  419. # """
  420. # if len(video_ids) == 1:
  421. # sql = "set hg_experimental_enable_shard_pruning=off; " \
  422. # "SELECT video_id " \
  423. # "FROM {} " \
  424. # "WHERE audit_status = 5 " \
  425. # "AND applet_rec_status IN (1, -6) " \
  426. # "AND open_status = 1 " \
  427. # "AND payment_status = 0 " \
  428. # "AND encryption_status != 5 " \
  429. # "AND transcoding_status = 3 " \
  430. # "AND video_id IN ({});".format(config_.VIDEO_STATUS, video_ids[0])
  431. # else:
  432. # sql = "set hg_experimental_enable_shard_pruning=off; " \
  433. # "SELECT video_id " \
  434. # "FROM {} " \
  435. # "WHERE audit_status = 5 " \
  436. # "AND applet_rec_status IN (1, -6) " \
  437. # "AND open_status = 1 " \
  438. # "AND payment_status = 0 " \
  439. # "AND encryption_status != 5 " \
  440. # "AND transcoding_status = 3 " \
  441. # "AND video_id IN {};".format(config_.VIDEO_STATUS, tuple(video_ids))
  442. #
  443. # hologres_helper = HologresHelper()
  444. # data = hologres_helper.get_data(sql=sql)
  445. # filtered_videos = [int(temp[0]) for temp in data]
  446. # return filtered_videos
  447. def filter_video_viewed(self, video_ids, types=(1, 6,)):
  448. """
  449. 调用后端接口过滤用户已观看视频
  450. :param video_ids: 视频id列表 type-list
  451. :param types: 过滤参数 type-tuple, 默认(1, )
  452. 1-已观看 2-视频状态 3-是否进入老年人社区 4-话题状态 5-推荐状态 6-白名单过滤 7-涉政视频过滤
  453. :return: filtered_videos
  454. """
  455. # 获取对应端的过滤参数types
  456. types = config_.FILTER_VIEWED_TYPES_CONFIG.get(self.app_type, None)
  457. if types is None:
  458. types = config_.FILTER_VIEWED_TYPES_CONFIG.get('other')
  459. request_data = {"appType": self.app_type,
  460. "mid": self.mid,
  461. "uid": self.uid,
  462. "types": list(types),
  463. "videoIds": video_ids}
  464. # print(request_data)
  465. # 调用http接口
  466. result = request_post(request_url=config_.VIDEO_FILTER_URL, request_data=request_data, timeout=(0.1, 1))
  467. # print("result:", result)
  468. if result is None:
  469. # print("result is None")
  470. # log_.info('过滤失败,types: {}'.format(types))
  471. return []
  472. if result['code'] != 0:
  473. # log_.info('过滤失败,types: {}'.format(types))
  474. return []
  475. filtered_videos = result['data']
  476. return filtered_videos
  477. def filter_video_viewed_new(self, video_ids):
  478. """
  479. 调用后端接口过滤用户已观看视频
  480. :param video_ids: 视频id列表 type-list
  481. :param types: 过滤参数 type-tuple, 默认(1, )
  482. 1-已观看 2-视频状态 3-是否进入老年人社区 4-话题状态 5-推荐状态 6-白名单过滤 7-涉政视频过滤
  483. :return: filtered_videos
  484. """
  485. # 获取对应端的过滤参数types
  486. st_time = time.time()
  487. types = config_.FILTER_VIEWED_TYPES_CONFIG.get(self.app_type, None)
  488. #print(types)
  489. if types is None:
  490. types = config_.FILTER_VIEWED_TYPES_CONFIG.get('other')
  491. if 6 in types:
  492. types = list(types)
  493. types.remove(6)
  494. #print(types)
  495. request_data = {"appType": self.app_type,
  496. "mid": self.mid,
  497. "uid": self.uid,
  498. "types": list(types),
  499. "videoIds": video_ids}
  500. # 调用http接口
  501. result = request_post(request_url=config_.VIDEO_FILTER_URL, request_data=request_data, timeout=(0.1, 1))
  502. #print(f"view res: {result}\nexecute_time: {(time.time() - st_time) * 1000}")
  503. if result is None:
  504. # log_.info('过滤失败,types: {}'.format(types))
  505. return []
  506. if result['code'] != 0:
  507. # log_.info('过滤失败,types: {}'.format(types))
  508. return []
  509. filtered_videos = result['data']
  510. return filtered_videos
  511. def filter_shield_video(self, video_ids, shield_key_name_list):
  512. """
  513. 过滤屏蔽视频视频
  514. :param video_ids: 需过滤的视频列表 type-list
  515. :param shield_key_name_list: 过滤视频 redis-key
  516. :return: filtered_videos 过滤后的列表 type-list
  517. """
  518. # print("filter_shield_video:", len(filter_shield_video))
  519. if len(video_ids) == 0:
  520. return video_ids
  521. # 根据Redis缓存中的数据过滤
  522. redis_helper = RedisHelper()
  523. for shield_key_name in shield_key_name_list:
  524. video_ids = [
  525. int(video_id) for video_id in video_ids
  526. if not redis_helper.data_exists_with_set(key_name=shield_key_name, value=video_id)
  527. ]
  528. # shield_videos_list = redis_helper.get_data_from_set(key_name=shield_key_name)
  529. # if not shield_videos_list:
  530. # continue
  531. # shield_videos = [int(video) for video in shield_videos_list]
  532. # video_ids = [int(video_id) for video_id in video_ids if int(video_id) not in shield_videos]
  533. # print("video_ids:", len(video_ids))
  534. return video_ids
  535. def new_filter_video(self):
  536. """视频过滤"""
  537. # 1. 预曝光过滤
  538. st_pre = time.time()
  539. #print("new_filter video_ids:", self.video_ids)
  540. filtered_pre_result = self.filter_video_previewed(self.video_ids)
  541. if not filtered_pre_result:
  542. return None
  543. # log_.info({
  544. # 'logTimestamp': int(time.time() * 1000),
  545. # 'request_id': self.request_id,
  546. # 'app_type': self.app_type,
  547. # 'mid': self.mid,
  548. # 'uid': self.uid,
  549. # 'operation': 'preview_filter',
  550. # 'request_videos': self.video_ids,
  551. # 'preview_filter_result': filtered_pre_result,
  552. # 'executeTime': (time.time() - st_pre) * 1000
  553. # })
  554. #2. 视频已曝光过滤
  555. st_viewed = time.time()
  556. #print("---filtered viewed---")
  557. #print("filtered_pre_result:",filtered_pre_result)
  558. filtered_viewed_result = self.filter_video_viewed(video_ids=filtered_pre_result)
  559. if not filtered_viewed_result:
  560. return None
  561. return filtered_viewed_result
  562. def new_flow_video(self, vid_list, flow_vids_set, region_code, shield_config):
  563. flow_video_list = []
  564. normal_video_list = []
  565. for v_id in vid_list:
  566. if v_id in flow_vids_set:
  567. flow_video_list.append(v_id)
  568. else:
  569. normal_video_list.append(v_id)
  570. shield_key_name_list = shield_config.get(region_code, None)
  571. if shield_key_name_list is not None:
  572. filtered_shield_video_ids = self.filter_shield_video(
  573. video_ids=flow_video_list, shield_key_name_list=shield_key_name_list
  574. )
  575. return normal_video_list, filtered_shield_video_ids
  576. else:
  577. return normal_video_list, flow_video_list
  578. def filter_movie_religion_video(self, video_ids):
  579. """过滤白名单视频(影视,宗教)"""
  580. # 影视 + 宗教: rov.filter.movie.{videoId}
  581. # 宗教: rov.filter.religion.{videoId}
  582. st_time = time.time()
  583. if self.app_type not in [config_.APP_TYPE['WAN_NENG_VIDEO'],
  584. config_.APP_TYPE['LAO_HAO_KAN_VIDEO'],
  585. config_.APP_TYPE['ZUI_JING_QI'],
  586. config_.APP_TYPE['H5']]:
  587. # 过滤 影视 + 宗教
  588. keys = [f"rov.filter.movie.{video_id}" for video_id in video_ids]
  589. elif self.app_type in [config_.APP_TYPE['WAN_NENG_VIDEO'],
  590. config_.APP_TYPE['ZUI_JING_QI'],
  591. config_.APP_TYPE['H5']]:
  592. # 过滤 影视 + 宗教
  593. keys = [f"rov.filter.religion.{video_id}" for video_id in video_ids]
  594. else:
  595. #print(f"m_r res: {video_ids}\nexecute_time: {(time.time() - st_time) * 1000}")
  596. return video_ids
  597. redis_helper = RedisHelper(redis_info=config_.REDIS_INFO_FILTER)
  598. filter_videos = []
  599. for i in range(len(keys)//1000 + 1):
  600. video_ids_temp = video_ids[i*1000:(i+1)*1000]
  601. if len(video_ids_temp) == 0:
  602. break
  603. mget_res = redis_helper.mget(keys=keys[i*1000:(i+1)*1000])
  604. filter_videos.extend([int(data) for data in mget_res if data is not None])
  605. if len(filter_videos) > 0:
  606. filtered_videos = set(video_ids) - set(filter_videos)
  607. #print(f"m_r res: {list(filtered_videos)}\nexecute_time: {(time.time() - st_time) * 1000}")
  608. return list(filtered_videos)
  609. else:
  610. #print(f"m_r res: {video_ids}\nexecute_time: {(time.time() - st_time) * 1000}")
  611. return video_ids
  612. def filter_videos_new(self, region_code=None, shield_config=None, flow_set=None):
  613. """视频过滤"""
  614. # 预曝光过滤
  615. st_pre = time.time()
  616. #print("self.video_ids:", len(self.video_ids))
  617. filtered_pre_result = self.filter_video_previewed(self.video_ids)
  618. if not filtered_pre_result:
  619. return None
  620. #print("filtered_pre_result:", len(filtered_pre_result))
  621. #print(filtered_pre_result)
  622. # 视频已曝光过滤/白名单过滤
  623. st_viewed = time.time()
  624. t = [
  625. gevent.spawn(self.filter_video_viewed_new, filtered_pre_result),
  626. gevent.spawn(self.filter_movie_religion_video, filtered_pre_result)]
  627. gevent.joinall(t)
  628. filtered_result_list = [i.get() for i in t]
  629. #print("filtered_result_list1:",filtered_result_list[0])
  630. #print("filtered_result_list2:",filtered_result_list[1])
  631. filtered_viewed_set = set('')
  632. for i in filtered_result_list[0]:
  633. filtered_viewed_set.add(int(i))
  634. filter_video_set =set('')
  635. for j in filtered_result_list[1]:
  636. filter_video_set.add(int(j))
  637. filtered_viewed_result = list(filtered_viewed_set & filter_video_set)
  638. #print(f"view&m_r res: {filtered_viewed_result}\nexecute_time: {(time.time() - st_viewed) * 1000}")
  639. #print("filtered:",len(filtered_viewed_result))
  640. if not filtered_viewed_result:
  641. return None
  642. filtered_viewed_videos = [int(video_id) for video_id in filtered_viewed_result]
  643. #print("result:", filtered_viewed_videos)
  644. if flow_set is None:
  645. return filtered_viewed_videos
  646. else:
  647. # 流量池视频需过滤屏蔽视频
  648. if region_code is None or shield_config is None:
  649. return filtered_viewed_videos
  650. else:
  651. normal_recall_ids = []
  652. left_flow_ids = []
  653. for vid in filtered_viewed_videos:
  654. if vid in flow_set:
  655. left_flow_ids.append(vid)
  656. else:
  657. normal_recall_ids.append(vid)
  658. shield_key_name_list = shield_config.get(region_code, None)
  659. if shield_key_name_list is not None:
  660. filtered_shield_video_ids = self.filter_shield_video(
  661. video_ids=left_flow_ids, shield_key_name_list=shield_key_name_list
  662. )
  663. return normal_recall_ids+filtered_shield_video_ids
  664. else:
  665. return filtered_viewed_videos
  666. def filter_videos_status(self, pool_type='rov', region_code=None, shield_config=None):
  667. """视频过滤"""
  668. # todo: 添加app和region的风险过滤。
  669. st_viewed = time.time()
  670. videos_filtered = self.filter_videos_with_risk_video(self.video_ids, self.app_type, region_code)
  671. # log_.info({
  672. # 'logTimestamp': int(time.time() * 1000),
  673. # 'pool_type': "zhangbo-filter-pool_type",
  674. # 'request_id': self.request_id,
  675. # 'app_type': self.app_type,
  676. # 'mid': "zhangbo-filter_videos_status",
  677. # 'uid': self.uid,
  678. # 'operation': 'shield_filter',
  679. # 'request_videos': self.video_ids,
  680. # 'shield_filter_result': videos_filtered,
  681. # 'executeTime': (time.time() - st_viewed) * 1000
  682. # })
  683. # 预曝光过滤
  684. st_pre = time.time()
  685. filtered_pre_result = self.filter_video_previewed(videos_filtered)
  686. # print("filtered_pre:", (time.time()-st_pre)*1000)
  687. # et_pre = time.time()
  688. # log_.info({
  689. # 'logTimestamp': int(time.time() * 1000),
  690. # 'request_id': self.request_id,
  691. # 'app_type': self.app_type,
  692. # 'mid': self.mid,
  693. # 'uid': self.uid,
  694. # 'operation': 'preview_filter',
  695. # 'request_videos': self.video_ids,
  696. # 'preview_filter_result': filtered_pre_result,
  697. # 'executeTime': (time.time() - st_pre) * 1000
  698. # })
  699. if not filtered_pre_result:
  700. return None
  701. # 视频状态过滤采用离线定时过滤方案
  702. # 视频状态过滤
  703. # st_status = time.time()
  704. # filtered_status_result = self.filter_video_status(video_ids=filtered_pre_result)
  705. # et_status = time.time()
  706. # log_.info('filter by video status: result = {}, execute time = {}ms'.format(
  707. # filtered_status_result, (et_status - st_status) * 1000))
  708. # if not filtered_status_result:
  709. # return None
  710. # 视频已曝光过滤
  711. st_viewed = time.time()
  712. filtered_viewed_result = self.filter_video_viewed_status(video_ids=filtered_pre_result)
  713. # print("filtered_pre:", (time.time() - st_viewed) * 1000)
  714. # et_viewed = time.time()
  715. # log_.info({
  716. # 'logTimestamp': int(time.time() * 1000),
  717. # 'pool_type': pool_type,
  718. # 'request_id': self.request_id,
  719. # 'app_type': self.app_type,
  720. # 'mid': self.mid,
  721. # 'uid': self.uid,
  722. # 'operation': 'view_filter',
  723. # 'request_videos': filtered_pre_result,
  724. # 'view_filter_result': filtered_viewed_result,
  725. # 'executeTime': (time.time() - st_viewed) * 1000
  726. # })
  727. if not filtered_viewed_result:
  728. return None
  729. filtered_viewed_videos = [int(video_id) for video_id in filtered_viewed_result]
  730. return filtered_viewed_videos
  731. # if pool_type == 'flow' or pool_type=='normal':
  732. # # 流量池视频需过滤屏蔽视频
  733. # if region_code is None or shield_config is None:
  734. # return filtered_viewed_videos
  735. # else:
  736. # shield_key_name_list = shield_config.get(region_code, None)
  737. # if shield_key_name_list is not None:
  738. # filtered_shield_video_ids = self.filter_shield_video(
  739. # video_ids=filtered_viewed_videos, shield_key_name_list=shield_key_name_list
  740. # )
  741. # log_.info({
  742. # 'logTimestamp': int(time.time() * 1000),
  743. # 'pool_type': pool_type,
  744. # 'request_id': self.request_id,
  745. # 'app_type': self.app_type,
  746. # 'mid': self.mid,
  747. # 'uid': self.uid,
  748. # 'operation': 'shield_filter',
  749. # 'request_videos': filtered_viewed_videos,
  750. # 'shield_filter_result': filtered_shield_video_ids,
  751. # 'executeTime': (time.time() - st_viewed) * 1000
  752. # })
  753. # # print("filtered_pre flow:", (time.time() - st_viewed) * 1000)
  754. # return filtered_shield_video_ids
  755. # else:
  756. # return filtered_viewed_videos
  757. # else:
  758. # return filtered_viewed_videos
  759. def filter_video_viewed_status(self, video_ids, types=(1, 6,)):
  760. """
  761. 调用后端接口过滤用户已观看视频
  762. :param video_ids: 视频id列表 type-list
  763. :param types: 过滤参数 type-tuple, 默认(1, )
  764. 1-已观看 2-视频状态 3-是否进入老年人社区 4-话题状态 5-推荐状态 6-白名单过滤 7-涉政视频过滤
  765. :return: filtered_videos
  766. """
  767. # 获取对应端的过滤参数types
  768. types = config_.FILTER_VIEWED_TYPES_CONFIG.get(self.app_type, None)
  769. if types is None:
  770. types = config_.FILTER_VIEWED_TYPES_CONFIG.get('other')
  771. types = list(types)
  772. types.append(2)
  773. request_data = {"appType": self.app_type,
  774. "mid": self.mid,
  775. "uid": self.uid,
  776. "types": types,
  777. "videoIds": video_ids}
  778. # print(request_data)
  779. # 调用http接口
  780. result = request_post(request_url=config_.VIDEO_FILTER_URL, request_data=request_data, timeout=(0.1, 1))
  781. # print("result:", result)
  782. if result is None:
  783. # print("result is None")
  784. # log_.info('过滤失败,types: {}'.format(types))
  785. return []
  786. if result['code'] != 0:
  787. # log_.info('过滤失败,types: {}'.format(types))
  788. return []
  789. filtered_videos = result['data']
  790. return filtered_videos
  791. def filter_videos_with_risk_video(self, video_ids, app_type, region_code):
  792. # 0 用一个开关控制,是否过滤生效。 便于回滚功能。
  793. flag = param_update_risk_filter_flag()
  794. if not flag:
  795. return video_ids[0: min(20, len(video_ids))]
  796. # 1 判断是否过滤,不展示的app+区域列表,-1必须过滤
  797. app_region_filtered = param_update_risk_rule()
  798. if app_type in app_region_filtered.keys():
  799. if_filtered = False
  800. if region_code in app_region_filtered[app_type]:
  801. if_filtered = True
  802. else:
  803. if_filtered = True
  804. if region_code == -1:
  805. if_filtered = True
  806. if not if_filtered:
  807. return video_ids[0: min(20, len(video_ids))]
  808. # 2 确认过滤,获取风险video列表param_update_risk_videos
  809. videos_with_risk = param_update_risk_videos()
  810. # 3 过滤 返回结果
  811. video_ids_new = [i for i in video_ids if i not in videos_with_risk]
  812. # print(flag)
  813. # print(app_region_filtered)
  814. # print(video_ids)
  815. # print(app_type)
  816. # print(region_code)
  817. # print(videos_with_risk)
  818. # print(video_ids_new)
  819. # print(len(video_ids))
  820. # print(len(video_ids_new))
  821. return video_ids_new[0: min(20, len(video_ids_new))]
  822. if __name__ == '__main__':
  823. user = [
  824. ('weixin_openid_o0w175fDc8pNnywrYN49E341tKfI', ''),
  825. ('weixin_openid_o0w175YwC3hStzcR5DAQdbgzdMeI', ''),
  826. ('weixin_openid_o0w175ftZDl6VJVDx9la3WVPh7mU', '15900461'),
  827. ('weixin_openid_o0w175SPqpCVRcp7x1XvnX4qpIvI', '19659040'),
  828. ('weixin_openid_o0w175cOnguapyWIrDrHkOWl4oFQ', '31210128'),
  829. ('weixin_openid_o0w175UXYId-o71e1Q3SOheYNteQ', '33099722'),
  830. ('weixin_openid_o0w175QQ5b42AtOe50bchrFgcttA', ''),
  831. ('weixin_openid_o0w175bgaPlfLsp3YLDKWqLWtXX8', '35371534'),
  832. ('weixin_openid_o0w175eRpvbmV6nOhM1VTyyLICWA', '30488803'),
  833. ('weixin_openid_o0w175TZYvG47pQkOjyJFoxQuqsw', '')
  834. ]
  835. video_df = pd.read_csv('./data/videoids.csv')
  836. videoid_list = video_df['videoid'].tolist()
  837. for mid, uid in user:
  838. video_ids = random.sample(videoid_list, 1000)
  839. start_time = time.time()
  840. filter_ = FilterVideos(request_id=f'{mid} - {uid}', app_type=0, mid=mid, uid=uid, video_ids=video_ids)
  841. res = filter_.filter_videos_new()
  842. print(f"res: {res}\nexecute_time: {(time.time() - start_time) * 1000}")
  843. # filter_.filter_video_status(video_ids=[1, 3, 5])
  844. # videos = [{'videoId': 9034659, 'flowPool': '3#11#3#1637824188547'}, {'videoId': 9035052, 'flowPool': '3#11#3#1637824172827'}]
  845. # res = get_videos_remain_view_count(4, videos)
  846. # print(res)
  847. # text = '测试 @李倩'
  848. # send_msg_to_feishu(text)
  849. # update_video_w_h_rate(video_id=113, key_name='')
  850. # mid = "weixin_openid_obHDW5c4g3aULfCWh-68LcUSxCB"
  851. # request_url = f"{config_.GET_USER_30DayReturnCnt_URL}{mid}"
  852. # res = request_get(request_url=request_url, timeout=100)
  853. # res = get_user_has30day_return(mid=mid)
  854. # print(res, type(res))