Merge branch 'master' of https://ai-evolution.top:18182/xiangrifeng/jxgzcd

# Conflicts:
#	README.md

test_db_connection.py

@@ -0,0 +1,109 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+数据库连接测试脚本
+连接到MySQL数据库并测试连接是否成功
+支持Windows和Linux (CentOS) 环境
+"""
+
+import pymysql
+import sys
+import os
+
+# 设置控制台输出编码（Windows需要，Linux默认UTF-8）
+if sys.platform == 'win32':
+    try:
+        os.system('chcp 65001 >nul 2>&1')
+    except:
+        pass
+else:
+    # Linux环境下设置UTF-8编码
+    import locale
+    try:
+        locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
+    except:
+        try:
+            locale.setlocale(locale.LC_ALL, 'C.UTF-8')
+        except:
+            pass
+
+def test_db_connection():
+    """测试数据库连接"""
+    # 数据库连接配置
+    db_config = {
+        'host': '127.0.0.1',
+        'port': 3306,
+        'user': 'prod',
+        'password': 'hmdmxjIvfIjIoflL',
+        'database': 'iot',
+        'charset': 'utf8mb4',
+        'connect_timeout': 10  # 连接超时时间（秒）
+    }
+    
+    connection = None
+    try:
+        # 尝试连接数据库
+        print("正在尝试连接到数据库...")
+        print(f"主机: {db_config['host']}")
+        print(f"端口: {db_config['port']}")
+        print(f"用户名: {db_config['user']}")
+        print("-" * 50)
+        
+        connection = pymysql.connect(**db_config)
+        
+        # 如果连接成功，执行一个简单查询来验证
+        with connection.cursor() as cursor:
+            cursor.execute("SELECT VERSION()")
+            version = cursor.fetchone()
+            print("[成功] 数据库连接成功！")
+            print(f"[成功] MySQL版本: {version[0]}")
+            
+            # 获取当前数据库名称
+            cursor.execute("SELECT DATABASE()")
+            db_name = cursor.fetchone()
+            if db_name[0]:
+                print(f"[成功] 当前数据库: {db_name[0]}")
+            else:
+                print("[成功] 当前未选择数据库")
+            
+            # 获取连接ID
+            cursor.execute("SELECT CONNECTION_ID()")
+            conn_id = cursor.fetchone()
+            print(f"[成功] 连接ID: {conn_id[0]}")
+        
+        return True
+        
+    except pymysql.Error as e:
+        error_code, error_msg = e.args
+        print("[失败] 数据库连接失败！")
+        print(f"[失败] 错误代码: {error_code}")
+        print(f"[失败] 错误信息: {error_msg}")
+        return False
+        
+    except Exception as e:
+        print("[失败] 发生未知错误！")
+        print(f"[失败] 错误信息: {e}")
+        return False
+        
+    finally:
+        # 关闭数据库连接
+        if connection:
+            connection.close()
+            print("-" * 50)
+            print("[成功] 数据库连接已关闭")
+
+if __name__ == "__main__":
+    print("=" * 50)
+    print("数据库连接测试")
+    print("=" * 50)
+    
+    success = test_db_connection()
+    
+    print("=" * 50)
+    if success:
+        print("测试结果: 成功")
+        sys.exit(0)
+    else:
+        print("测试结果: 失败")
+        sys.exit(1)
+