liqian преди 3 години
родител
ревизия
f1bba95ca0
променени са 1 файла, в които са добавени 21 реда и са изтрити 9 реда
  1. 21 9
      utils.py

+ 21 - 9
utils.py

@@ -98,15 +98,27 @@ def filter_video_status(video_ids):
     :param video_ids: 视频id列表 type-list
     :return: filtered_videos
         """
-    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))
+    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]