#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
更新数据库表字符集为utf8mb4，以支持emoji和其他特殊字符
"""

import logging
from cryptoai.utils.db_manager import get_db_manager
from sqlalchemy import text

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger('update_db_charset')

def update_table_charset():
    """更新数据库表字符集为utf8mb4"""
    try:
        # 获取数据库管理器
        db_manager = get_db_manager()
        
        if not db_manager.engine:
            logger.error("数据库连接失败")
            return False
        
        # 创建会话
        session = db_manager.Session()
        
        try:
            # 更新数据库字符集
            session.execute(text("ALTER DATABASE CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;"))
            
            # 更新agent_feeds表字符集
            session.execute(text("""
                ALTER TABLE agent_feeds 
                CONVERT TO CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
            """))
            
            # 特别更新content列的字符集
            session.execute(text("""
                ALTER TABLE agent_feeds 
                MODIFY content TEXT CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;
            """))
            
            session.commit()
            logger.info("成功更新数据库表字符集为utf8mb4")
            return True
            
        except Exception as e:
            session.rollback()
            logger.error(f"更新数据库表字符集失败: {e}")
            return False
            
        finally:
            session.close()
            
    except Exception as e:
        logger.error(f"更新数据库表字符集失败: {e}")
        return False

if __name__ == "__main__":
    update_table_charset()