#!/usr/bin/env python3
"""
测试所有模型管理修复
"""
import urllib.request
import json

def test_complete_fixes():
    try:
        print("=== 综合测试模型管理修复 ===")
        
        # 1. 测试基本模型列表
        print("\n1. 测试基本模型列表:")
        url = 'http://localhost:5000/api/models'
        with urllib.request.urlopen(url) as response:
            data = response.read().decode('utf-8')
            result = json.loads(data)
            
            print(f"状态码: {response.status}")
            models = result.get('data', [])
            pagination = result.get('pagination', {})
            
            print(f"模型数量: {len(models)}")
            print(f"总数: {pagination.get('total', 0)}")
            
            # 检查每个模型的关键字段
            for i, model in enumerate(models):
                print(f"\n模型 {i+1}:")
                model_id = model.get('model_id', '')
                filename = model.get('filename', '')
                metrics = model.get('metrics', {})
                
                print(f"  model_id: '{model_id}' {'✓' if model_id else '✗'}")
                print(f"  filename: '{filename}' {'✓' if filename else '✗'}")
                print(f"  product_id: '{model.get('product_id', '')}' ✓")
                print(f"  model_type: '{model.get('model_type', '')}' ✓")
                print(f"  metrics: {len(metrics)} 个指标 {'✓' if metrics else '✗'}")
                
                if metrics:
                    print(f"    - R2: {metrics.get('R2', 'N/A')}")
                    print(f"    - RMSE: {metrics.get('RMSE', 'N/A')}")
                    print(f"    - MAE: {metrics.get('MAE', 'N/A')}")
                
                # 模型能否删除（基于model_id）
                if model_id:
                    print(f"  删除URL: /api/models/{model_id} ✓")
                else:
                    print(f"  删除URL: 无法构造 ✗")
        
        # 2. 测试分页功能
        print("\n\n2. 测试分页功能:")
        url = 'http://localhost:5000/api/models?page=1&page_size=2'
        with urllib.request.urlopen(url) as response:
            data = response.read().decode('utf-8')
            result = json.loads(data)
            
            models = result.get('data', [])
            pagination = result.get('pagination', {})
            
            print(f"请求第1页，每页2条")
            print(f"返回数量: {len(models)}")
            print(f"分页信息:")
            print(f"  当前页: {pagination.get('page', 0)}")
            print(f"  每页大小: {pagination.get('page_size', 0)}")
            print(f"  总数: {pagination.get('total', 0)}")
            print(f"  总页数: {pagination.get('total_pages', 0)}")
            print(f"  有下一页: {pagination.get('has_next', False)}")
        
        # 3. 测试单个模型删除URL（不实际删除）
        if models and models[0].get('model_id'):
            model_id = models[0]['model_id']
            print(f"\n3. 测试删除端点（不实际删除）:")
            print(f"模型ID: {model_id}")
            print(f"删除URL: DELETE /api/models/{model_id}")
            print("前端应该能正确构造此URL")
        
        # 4. 总结
        print(f"\n\n=== 修复状态总结 ===")
        total_models = pagination.get('total', 0)
        working_models = sum(1 for m in models if m.get('model_id') and m.get('filename'))
        models_with_metrics = sum(1 for m in models if m.get('metrics'))
        
        print(f"✓ 分页功能: 已实现")
        print(f"✓ 总模型数: {total_models}")
        print(f"✓ 可用模型: {working_models}/{len(models)} (有model_id)")
        print(f"✓ 有指标模型: {models_with_metrics}/{len(models)}")
        print(f"✓ 删除功能: {'可用' if working_models > 0 else '不可用'}")
        
        if working_models == len(models) and models_with_metrics > 0:
            print("\n🎉 所有功能修复成功！")
        else:
            print(f"\n⚠️ 仍需要重启API服务器以应用所有修复")
            
    except Exception as e:
        print(f"测试失败: {e}")

if __name__ == "__main__":
    test_complete_fixes()