liqian 3 gadi atpakaļ
vecāks
revīzija
5cf7129fdf
2 mainītis faili ar 2 papildinājumiem un 2 dzēšanām
  1. 1 1
      relevant_top_videos.py
  2. 1 1
      videos_filter.py

+ 1 - 1
relevant_top_videos.py

@@ -22,7 +22,7 @@ def get_relevant_videos_with_excel():
         recommend_vid = int(df.iloc[i]['recommendVid'])
         # 状态过滤
         filtered_videos = filter_video_status(video_ids=[recommend_vid])
-        if not filtered_videos or len(filtered_videos) == 0:
+        if filtered_videos is None or len(filtered_videos) == 0:
             continue
         # 将时间转换为10位时间戳
         start_time = int(time.mktime(time.strptime(df.iloc[i]['startTime'].split(r'.')[0], '%Y-%m-%d %H:%M:%S')))

+ 1 - 1
videos_filter.py

@@ -18,7 +18,7 @@ def filter_relevant_videos():
     # 读取需要过滤的头部视频id
     redis_helper = RedisHelper()
     head_videos = redis_helper.get_data_from_set(key_name=config_.RELEVANT_TOP_VIDEOS_KEY_NAME)
-    if len(head_videos) == 0:
+    if head_videos is None or len(head_videos) == 0:
         return
 
     # 过滤