zhangyong 1 tahun lalu
induk
melakukan
21d265600a
1 mengubah file dengan 3 tambahan dan 1 penghapusan
  1. 3 1
      video_agc/agc_video_method.py

+ 3 - 1
video_agc/agc_video_method.py

@@ -26,6 +26,8 @@ class AgcVidoe():
     def get_url_list(cls, user_list, mark, limit_count):
         for i in range(5):
             user = random.choice(user_list)
+            if "-" in mark:
+                user = str(user).replace('(', '').replace(')', '').replace(',', '')
             current_time = datetime.now()
             three_days_ago = current_time - timedelta(days=3)
             formatted_current_time = current_time.strftime("%Y-%m-%d")
@@ -34,7 +36,7 @@ class AgcVidoe():
                                 SELECT video_id
                                 FROM agc_video_deposit b
                                 WHERE a.oss_object_key = b.oss_object_key AND b.time >= '{formatted_three_days_ago}' AND b.time <= '{formatted_current_time}'
-                            ) AND a.account_id = {user} and a.`status` = 1 and a.mark = '{mark}' limit {limit_count};"""
+                            ) AND a.account_id = '{user}' and a.`status` = 1 and a.mark = '{mark}' limit {limit_count};"""
             print(url_list)
             url_list = MysqlHelper.get_values(url_list, "prod")
             if url_list: