PCM_Report/test_sqlserver_writer.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
测试 SQL Server 写入功能
"""
from sqlserver_writer import SQLServerWriter
from datetime import datetime

def test_connection():
    """测试数据库连接"""
    print("=" * 60)
    print("测试 SQL Server 连接")
    print("=" * 60)
    
    # 从 work_order_db_config.json 加载配置
    import json
    from pathlib import Path
    
    config_file = Path(__file__).parent / 'work_order_db_config.json'
    
    if not config_file.exists():
        print(f"❌ 配置文件不存在: {config_file}")
        print("请先创建 work_order_db_config.json 文件")
        return False
    
    with open(config_file, 'r', encoding='utf-8') as f:
        db_config = json.load(f)
    
    config = {
        'host': db_config.get('host', 'localhost'),
        'port': db_config.get('port', 1433),
        'database': db_config.get('database', ''),
        'username': db_config.get('username', ''),
        'password': db_config.get('password', ''),
    }
    
    print(f"使用配置: host={config['host']}, port={config['port']}, database={config['database']}")
    
    writer = SQLServerWriter(config)
    
    if writer.connect():
        print("✅ 连接成功")
        writer.disconnect()
        return True
    else:
        print("❌ 连接失败")
        return False

def test_write_data():
    """测试数据写入"""
    print("\n" + "=" * 60)
    print("测试数据写入")
    print("=" * 60)
    
    # 从 work_order_db_config.json 加载配置
    import json
    from pathlib import Path
    
    config_file = Path(__file__).parent / 'work_order_db_config.json'
    
    if not config_file.exists():
        print(f"❌ 配置文件不存在: {config_file}")
        return False
    
    with open(config_file, 'r', encoding='utf-8') as f:
        db_config = json.load(f)
    
    config = {
        'host': db_config.get('host', 'localhost'),
        'port': db_config.get('port', 1433),
        'database': db_config.get('database', ''),
        'username': db_config.get('username', ''),
        'password': db_config.get('password', ''),
    }
    
    # 准备测试数据
    test_data = {
        'order_no': 'TEST_' + datetime.now().strftime('%Y%m%d_%H%M%S'),
        'runin_date': datetime.now().date(),
        'operator_name': '测试员',
        'power_end_part_no': 'TEST-001',
        'motor_speed_rpm': 980,
        'ambient_temp_c': 25.5,
        'start_time': datetime.now(),
        'end_time': datetime.now(),
        'reviewer_name': '审核员',
        'is_normal': 1,
        'abnormal_desc': None,
        'remarks': '测试数据',
        # 温度数据示例
        'temp_main_1_t05': 45.2,
        'temp_main_1_t10': 48.5,
        'temp_main_1_t15': 51.3,
        'temp_main_1_t20': 53.8,
        'temp_main_1_t25': 55.2,
        'temp_main_1_t30': 56.1,
        'temp_main_1_t35': 56.8,
    }
    
    writer = SQLServerWriter(config)
    
    if not writer.connect():
        print("❌ 无法连接到数据库")
        return False
    
    try:
        success = writer.write_pump_600_data(test_data)
        
        if success:
            print(f"✅ 数据写入成功")
            print(f"   工单号: {test_data['order_no']}")
            return True
        else:
            print("❌ 数据写入失败")
            return False
    finally:
        writer.disconnect()

if __name__ == "__main__":
    print("SQL Server 写入功能测试")
    print()
    
    # 测试连接
    if test_connection():
        # 测试写入
        test_write_data()
    
    print("\n" + "=" * 60)
    print("测试完成")
    print("=" * 60)