Bläddra i källkod

add filter video status

liqian 3 år sedan
förälder
incheckning
3e1b3534cf
1 ändrade filer med 12 tillägg och 3 borttagningar
  1. 12 3
      top_video_list.py

+ 12 - 3
top_video_list.py

@@ -1,5 +1,5 @@
 import traceback
-from utils import request_post
+from utils import request_post, filter_video_status
 from db_helper import RedisHelper
 from config import set_config
 from log import Log
@@ -17,11 +17,20 @@ def get_top_video_list():
         if result['code'] != 0:
             log_.info('获取置顶视频失败!')
             return
+
         data = result['data']
-        log_.info('data = {}'.format(data))
+        # 视频状态过滤
+        video_ids = [item['videoId'] for item in data]
+        log_.info('video_ids = {}'.format(video_ids))
+        filter_videos = filter_video_status(video_ids=video_ids)
+        log_.info('filter_videos = {}'.format(filter_videos))
+        value = [item for item in data if item['videoId'] in filter_videos]
+        log_.info('value = {}'.format(value))
+        # 写入redis
         redis_helper = RedisHelper()
-        redis_helper.set_data_to_redis(key_name=config_.TOP_VIDEO_LIST_KEY_NAME, value=str(data), expire_time=5*60)
+        redis_helper.set_data_to_redis(key_name=config_.TOP_VIDEO_LIST_KEY_NAME, value=str(value), expire_time=5 * 60)
         log_.info('置顶视频更新成功!')
+
     except Exception as e:
         log_.error('置顶视频更新失败!')
         log_.error(traceback.format_exc())