74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
为API添加一个新的测试端点，用于验证修复
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def add_test_endpoint():
|
|||
|
|
"""在API文件末尾添加一个测试端点"""
|
|||
|
|
|
|||
|
|
endpoint_code = '''
|
|||
|
|
|
|||
|
|
# 测试端点 - 用于验证ModelManager修复
|
|||
|
|
@app.route('/api/models/test', methods=['GET'])
|
|||
|
|
def test_models_fix():
|
|||
|
|
"""
|
|||
|
|
测试端点 - 验证ModelManager修复是否生效
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
from utils.model_manager import ModelManager
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 强制创建新的ModelManager实例
|
|||
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
project_root = os.path.dirname(current_dir)
|
|||
|
|
model_dir = os.path.join(project_root, 'saved_models')
|
|||
|
|
manager = ModelManager(model_dir)
|
|||
|
|
|
|||
|
|
models = manager.list_models()
|
|||
|
|
|
|||
|
|
# 简化的响应格式
|
|||
|
|
test_result = {
|
|||
|
|
"status": "success",
|
|||
|
|
"test_name": "ModelManager修复测试",
|
|||
|
|
"model_dir": manager.model_dir,
|
|||
|
|
"dir_exists": os.path.exists(manager.model_dir),
|
|||
|
|
"models_found": len(models),
|
|||
|
|
"models": []
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for model in models:
|
|||
|
|
test_result["models"].append({
|
|||
|
|
"filename": model.get('filename', 'MISSING'),
|
|||
|
|
"model_id": model.get('filename', '').replace('.pth', '') if model.get('filename') else 'GENERATED_MISSING',
|
|||
|
|
"product_id": model.get('product_id', 'MISSING'),
|
|||
|
|
"model_type": model.get('model_type', 'MISSING')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return jsonify(test_result)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
return jsonify({
|
|||
|
|
"status": "error",
|
|||
|
|
"message": str(e),
|
|||
|
|
"test_name": "ModelManager修复测试"
|
|||
|
|
}), 500
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
# 读取当前API文件
|
|||
|
|
with open('server/api.py', 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
|
|||
|
|
# 检查是否已经添加了测试端点
|
|||
|
|
if '/api/models/test' in content:
|
|||
|
|
print("测试端点已存在")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 在文件末尾添加测试端点
|
|||
|
|
with open('server/api.py', 'a', encoding='utf-8') as f:
|
|||
|
|
f.write(endpoint_code)
|
|||
|
|
|
|||
|
|
print("已添加测试端点: /api/models/test")
|
|||
|
|
print("重启API服务器后访问: http://localhost:5000/api/models/test")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
add_test_endpoint()
|