73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
|
from flask_prompt_master import create_app, db
|
||
|
|
import pymysql
|
||
|
|
from flask_prompt_master.init_db import templates
|
||
|
|
|
||
|
|
def insert_all_templates():
|
||
|
|
"""向 container_orchestration_assistant 表插入所有模板数据"""
|
||
|
|
try:
|
||
|
|
# 连接MySQL数据库
|
||
|
|
conn = pymysql.connect(
|
||
|
|
host='localhost',
|
||
|
|
user='root',
|
||
|
|
password='123456',
|
||
|
|
database='food_db',
|
||
|
|
charset='utf8mb4'
|
||
|
|
)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
|
||
|
|
# 先清空表
|
||
|
|
cursor.execute("TRUNCATE TABLE container_orchestration_assistant")
|
||
|
|
|
||
|
|
# SQL 插入语句
|
||
|
|
sql = """
|
||
|
|
INSERT INTO container_orchestration_assistant
|
||
|
|
(name, description, category, industry, profession, sub_category, system_prompt)
|
||
|
|
VALUES (%(name)s, %(description)s, %(category)s, %(industry)s, %(profession)s,
|
||
|
|
%(sub_category)s, %(system_prompt)s)
|
||
|
|
"""
|
||
|
|
|
||
|
|
# 遍历所有模板数据并插入
|
||
|
|
success_count = 0
|
||
|
|
for template in templates:
|
||
|
|
try:
|
||
|
|
# 准备模板数据
|
||
|
|
template_data = {
|
||
|
|
'name': template['name'],
|
||
|
|
'description': template['description'],
|
||
|
|
'category': template.get('category', ''),
|
||
|
|
'industry': template.get('industry', ''),
|
||
|
|
'profession': template.get('profession', ''),
|
||
|
|
'sub_category': template.get('sub_category', ''),
|
||
|
|
'system_prompt': template['system_prompt']
|
||
|
|
}
|
||
|
|
|
||
|
|
# 执行插入
|
||
|
|
cursor.execute(sql, template_data)
|
||
|
|
success_count += 1
|
||
|
|
|
||
|
|
print(f"成功插入模板: {template['name']}")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"插入模板 {template['name']} 失败: {str(e)}")
|
||
|
|
continue
|
||
|
|
|
||
|
|
# 提交事务
|
||
|
|
conn.commit()
|
||
|
|
|
||
|
|
print("\n=== 数据插入完成 ===")
|
||
|
|
print(f"成功插入 {success_count} 个模板")
|
||
|
|
print(f"总计 {len(templates)} 个模板")
|
||
|
|
print("===================")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
print(f"数据库连接失败: {str(e)}")
|
||
|
|
if 'conn' in locals():
|
||
|
|
conn.rollback()
|
||
|
|
finally:
|
||
|
|
if 'cursor' in locals():
|
||
|
|
cursor.close()
|
||
|
|
if 'conn' in locals():
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
insert_all_templates()
|