from flask_prompt_master import create_app, db
from flask_prompt_master.models import PromptTemplate
import pymysql

def migrate_to_mysql():
    """迁移数据到MySQL"""
    app = create_app()
    
    with app.app_context():
        # 创建所有表
        db.create_all()
        
        try:
            # 检查是否已有数据
            if PromptTemplate.query.first() is None:
                # 执行init_db中的初始化
                from flask_prompt_master.init_db import init_db
                init_db()
                print("模板数据迁移完成！")
            else:
                print("数据库中已存在模板数据，跳过迁移。")
                
        except Exception as e:
            print(f"迁移过程中出错: {str(e)}")
            db.session.rollback()
        
if __name__ == '__main__':
    migrate_to_mysql()