51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
VWED.api 模块 - API接口注册
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import Dict, Any, Callable
|
|||
|
|
from ..script_registry_service import get_global_registry
|
|||
|
|
|
|||
|
|
|
|||
|
|
class VWEDApiModule:
|
|||
|
|
"""VWED.api 模块 - API接口注册"""
|
|||
|
|
|
|||
|
|
def __init__(self, script_id: str):
|
|||
|
|
self.script_id = script_id
|
|||
|
|
self.registry = get_global_registry()
|
|||
|
|
|
|||
|
|
def register_route(self, path: str, method: str = "GET", handler: Callable = None,
|
|||
|
|
description: str = "", params: Dict = None,
|
|||
|
|
parameters: Dict = None, response_schema: Dict = None):
|
|||
|
|
"""分离式API接口注册（用户要求的新格式）
|
|||
|
|
|
|||
|
|
使用方式：
|
|||
|
|
VWED.api.register_route(
|
|||
|
|
path="/calculate",
|
|||
|
|
method="POST",
|
|||
|
|
handler=calculate_handler,
|
|||
|
|
description="数学计算接口",
|
|||
|
|
params={"a": 0, "b": 0, "operation": "add"} # 简化的参数定义
|
|||
|
|
)
|
|||
|
|
"""
|
|||
|
|
# 验证HTTP方法
|
|||
|
|
valid_methods = ["GET", "POST", "PUT", "DELETE"]
|
|||
|
|
if method.upper() not in valid_methods:
|
|||
|
|
raise ValueError(f"method参数错误：'{method}' 不是有效的HTTP请求方式。支持的方式有：{', '.join(valid_methods)}")
|
|||
|
|
|
|||
|
|
if handler is None:
|
|||
|
|
raise ValueError("handler参数不能为空")
|
|||
|
|
|
|||
|
|
self.registry.register_api_route(
|
|||
|
|
path=path,
|
|||
|
|
method=method.upper(),
|
|||
|
|
handler=handler,
|
|||
|
|
script_id=self.script_id,
|
|||
|
|
description=description,
|
|||
|
|
params=params, # 传递简化的参数格式
|
|||
|
|
parameters=parameters,
|
|||
|
|
response_schema=response_schema
|
|||
|
|
)
|
|||
|
|
return handler
|