"""
pip install aiomysql
"""
import logging

from aiomysql import create_pool

from Config.Config import *

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

# MySQL 配置
MYSQL_CONFIG = {
    "host": MYSQL_HOST,
    "port": MYSQL_PORT,
    "user": MYSQL_USER,
    "password": MYSQL_PASSWORD,
    "db": MYSQL_DB_NAME,
    "minsize": 1,
    "maxsize": 20,
    "charset": "utf8mb4"
}


# 初始化 MySQL 连接池
async def init_mysql_pool():
    return await create_pool(**MYSQL_CONFIG)


# 保存聊天记录到 MySQL
async def save_chat_to_mysql(mysql_pool, person_id, prompt, result, audio_url, duration, input_type=1, output_type=1,
                             input_image_type=0, image_width=0, image_height=0):
    async with mysql_pool.acquire() as conn:
        await conn.ping()  # 重置连接
        async with conn.cursor() as cur:
            await cur.execute(
                "INSERT INTO t_chat_log (person_id, user_input, model_response,audio_url,duration,input_type,output_type,input_image_type,image_width,image_height,create_time) VALUES (%s, %s, %s, %s, %s, %s, %s,%s,%s,%s,NOW())",
                (person_id, prompt, result, audio_url, duration, input_type, output_type, input_image_type, image_width,
                 image_height)
            )
            await conn.commit()