zhangyong 1 anno fa
parent
commit
7ae7afd0e7
3 ha cambiato i file con 5 aggiunte e 4 eliminazioni
  1. 1 1
      common/material.py
  2. 2 1
      main.py
  3. 2 2
      video_stitching/video_stitching.py

+ 1 - 1
common/material.py

@@ -25,7 +25,7 @@ class Material():
                 account_name = row[2]
                 account_link = row[3]
                 user_id = account_link.split("user/")[1]
-                insert_sql = f"""INSERT INTO video_user_id (name, user_id, channel) values ("{account_name}", "{user_id}", "{channel}")"""
+                insert_sql = f"""INSERT INTO video_user_id (name, user_id, channel) values ('{account_name}', '{user_id}', '{channel}')"""
                 MysqlHelper.update_values(
                     sql=insert_sql,
                     env="prod",

+ 2 - 1
main.py

@@ -14,7 +14,8 @@ from video_stitching.video_stitching import VideoStitching
 def get_account_id():
     current_time = datetime.now()
     formatted_time = current_time.strftime("%Y-%m-%d")
-    count = f"""SELECT COUNT(*) AS total_count FROM ( SELECT audio, account_id FROM video_audio WHERE time = "{formatted_time}" GROUP BY audio, account_id) AS subquery;"""
+    count = f"""SELECT COUNT(*) AS total_count FROM ( SELECT audio, account_id FROM video_audio WHERE time = '{formatted_time}' GROUP BY audio, account_id) AS subquery;"""
+
     count = MysqlHelper.get_values(count, "prod")
     return count
 if __name__ == '__main__':

+ 2 - 2
video_stitching/video_stitching.py

@@ -52,7 +52,7 @@ class VideoStitching():
         current_time = datetime.now()
         formatted_time = current_time.strftime("%Y-%m-%d")
         for j in audio_url:
-            insert_sql = f"""INSERT INTO video_audio (audio, video_id, account_id, oss_object_key, time, tag) values ("{i}", "{j[0]}", "{j[1]}", "{j[2]}", "{formatted_time}")"""
+            insert_sql = f"""INSERT INTO video_audio (audio, video_id, account_id, oss_object_key, time, tag) values ('{i}', '{j[0]}', '{j[1]}', '{j[2]}', '{formatted_time}')"""
             MysqlHelper.update_values(
                 sql=insert_sql,
                 env="prod",
@@ -80,7 +80,7 @@ class VideoStitching():
         url_list = f"""SELECT a.video_id,a.account_id,a.oss_object_key FROM video_url a WHERE NOT EXISTS (
                 SELECT video_id
                 FROM video_audio b
-                WHERE a.video_id = b.video_id AND b.audio = "{audio_id}"
+                WHERE a.video_id = b.video_id AND b.audio = '{audio_id}'
             ) AND a.account_id = {account} ;"""
         url_list = MysqlHelper.get_values(url_list, "prod")
         return url_list