update filter_video_status

utils.py

@@ -97,31 +97,36 @@ def filter_video_status(video_ids):
     对视频状态进行过滤
     :param video_ids: 视频id列表 type-list
     :return: filtered_videos
-        """
-    if len(video_ids) == 1:
-        sql = "SELECT video_id " \
-              "FROM {} " \
-              "WHERE audit_status = 5 " \
-              "AND applet_rec_status IN (1, -6) " \
-              "AND open_status = 1 " \
-              "AND payment_status = 0 " \
-              "AND encryption_status IS NULL " \
-              "AND transcoding_status = 3 " \
-              "AND video_id IN ({});".format(config_.VIDEO_STATUS, video_ids[0])
-    else:
-        sql = "SELECT video_id " \
-              "FROM {} " \
-              "WHERE audit_status = 5 " \
-              "AND applet_rec_status IN (1, -6) " \
-              "AND open_status = 1 " \
-              "AND payment_status = 0 " \
-              "AND encryption_status IS NULL " \
-              "AND transcoding_status = 3 " \
-              "AND video_id IN {};".format(config_.VIDEO_STATUS, tuple(video_ids))
-
-    hologres_helper = HologresHelper()
-    data = hologres_helper.get_data(sql=sql)
-    filtered_videos = [temp[0] for temp in data]
+    """
+    filtered_videos = []
+    for i in range(len(video_ids) // 50 + 1):
+        start = i * 50
+        end = (i+1) * 50
+        videos = video_ids[start:end]
+        if len(videos) == 1:
+            sql = "SELECT video_id " \
+                  "FROM {} " \
+                  "WHERE audit_status = 5 " \
+                  "AND applet_rec_status IN (1, -6) " \
+                  "AND open_status = 1 " \
+                  "AND payment_status = 0 " \
+                  "AND encryption_status IS NULL " \
+                  "AND transcoding_status = 3 " \
+                  "AND video_id IN ({});".format(config_.VIDEO_STATUS, videos[0])
+        else:
+            sql = "SELECT video_id " \
+                  "FROM {} " \
+                  "WHERE audit_status = 5 " \
+                  "AND applet_rec_status IN (1, -6) " \
+                  "AND open_status = 1 " \
+                  "AND payment_status = 0 " \
+                  "AND encryption_status IS NULL " \
+                  "AND transcoding_status = 3 " \
+                  "AND video_id IN {};".format(config_.VIDEO_STATUS, tuple(videos))
+
+        hologres_helper = HologresHelper()
+        data = hologres_helper.get_data(sql=sql)
+        filtered_videos.extend([temp[0] for temp in data])
     return filtered_videos