c
from flask_prompt_master import create_app, db
import pymysql

def insert_data():
    """向 container_orchestration_assistant 表插入数据"""
    try:
        # 连接MySQL数据库
        conn = pymysql.connect(
            host='localhost',
            user='root',
            password='123456',
            database='food_db',
            charset='utf8mb4'
        )
        cursor = conn.cursor()
        
        # 准备插入的数据
        template_data = {
            'name': '容器编排助手',
            'description': '优化容器化、自动化部署相关的提示词',
            'category': '架构设计',
            'industry': '技术研发',
            'profession': '高级工程师',
            'sub_category': '容器编排',
            'system_prompt': """你是一个专业的容器编排提示词专家。对于容器化需求，你需要：
1. 明确容器化策略
   - 确定容器化范围
   - 选择基础镜像
   - 规划镜像分层
   - 优化构建过程

2. 设计编排配置
   - 定义服务组件
   - 配置资源限制
   - 设置环境变量
   - 管理配置文件

3. 规划资源调度
   - 节点亲和性策略
   - 容器分布规则
   - 资源预留配置
   - 弹性伸缩方案

4. 考虑服务发现
   - 服务注册机制
   - 负载均衡策略
   - 健康检查配置
   - 网络通信规则

5. 注意状态管理
   - 持久化存储方案
   - 数据备份策略
   - 日志收集管理
   - 监控告警集成"""
        }
        
        # 构建 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)
        """
        
        # 执行插入
        cursor.execute(sql, template_data)
        
        # 提交事务
        conn.commit()
        
        print("=== 数据插入成功 ===")
        print(f"名称: {template_data['name']}")
        print(f"描述: {template_data['description']}")
        print(f"分类: {template_data['category']}")
        print(f"行业: {template_data['industry']}")
        print(f"职业: {template_data['profession']}")
        print(f"子类别: {template_data['sub_category']}")
        print("===================")
            
    except Exception as e:
        print(f"数据库操作失败: {str(e)}")
        conn.rollback()
    finally:
        cursor.close()
        conn.close()

if __name__ == '__main__':
    insert_data()
	
	
#插入全部数据
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()