import pymysql
from configs.config import MYSQL_HOST, MYSQL_PORT, MYSQL_NAME, MYSQL_USER, MYSQL_PASS

# 创建 MySQL 连接
def create_connection():
    return pymysql.connect(
        host=MYSQL_HOST,
        user=MYSQL_USER,
        port=MYSQL_PORT,  # 应该使用 MYSQL_PORT 而不是 MYSQL_HOST
        password=MYSQL_PASS,
        database=MYSQL_NAME,
        local_infile=True,
        cursorclass=pymysql.cursors.DictCursor
    )

# 连接 MySQL (开启 MySQL 服务)
conn = create_connection()

# 获取 MySQL 连接
def get_cursor():
    global conn
    try:
        conn.ping()
        return conn.cursor()
    except Exception:
        conn = create_connection()
        return conn.cursor()