#!/usr/bin/env python3
"""修复 condition_json 字段类型"""

import asyncio
from sqlalchemy.ext.asyncio import create_async_engine
from sqlalchemy import text

DB_CONFIG = {
    'host': '123.57.81.127',
    'port': 3306,
    'user': 'cpy_admin',
    'password': 'Tgzz2025+',
    'database': 'ruoyi-fastapi'
}

DATABASE_URL = (
    f"mysql+asyncmy://{DB_CONFIG['user']}:{DB_CONFIG['password']}@"
    f"{DB_CONFIG['host']}:{DB_CONFIG['port']}/{DB_CONFIG['database']}"
)

async def fix():
    engine = create_async_engine(DATABASE_URL, echo=False)
    async with engine.connect() as conn:
        print("Fixing condition_json column type...")
        await conn.execute(text("""
            ALTER TABLE test_item MODIFY COLUMN condition_json TEXT NULL COMMENT '测试条件模板JSON'
        """))
        await conn.commit()
        print("[OK] condition_json changed to TEXT")
    await engine.dispose()

if __name__ == '__main__':
    asyncio.run(fix())