debug

pool_predict.py

@@ -28,7 +28,9 @@ def get_videos_from_flow_pool(app_type, size=1000):
         if result['code'] != 0:
             log_.info('batch_flag: {}, 获取流量池视频失败'.format(batch_flag))
             break
-        videos.append(result['data'])
+        if not result['data']:
+            break
+        videos.extend(result['data'])
     return videos
 
 
@@ -74,11 +76,16 @@ def predict(app_type):
         return None
     # video_id 与 flow_pool 进行mapping
     video_ids = set()
-    log_.info('流量池中视频数：{}'.format(len(video_ids)))
+    log_.info('流量池中视频数：{}'.format(len(videos)))
     mapping = {}
     for video in videos:
-        video_ids.add(video['videoId'])
-        mapping[video['videoId']] = video['flowPool']
+        video_id = video['videoId']
+        video_ids.add(video_id)
+        if video_id in mapping:
+            mapping[video_id].append(video['flowPool'])
+        else:
+            mapping[video_id] = [video['flowPool']]
+
     # 对视频状态进行过滤
     filtered_videos = filter_video_status(list(video_ids))
     log_.info('filter videos status finished, filtered_videos nums={}'.format(len(filtered_videos)))

utils.py

@@ -101,7 +101,7 @@ def filter_video_status(video_ids):
     sql = "SELECT video_id " \
           "FROM {} " \
           "WHERE audit_status = 5 " \
-          "AND applet_rec_status IN (1, 6) " \
+          "AND applet_rec_status IN (1, -6) " \
           "AND open_status = 1 " \
           "AND payment_status = 0 " \
           "AND encryption_status IS NULL " \