#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
训练测试 - 正确的编码配置
测试训练器是否正常输出日志，保留所有中文和emoji字符
"""

import os
import sys
import subprocess

def setup_encoding():
    """设置正确的UTF-8编码环境"""
    os.environ['PYTHONIOENCODING'] = 'utf-8'
    os.environ['PYTHONLEGACYWINDOWSSTDIO'] = '0'
    
    # Windows控制台编码设置
    if os.name == 'nt':
        try:
            # 设置控制台代码页为UTF-8
            subprocess.run(['chcp', '65001'], 
                         capture_output=True, 
                         shell=True, 
                         check=False)
            
            # 重新配置标准输出
            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"警告：编码设置失败: {e}")

def test_direct_training():
    """测试直接训练功能"""
    
    print("=" * 60)
    print("🧪 训练日志输出测试开始")
    print("💡 目标: 验证训练器控制台输出是否正常")
    print("📋 策略: 保留所有中文和emoji，通过编码配置解决乱码")
    print("=" * 60)
    
    # 添加server路径
    current_dir = os.path.dirname(os.path.abspath(__file__))
    server_dir = os.path.join(current_dir, 'server')
    sys.path.insert(0, server_dir)
    
    try:
        print("📦 正在导入训练器模块...")
        from trainers.transformer_trainer import train_product_model_with_transformer
        print("✅ 训练器模块导入成功")
        
        print("\n🚀 开始训练测试")
        print("📊 产品ID: P001")
        print("🤖 模型类型: Transformer")
        print("⚙️ 训练轮次: 2 (快速测试)")
        print("🎯 期望: 看到详细的训练进度输出")
        print("-" * 60)
        
        # 调用训练器 - 应该能正常显示所有中文和emoji
        result = train_product_model_with_transformer(
            product_id='P001',
            epochs=2,
            training_mode='product'
        )
        
        print("-" * 60)
        print("🎉 训练测试完成!")
        
        if result:
            model, metrics, version = result
            print(f"📊 返回结果:")
            print(f"   模型对象: {type(model).__name__}")
            print(f"   模型版本: {version}")
            
            if metrics:
                print(f"   训练指标:")
                for key, value in metrics.items():
                    if isinstance(value, float):
                        print(f"     {key}: {value:.4f}")
                    else:
                        print(f"     {key}: {value}")
            else:
                print("   ⚠️  训练指标为空")
                
            print("✅ 训练成功完成，返回了完整结果")
        else:
            print("❌ 训练返回None，可能存在问题")
            
    except Exception as e:
        print(f"❌ 训练过程发生错误: {e}")
        print("\n🔍 错误详情:")
        import traceback
        traceback.print_exc()
        
        # 提供解决建议
        print("\n💡 可能的解决方案:")
        print("1. 确认数据文件存在: pharmacy_sales_multi_store.csv")
        print("2. 检查编码环境变量: PYTHONIOENCODING=utf-8")
        print("3. 使用批处理文件启动: .\\启动API服务器.bat")
    
    print("\n" + "=" * 60)
    print("🏁 训练日志输出测试结束")
    print("💬 如果看到了包含emoji和中文的训练进度，说明编码配置成功")
    print("📝 重要: 本次测试保留了所有原始中文和emoji字符")
    print("=" * 60)

def main():
    """主函数"""
    # 设置编码环境
    setup_encoding()
    
    print("🔧 编码环境配置:")
    print(f"   PYTHONIOENCODING: {os.environ.get('PYTHONIOENCODING', '未设置')}")
    print(f"   PYTHONLEGACYWINDOWSSTDIO: {os.environ.get('PYTHONLEGACYWINDOWSSTDIO', '未设置')}")
    print(f"   系统平台: {os.name}")
    print()
    
    # 运行训练测试
    test_direct_training()

if __name__ == "__main__":
    main()