82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
#!/usr/bin/env python
|
||
"""
|
||
直接测试API端点逻辑
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
sys.path.append('server')
|
||
|
||
def simulate_api_call():
|
||
"""模拟API调用"""
|
||
print("=== 模拟API端点调用 ===")
|
||
|
||
try:
|
||
# 模拟从server目录运行
|
||
original_cwd = os.getcwd()
|
||
server_dir = os.path.join(original_cwd, 'server')
|
||
os.chdir(server_dir)
|
||
|
||
# 重新添加路径
|
||
sys.path.insert(0, server_dir)
|
||
|
||
print(f"当前工作目录: {os.getcwd()}")
|
||
|
||
# 直接导入模型管理器（模拟API中的导入）
|
||
from utils.model_manager import model_manager
|
||
|
||
print(f"模型管理器目录: {model_manager.model_dir}")
|
||
print(f"目录存在: {os.path.exists(model_manager.model_dir)}")
|
||
|
||
if os.path.exists(model_manager.model_dir):
|
||
files = os.listdir(model_manager.model_dir)
|
||
print(f"目录文件: {files}")
|
||
|
||
# 调用API中的逻辑
|
||
models = model_manager.list_models()
|
||
print(f"原始模型数量: {len(models)}")
|
||
|
||
# 格式化响应数据（与API相同）
|
||
formatted_models = []
|
||
for model in models:
|
||
formatted_model = {
|
||
'model_id': model.get('filename', '').replace('.pth', ''),
|
||
'product_id': model.get('product_id', ''),
|
||
'product_name': model.get('product_name', model.get('product_id', '')),
|
||
'model_type': model.get('model_type', ''),
|
||
'training_mode': model.get('training_mode', 'product'),
|
||
'store_id': model.get('store_id'),
|
||
'aggregation_method': model.get('aggregation_method'),
|
||
'version': model.get('version', 'v1'),
|
||
'created_at': model.get('created_at', model.get('modified_at', '')),
|
||
'file_size': model.get('file_size', 0),
|
||
'metrics': model.get('metrics', {}),
|
||
'config': model.get('config', {})
|
||
}
|
||
formatted_models.append(formatted_model)
|
||
|
||
print(f"格式化后模型数量: {len(formatted_models)}")
|
||
|
||
if formatted_models:
|
||
print("API将返回:")
|
||
import json
|
||
print(json.dumps({
|
||
"status": "success",
|
||
"data": formatted_models,
|
||
"total": len(formatted_models)
|
||
}, indent=2, ensure_ascii=False))
|
||
else:
|
||
print("API将返回空数据")
|
||
|
||
return formatted_models
|
||
|
||
except Exception as e:
|
||
print(f"模拟失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return []
|
||
finally:
|
||
os.chdir(original_cwd)
|
||
|
||
if __name__ == "__main__":
|
||
simulate_api_call() |