"""影刀 RPA：用 local_auth 连本机 mysql-tunnel，返回 conn。

须先启动 mysql-tunnel.exe（窗口点「启动」），listen / local_auth 与下面一致。
默认值对齐 config.example.yaml；按你的 config.yaml 改 port / password / db。
"""

import pymysql


def main(args):
    conn = pymysql.connect(
        host="127.0.0.1",
        port=13306,
        user="root",
        password="123456",
        db="your_database",
    )
    print("MySQL数据库连接成功")
    args["conn"] = conn
    return args


if __name__ == "__main__":
    out = main({})
    conn = out["conn"]
    try:
        with conn.cursor() as cur:
            cur.execute("SELECT 1")
            print("SELECT 1 ->", cur.fetchone())
    finally:
        conn.close()