#!/usr/bin/env python3
"""
简单调试脚本
"""
import sys
import os

# 添加server路径
server_path = os.path.join(os.path.dirname(__file__), 'server')
sys.path.append(server_path)

print("=== 简单调试测试 ===")

# 1. 测试基本导入
print("1. 测试基本导入...")
try:
    import pandas as pd
    print("OK pandas导入成功")
except Exception as e:
    print(f"ERROR pandas导入失败: {e}")

try:
    import torch
    print("OK torch导入成功")
except Exception as e:
    print(f"ERROR torch导入失败: {e}")

# 2. 测试进度管理器导入
print("\n2. 测试进度管理器导入...")
try:
    from utils.training_progress import progress_manager
    print("OK 进度管理器导入成功")
    print(f"进度管理器类型: {type(progress_manager)}")
except Exception as e:
    print(f"ERROR 进度管理器导入失败: {e}")

# 3. 测试训练器导入
print("\n3. 测试训练器导入...")
try:
    from trainers.mlstm_trainer import train_product_model_with_mlstm
    print("OK mLSTM训练器导入成功")
    
    # 检查函数参数
    import inspect
    sig = inspect.signature(train_product_model_with_mlstm)
    params = list(sig.parameters.keys())
    print(f"函数参数: {params}")
    
    has_socketio = 'socketio' in params
    has_task_id = 'task_id' in params
    print(f"有socketio参数: {has_socketio}")
    print(f"有task_id参数: {has_task_id}")
    
except Exception as e:
    print(f"ERROR mLSTM训练器导入失败: {e}")
    import traceback
    traceback.print_exc()

# 4. 测试数据文件
print("\n4. 测试数据文件...")
if os.path.exists('pharmacy_sales_multi_store.csv'):
    print("OK 数据文件存在")
    try:
        df = pd.read_csv('pharmacy_sales_multi_store.csv', nrows=5)  # 只读前5行
        print(f"OK 数据文件可读，列: {list(df.columns)}")
    except Exception as e:
        print(f"ERROR 数据文件读取失败: {e}")
else:
    print("ERROR 数据文件不存在")

print("\n=== 简单调试完成 ===")