#!/usr/bin/env python3
"""
直接测试ModelManager功能
"""
import sys
import os

# 添加server目录到路径
server_path = os.path.join(os.path.dirname(__file__), 'server')
sys.path.insert(0, server_path)

from utils.model_manager import ModelManager

def test_model_manager():
    print("=== 直接测试ModelManager ===")
    
    # 创建ModelManager实例
    model_manager = ModelManager()
    
    # 获取模型列表
    models = model_manager.list_models()
    
    print(f"找到模型数量: {len(models)}")
    
    for i, model in enumerate(models):
        print(f"\n--- 模型 {i+1} (直接从ModelManager) ---")
        
        # 显示所有字段
        for key, value in model.items():
            print(f"{key}: '{value}' ({type(value).__name__})")
        
        # 特别检查filename
        filename = model.get('filename', 'MISSING')
        if filename != 'MISSING':
            print(f"✓ filename字段存在: {filename}")
            expected_model_id = filename.replace('.pth', '')
            print(f"✓ 期望model_id: {expected_model_id}")
        else:
            print("✗ filename字段缺失!")

if __name__ == "__main__":
    test_model_manager()