update sql

utils.py

@@ -98,35 +98,32 @@ def filter_video_status(video_ids):
     :param video_ids: 视频id列表 type-list
     :return: filtered_videos
     """
-    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])
+    if len(video_ids) == 1:
+        sql = "set hg_experimental_enable_shard_pruning=off; " \
+              "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 = "set hg_experimental_enable_shard_pruning=off; " \
+              "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]
     return filtered_videos