2024-06-12

数据表用config文件来配置

applications/routes.py

@@ -8,6 +8,7 @@ from quart import Blueprint, jsonify, request
 
 from applications.functions.log import logging
 from applications.schedule import recall_videos, search_videos, re_search_videos
+from applications.static.config import db_article
 
 my_blueprint = Blueprint('LongArticles', __name__)
 
@@ -62,7 +63,7 @@ def Routes(mysql_client):
         if "video_id=" in title:
             video_id = title.split("video_id=")[-1]
             insert_sql = f"""
-                            INSERT INTO long_articles_video
+                            INSERT INTO {db_article}
                                 (trace_id, gh_id, article_title, article_text, account_name, recall_video_id1)
                             VALUES 
                                 ('{trace_id}', '{gh_id}', '{title}', '{contents}', '{account_name}', '{video_id}');"""
@@ -75,7 +76,7 @@ def Routes(mysql_client):
             return jsonify(res)
         else:
             insert_sql = f"""
-                INSERT INTO long_articles_video
+                INSERT INTO {db_article}
                     (trace_id, gh_id, article_title, article_text, account_name)
                 VALUES 
                     ('{trace_id}', '{gh_id}', '{title}', '{contents}', '{account_name}');"""

applications/schedule/process_schedule.py

@@ -6,6 +6,7 @@ import json
 
 from applications.functions.common import request_for_info, create_gzh_path
 from applications.functions.log import logging
+from applications.static.config import db_article
 
 
 async def return_info_v2(video_id, trace_id, mysql_client):
@@ -44,7 +45,7 @@ async def return_info_v2(video_id, trace_id, mysql_client):
         "paragraphPosition": 0.25
     }
     update_result_sql = f"""
-                        UPDATE long_articles_video
+                        UPDATE {db_article}
                         SET
                             result1 = '{json.dumps(result, ensure_ascii=False)}',
                             success = 1
@@ -105,7 +106,7 @@ async def return_info(video_id, kimi_title, trace_id, mysql_client, index):
         "paragraphPosition": index * 0.25
     }
     update_result_sql = f"""
-                    UPDATE long_articles_video
+                    UPDATE {db_article}
                     SET
                         result{index} = '{json.dumps(result, ensure_ascii=False)}',
                         success = 1
@@ -136,7 +137,7 @@ async def recall_videos(trace_id, mysql_client):
     """
     select_sql = f"""
         SELECT recall_video_id1, recall_video_id2, recall_video_id3, kimi_title 
-        FROM long_articles_video
+        FROM {db_article}
         WHERE trace_id = '{trace_id}';
     """
     info_tuple = await mysql_client.async_select(select_sql)

applications/schedule/search_schedule.py

@@ -6,7 +6,7 @@ import json
 import time
 
 from applications.search import *
-from applications.static.config import gh_id_dict
+from applications.static.config import gh_id_dict, db_article
 from applications.functions.log import logging
 from applications.functions.video_item import VideoProducer
 from applications.functions.mysql import select_sensitive_words
@@ -358,7 +358,7 @@ async def search_videos(params, trace_id, gh_id, mysql_client):
     content_title = kimi_info['content_title'].replace("'", "").replace('"', "")
     content_keys = json.dumps(kimi_info['content_keys'], ensure_ascii=False)
     update_kimi_sql = f"""
-                    UPDATE long_articles_video SET
+                    UPDATE {db_article} SET
                     kimi_title = '{kimi_title}',
                     kimi_summary = '{content_title}',
                     kimi_keys = '{content_keys}'

applications/static/config.py

@@ -362,4 +362,10 @@ ab_test_config = {
     "gh_89ef4798d3ea": 1,
     "gh_a2901d34f75b": 1,
     "gh_b15de7c99912": 1
-}
+}
+
+# prod
+db_article = "long_articles_video"
+
+# dev
+# db_article = "long_articles_video_dev"