#\!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
简单训练测试 - 验证API日志输出修复
"""

import os
import sys
import urllib.request
import json
import time

# 设置UTF-8编码
os.environ['PYTHONIOENCODING'] = 'utf-8'
os.environ['PYTHONLEGACYWINDOWSSTDIO'] = '0'

# Windows系统额外配置
if os.name == 'nt':
    try:
        os.system('chcp 65001 >nul 2>&1')
        if hasattr(sys.stdout, 'reconfigure'):
            sys.stdout.reconfigure(encoding='utf-8', errors='replace')
            sys.stderr.reconfigure(encoding='utf-8', errors='replace')
    except Exception as e:
        print(f'Warning: Failed to set UTF-8 encoding: {e}')

def send_training_request():
    """发送训练请求"""
    training_data = {
        'product_id': 'P001',
        'model_type': 'transformer', 
        'epochs': 2,  # 短期训练以便快速验证
        'training_mode': 'product'
    }
    
    try:
        json_data = json.dumps(training_data).encode('utf-8')
        req = urllib.request.Request(
            'http://127.0.0.1:5000/api/training',
            data=json_data,
            headers={'Content-Type': 'application/json'}
        )
        
        with urllib.request.urlopen(req, timeout=30) as response:
            return response.getcode(), json.loads(response.read().decode('utf-8'))
    except Exception as e:
        return None, str(e)

def main():
    """主测试函数"""
    print('简单训练日志验证测试')
    print('=' * 50)
    print('目标: 验证API训练时的控制台输出')
    print('观察重点: API服务器控制台应该显示:')
    print('1. 训练任务线程启动消息')
    print('2. 参数配置信息')
    print('3. 训练器内部的详细进度')
    print('4. 带时间戳和线程ID的格式化输出')
    print('=' * 50)
    
    print('发送训练请求...')
    status, response = send_training_request()
    
    if status == 200:
        task_id = response.get('task_id')
        print(f'训练请求成功')
        print(f'任务ID: {task_id}')
        print(f'响应: {response.get("message")}')
        
        print('\n请观察API服务器控制台输出...')
        print('等待15秒让训练完成...')
        
        # 等待训练完成
        for i in range(15):
            time.sleep(1)
            print(f'等待中... {i+1}/15秒', end='\r')
        
        print('\n\n训练应该已完成')
        print('验证要点:')
        print('1. API服务器控制台是否显示了完整的训练日志')
        print('2. 日志格式是否包含 [时间戳][线程ID][任务ID][标签]')
        print('3. 中文字符和emoji是否正常显示')
        print('4. 训练器的进度信息是否可见')
        
    else:
        print(f'训练请求失败: {response}')

if __name__ == '__main__':
    main()