v0.7

修改判断是否存在的逻辑

deal/algorithm_deal.py

@@ -65,3 +65,4 @@ class ArticleMatchAccount(object):
         """
         return self.check_params() if self.check_params() else self.get_score_list()
 
+

deal/db_deal.py

@@ -7,10 +7,11 @@ async def insert_text_mysql(mysql_client, video_id, text, title):
     """
     :return:
     """
-    select_sql = "select video_id from video_text;"
+    select_sql = f"""select video_id from video_text where video_id = {video_id};"""
     result = await mysql_client.select(select_sql)
+    print(result)
     if result:
-        return {"duplicated": "vid already exists"}
+        return {"duplicated": "vid already exists", "vid": video_id}
     else:
         insert_sql = f"""
         insert into video_text
@@ -19,6 +20,7 @@ async def insert_text_mysql(mysql_client, video_id, text, title):
         ({video_id},'{text}', '{title}');
         """
         await mysql_client.async_insert(insert_sql)
+        return {"info": "success insert text into mysql", "vid": video_id}
 
 
 async def get_text_by_id(mysql_client, video_id):

routes/vta_routes.py

@@ -42,8 +42,7 @@ def VTARoutes(mysql_client):
         video_title = params['title']
         try:
             response = whisper(video_id)
-            await insert_text_mysql(mysql_client, video_id, response['text'], video_title)
-            result = {"info": "success insert text into mysql", "vid": video_id}
+            result = await insert_text_mysql(mysql_client, video_id, response['text'], video_title)
         except Exception as e:
             result = {"error": str(e), "vid": video_id}
         return jsonify(result)