Jelajahi Sumber

add filter in get_top_video_list

liqian 3 tahun lalu
induk
melakukan
063b59a2fd
1 mengubah file dengan 10 tambahan dan 3 penghapusan
  1. 10 3
      top_video_list.py

+ 10 - 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
@@ -18,9 +18,16 @@ def get_top_video_list():
             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('置顶视频更新失败!')