Parcourir la source

oss地址入库

zhangyong il y a 10 mois
Parent
commit
a9de64b506
2 fichiers modifiés avec 5 ajouts et 3 suppressions
  1. 2 2
      common/sql_help.py
  2. 3 1
      video_rewriting/video_prep.py

+ 2 - 2
common/sql_help.py

@@ -60,8 +60,8 @@ class sqlCollect():
 
 
     @classmethod
-    def insert_machine_making_data(cls, name: str, task_mark: str, channel_id: str, url: str, v_id: str, piaoquan_id: str, new_title: str, code: str, formatted_time, old_title: str):
-        insert_sql = f"""INSERT INTO machine_making_data (name, task_mark, channel, user, v_id, pq_uid, title, pq_vid, data_time, old_title) values ("{name}", "{task_mark}", "{channel_id}", "{url}", "{v_id}" , "{piaoquan_id}", "{new_title}", "{code}", "{formatted_time}", "{old_title}")"""
+    def insert_machine_making_data(cls, name: str, task_mark: str, channel_id: str, url: str, v_id: str, piaoquan_id: str, new_title: str, code: str, formatted_time, old_title: str, oss_object_key: str):
+        insert_sql = f"""INSERT INTO machine_making_data (name, task_mark, channel, user, v_id, pq_uid, title, pq_vid, data_time, old_title, oss_object_key) values ("{name}", "{task_mark}", "{channel_id}", "{url}", "{v_id}" , "{piaoquan_id}", "{new_title}", "{code}", "{formatted_time}", "{old_title}", "{oss_object_key}")"""
         MysqlHelper.update_values(
             sql=insert_sql
         )

+ 3 - 1
video_rewriting/video_prep.py

@@ -155,6 +155,8 @@ class getVideo:
                         old_title = video['old_title']
                         if old_title == "" or len(old_title) == 0 or old_title == None:
                             old_title = '⭕分享给大家一个视频!值得细品!'
+                            Common.logger("title").info(f"{name}的{task_mark}下的视频{url},标题为空,使用兜底标题生成片尾")
+
                         time.sleep(1)
                         pw_random_id = cls.random_id()
                         if channel_id == "票圈" or channel_id == "快手创作者版":
@@ -274,7 +276,7 @@ class getVideo:
                                 sqlCollect.insert_task(task_mark, v_id, mark, channel_id)  # 插入数据库
                                 current_time = datetime.now()
                                 formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
-                                sqlCollect.insert_machine_making_data(name, task_mark, channel_id, url, v_id, piaoquan_id, new_title, code, formatted_time, old_title)
+                                sqlCollect.insert_machine_making_data(name, task_mark, channel_id, url, v_id, piaoquan_id, new_title, code, formatted_time, old_title, oss_object_key)
                                 if title == "原标题" or title == "AI标题":
                                     values = [[name, task_mark, channel_id, url, str(v_id), piaoquan_id, old_title, title,new_title, str(code), formatted_time, str(rule)]]
                                 else: