同一个音频一天最多用2次

common/material.py

@@ -119,12 +119,13 @@ class Material():
     @classmethod
     def get_id(cls, audio_id):
         current_time = datetime.now()
-        previous_day = current_time - timedelta(days=1)
-        formatted_time = previous_day.strftime("%Y-%m-%d")
-        id_list = f"""select * FROM video_audio where time = '{formatted_time}' AND audio = '{audio_id}'"""
+        formatted_time = current_time.strftime("%Y-%m-%d")
+        id_list = f"""select account_id FROM video_audio where time = '{formatted_time}' AND audio = '{audio_id} ' GROUP BY account_id """
         id_list = MysqlHelper.get_values(id_list, "prod")
         return id_list
 
+
+
     # 获取音频类型
     @classmethod
     def get_audio_type(cls, video_type, count, channel_type):
@@ -188,10 +189,13 @@ class Material():
                         if audio_id:
                             list.append(number)
                         title_list.append(title)
-                    list = random.choice(list)
-                    audio_id = list['audio_id']
-                    srt = list['text']
-                    return audio_id, srt, title_list
+                    while True:
+                        list1 = random.choice(list)
+                        audio_id = list1['audio_id']
+                        srt = list1['text']
+                        id_list = cls.get_id(audio_id)
+                        if len(id_list) <= 2:
+                            return audio_id, srt, title_list
             else:
                 audio_type = [{"audio": "音画美文--美文类", "type": "6VXm7q"}, {"audio": "音画美文--通用类", "type": "aSNFl8"}]
                 audio_type = random.choice(audio_type)
@@ -210,7 +214,7 @@ class Material():
                     audio_id = list1['audio_id']
                     srt = list1['text']
                     id_list = cls.get_id(audio_id)
-                    if len(id_list) == 0:
+                    if len(id_list) <= 2:
                         return audio_id, srt, title_list