117 lines
3.2 KiB
Python
117 lines
3.2 KiB
Python
"""
|
|
模块注册和路由管理
|
|
提供统一的模块注册机制，方便添加新功能模块
|
|
"""
|
|
from typing import Dict, Callable, Optional, Any
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class ActionConfig:
|
|
"""操作配置"""
|
|
handler: Callable # 处理函数
|
|
module_name: str # 所属模块名称
|
|
description: Optional[str] = None # 描述
|
|
requires_auth: bool = True # 是否需要认证
|
|
header_module: Optional[str] = None # 使用的请求头模块名称
|
|
|
|
|
|
class ModuleRegistry:
|
|
"""模块注册表"""
|
|
|
|
def __init__(self):
|
|
self._actions: Dict[str, ActionConfig] = {}
|
|
self._modules: Dict[str, Dict[str, Any]] = {}
|
|
|
|
def register_action(
|
|
self,
|
|
action_name: str,
|
|
handler: Callable,
|
|
module_name: str = "default",
|
|
description: Optional[str] = None,
|
|
requires_auth: bool = True,
|
|
header_module: Optional[str] = None
|
|
):
|
|
"""
|
|
注册操作
|
|
|
|
Args:
|
|
action_name: 操作名称
|
|
handler: 处理函数
|
|
module_name: 模块名称
|
|
description: 描述
|
|
requires_auth: 是否需要认证
|
|
header_module: 请求头模块名称
|
|
"""
|
|
config = ActionConfig(
|
|
handler=handler,
|
|
module_name=module_name,
|
|
description=description,
|
|
requires_auth=requires_auth,
|
|
header_module=header_module
|
|
)
|
|
self._actions[action_name] = config
|
|
|
|
def get_action(self, action_name: str) -> Optional[ActionConfig]:
|
|
"""
|
|
获取操作配置
|
|
|
|
Args:
|
|
action_name: 操作名称
|
|
|
|
Returns:
|
|
操作配置，如果不存在返回None
|
|
"""
|
|
return self._actions.get(action_name)
|
|
|
|
def get_all_actions(self) -> Dict[str, ActionConfig]:
|
|
"""获取所有注册的操作"""
|
|
return self._actions.copy()
|
|
|
|
def register_module(self, module_name: str, module_instance: Any, **metadata):
|
|
"""
|
|
注册模块实例
|
|
|
|
Args:
|
|
module_name: 模块名称
|
|
module_instance: 模块实例
|
|
**metadata: 模块元数据
|
|
"""
|
|
self._modules[module_name] = {
|
|
'instance': module_instance,
|
|
**metadata
|
|
}
|
|
|
|
def get_module(self, module_name: str) -> Optional[Any]:
|
|
"""
|
|
获取模块实例
|
|
|
|
Args:
|
|
module_name: 模块名称
|
|
|
|
Returns:
|
|
模块实例，如果不存在返回None
|
|
"""
|
|
module_info = self._modules.get(module_name)
|
|
return module_info['instance'] if module_info else None
|
|
|
|
def get_actions_by_module(self, module_name: str) -> Dict[str, ActionConfig]:
|
|
"""
|
|
获取指定模块的所有操作
|
|
|
|
Args:
|
|
module_name: 模块名称
|
|
|
|
Returns:
|
|
操作字典
|
|
"""
|
|
return {
|
|
name: config
|
|
for name, config in self._actions.items()
|
|
if config.module_name == module_name
|
|
}
|
|
|
|
|
|
# 全局模块注册表实例
|
|
registry = ModuleRegistry()
|