liqian 3 years ago
parent
commit
f1bba95ca0
1 changed files with 21 additions and 9 deletions
  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]