Browse Source

update predict

liqian 2 years ago
parent
commit
c62d9dca35
1 changed files with 5 additions and 2 deletions
  1. 5 2
      pool_predict.py

+ 5 - 2
pool_predict.py

@@ -6,7 +6,7 @@ import traceback
 import random
 
 from config import set_config
-from utils import request_post, filter_video_status, send_msg_to_feishu
+from utils import request_post, filter_video_status, send_msg_to_feishu, filter_video_status_app
 from log import Log
 from db_helper import RedisHelper
 from odps import ODPS
@@ -111,7 +111,10 @@ def predict(app_type):
                 mapping[video_id] = [video['flowPool']]
 
         # 对视频状态进行过滤
-        filtered_videos = filter_video_status(list(video_ids))
+        if app_type == config_.APP_TYPE['APP']:
+            filtered_videos = filter_video_status_app(list(video_ids))
+        else:
+            filtered_videos = filter_video_status(list(video_ids))
         log_.info('filter videos status finished, filtered_videos nums={}'.format(len(filtered_videos)))
         if not filtered_videos:
             log_.info('流量池中视频状态不符合分发')