#!/usr/bin/env python3
"""
测试训练日志输出
"""
import requests
import json
import time

def test_training_logs():
    """测试训练API和日志输出"""
    
    # 1. 启动训练任务
    print("=== 启动训练任务 ===")
    training_data = {
        "product_id": "P001",
        "model_type": "mlstm",
        "epochs": 3,  # 只训练3个epoch进行测试
        "training_mode": "product"
    }
    
    try:
        response = requests.post(
            'http://localhost:5000/api/training/start',
            json=training_data,
            headers={'Content-Type': 'application/json'}
        )
        
        if response.status_code == 200:
            result = response.json()
            task_id = result.get('task_id')
            print(f"✅ 训练任务已启动，任务ID: {task_id}")
            
            # 2. 监控训练状态
            print("\n=== 监控训练状态 ===")
            for i in range(30):  # 最多监控30次
                time.sleep(2)
                
                status_response = requests.get(f'http://localhost:5000/api/training/status/{task_id}')
                if status_response.status_code == 200:
                    status_data = status_response.json()
                    status = status_data.get('status', 'unknown')
                    print(f"第{i+1}次检查 - 状态: {status}")
                    
                    if status == 'completed':
                        print("✅ 训练已完成!")
                        print(f"指标: {status_data.get('metrics', 'None')}")
                        break
                    elif status == 'failed':
                        print("❌ 训练失败!")
                        print(f"错误: {status_data.get('error', 'Unknown')}")
                        break
                else:
                    print(f"⚠️ 无法获取状态: {status_response.status_code}")
            
        else:
            print(f"❌ 启动训练失败: {response.status_code}")
            print(f"响应: {response.text}")
            
    except Exception as e:
        print(f"❌ 测试失败: {e}")

if __name__ == "__main__":
    test_training_logs()