"""
数据库连接配置
"""
import os
import pymysql
from contextlib import contextmanager
from lib.logger import log_info, log_error, log_warning

def get_db_config():
    """从环境变量或默认值获取数据库配置"""
    return {
        'host': os.environ.get('DB_HOST', 'haoslm2.xicp.net'),
        'port': int(os.environ.get('DB_PORT', 10216)),
        'user': os.environ.get('DB_USER', 'root'),
        'password': os.environ.get('DB_PASSWORD', 'DsideaL147258369'),
        'database': os.environ.get('DB_NAME', 'yltcharge'),
        'charset': 'utf8mb4',
        'cursorclass': pymysql.cursors.DictCursor,
        'autocommit': True,
        'connect_timeout': 30,
        'read_timeout': 60,
        'write_timeout': 60
    }

DB_CONFIG = get_db_config()


@contextmanager
def get_connection():
    """获取数据库连接（上下文管理器）"""
    try:
        conn = pymysql.connect(**DB_CONFIG)
        yield conn
    except pymysql.Error as e:
        log_error(f'数据库连接失败: {e}', 'db', exc_info=True)
        raise
    finally:
        try:
            conn.close()
        except:
            pass


def execute_query(sql, params=None):
    """执行查询并返回结果"""
    try:
        with get_connection() as conn:
            with conn.cursor() as cursor:
                cursor.execute(sql, params)
                result = cursor.fetchall()
                return result
    except Exception as e:
        log_error(f'查询失败: {e}\nSQL: {sql}\nParams: {params}', 'db', exc_info=True)
        raise


def execute_update(sql, params=None):
    """执行更新操作"""
    try:
        with get_connection() as conn:
            with conn.cursor() as cursor:
                affected = cursor.execute(sql, params)
                return cursor.rowcount
    except Exception as e:
        log_error(f'更新失败: {e}\nSQL: {sql}\nParams: {params}', 'db', exc_info=True)
        raise


def execute_insert(sql, params=None):
    """执行插入操作并返回插入ID"""
    try:
        with get_connection() as conn:
            with conn.cursor() as cursor:
                cursor.execute(sql, params)
                last_id = cursor.lastrowid
                return last_id
    except Exception as e:
        log_error(f'插入失败: {e}\nSQL: {sql}\nParams: {params}', 'db', exc_info=True)
        raise