20251222

AdminTool/test_db_connection.py

@@ -0,0 +1,127 @@
+# -*- coding: utf-8 -*-
+"""
+测试数据库连接和 app_deployments 表（通过 SSH 隧道）
+"""
+import json
+import pymysql
+from datetime import datetime
+import sys
+import io
+from sshtunnel import SSHTunnelForwarder
+
+# 设置 stdout 为 UTF-8
+sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
+
+def load_config():
+    with open('deploy_config.json', 'r', encoding='utf-8') as f:
+        return json.load(f)
+
+def test_connection():
+    print("="*60)
+    print("测试 MySQL 数据库连接（通过 SSH 隧道）")
+    print("="*60)
+    
+    config = load_config()
+    mysql_config = config['mysql']
+    
+    print(f"\n📋 SSH 服务器:")
+    print(f"  地址: {config['host']}")
+    print(f"  端口: {config['port']}")
+    print(f"  用户: {config['username']}")
+    
+    print(f"\n📋 MySQL 配置:")
+    print(f"  主机: {mysql_config['host']}")
+    print(f"  端口: {mysql_config['port']}")
+    print(f"  用户: {mysql_config['username']}")
+    print(f"  数据库: {mysql_config['database']}")
+    
+    tunnel = None
+    
+    try:
+        print("\n🔌 创建 SSH 隧道...")
+        tunnel = SSHTunnelForwarder(
+            (config['host'], int(config.get('port', 22))),
+            ssh_username=config['username'],
+            ssh_password=config['password'],
+            remote_bind_address=('127.0.0.1', int(mysql_config.get('port', 3306)))
+        )
+        tunnel.start()
+        print(f"✅ SSH 隧道已建立，本地端口: {tunnel.local_bind_port}")
+        
+        print("\n🔌 连接 MySQL...")
+        conn = pymysql.connect(
+            host='127.0.0.1',
+            port=tunnel.local_bind_port,
+            user=mysql_config['username'],
+            password=mysql_config['password'],
+            database=mysql_config['database'],
+            charset='utf8mb4',
+            cursorclass=pymysql.cursors.DictCursor
+        )
+        
+        print("✅ MySQL 连接成功！")
+        
+        # 测试创建表
+        print("\n📝 创建/检查 app_deployments 表...")
+        with conn.cursor() as cursor:
+            sql = """
+            CREATE TABLE IF NOT EXISTS app_deployments (
+                id INT PRIMARY KEY AUTO_INCREMENT,
+                version VARCHAR(50) UNIQUE NOT NULL,
+                deployed_at DATETIME NOT NULL,
+                is_current BOOLEAN DEFAULT FALSE,
+                file_size_mb DECIMAL(10, 2),
+                comment VARCHAR(500),
+                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+            )
+            """
+            cursor.execute(sql)
+            conn.commit()
+            print("✅ 表已就绪")
+        
+        # 查看表结构
+        print("\n📊 表结构:")
+        with conn.cursor() as cursor:
+            cursor.execute("DESCRIBE app_deployments")
+            for row in cursor.fetchall():
+                print(f"  {row['Field']}: {row['Type']} {'(主键)' if row['Key'] == 'PRI' else ''}")
+        
+        # 查看现有数据
+        print("\n📚 现有部署记录:")
+        with conn.cursor() as cursor:
+            cursor.execute("SELECT * FROM app_deployments ORDER BY deployed_at DESC")
+            records = cursor.fetchall()
+            
+            if records:
+                for i, rec in enumerate(records, 1):
+                    status = "✅ 当前" if rec['is_current'] else ""
+                    print(f"\n  [{i}] {rec['version']} {status}")
+                    print(f"      部署时间: {rec['deployed_at']}")
+                    print(f"      大小: {rec['file_size_mb']} MB")
+                    print(f"      备注: {rec['comment'] or '无'}")
+            else:
+                print("  (暂无记录)")
+        
+        conn.close()
+        
+        print("\n" + "="*60)
+        print("✅ 测试完成！数据库一切正常")
+        print("="*60)
+        
+    except Exception as e:
+        print(f"\n❌ 测试失败: {e}")
+        print("\n请检查:")
+        print("  1. SSH 服务器是否可访问")
+        print("  2. deploy_config.json 配置是否正确")
+        print("  3. 服务器上 MySQL 服务是否运行")
+        print("  4. 数据库用户权限是否足够")
+        return False
+    finally:
+        if tunnel:
+            tunnel.stop()
+            print("\n🔌 SSH 隧道已关闭")
+    
+    return True
+
+if __name__ == "__main__":
+    test_connection()