#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
测试SQL Server数据库连接
"""
from work_order_query import WORK_ORDER_DB_CONFIG, reload_config

def test_connection():
    """测试数据库连接"""
    print("=" * 50)
    print("测试SQL Server数据库连接")
    print("=" * 50)
    
    # 重新加载配置
    reload_config()
    
    print(f"连接配置:")
    print(f"  服务器: {WORK_ORDER_DB_CONFIG['host']}:{WORK_ORDER_DB_CONFIG['port']}")
    print(f"  数据库: {WORK_ORDER_DB_CONFIG['database']}")
    print(f"  用户名: {WORK_ORDER_DB_CONFIG['username']}")
    print()
    
    # 尝试pyodbc
    try:
        import pyodbc
        print("✓ pyodbc已安装")
        
        # 列出可用的ODBC驱动
        drivers = pyodbc.drivers()
        print(f"  可用的ODBC驱动: {drivers}")
        
        # 查找SQL Server驱动
        sql_drivers = [d for d in drivers if 'SQL' in d.upper()]
        if sql_drivers:
            print(f"  找到SQL Server驱动: {sql_drivers}")
        else:
            print("  ⚠ 未找到SQL Server ODBC驱动")
    except ImportError:
        print("✗ pyodbc未安装")
    
    print()
    
    # 尝试pymssql
    try:
        import pymssql
        print("✓ pymssql已安装")
        
        # 测试连接
        try:
            conn = pymssql.connect(
                server=WORK_ORDER_DB_CONFIG['host'],
                port=WORK_ORDER_DB_CONFIG['port'],
                user=WORK_ORDER_DB_CONFIG['username'],
                password=WORK_ORDER_DB_CONFIG['password'],
                database=WORK_ORDER_DB_CONFIG['database']
            )
            print("✓ pymssql连接成功!")
            
            # 测试查询
            cursor = conn.cursor()
            cursor.execute("SELECT @@VERSION")
            row = cursor.fetchone()
            if row:
                print(f"  SQL Server版本: {row[0][:50]}...")
            
            cursor.close()
            conn.close()
        except Exception as e:
            print(f"✗ pymssql连接失败: {e}")
    except ImportError:
        print("✗ pymssql未安装")
    
    print()
    print("测试完成！")
    print()
    
    # 测试工单查询
    test_query = input("是否测试工单查询？(y/n): ")
    if test_query.lower() == 'y':
        work_order_no = input("请输入工单号进行测试: ")
        if work_order_no:
            from work_order_query import query_work_order
            result = query_work_order(work_order_no)
            if result:
                print("查询成功！")
                for key, value in result.items():
                    print(f"  {key}: {value}")
            else:
                print("未找到该工单号的信息")

if __name__ == "__main__":
    test_connection()
    input("\n按Enter键退出...")