app-init

tests/rpa_tunnel.py

@@ -0,0 +1,31 @@
+"""影刀 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()