ErisPulse 2.3.4.dev0__py3-none-any.whl → 2.3.4.dev2__py3-none-any.whl

ErisPulse/loaders/base_loader.py

@@ -0,0 +1,162 @@
+"""
+ErisPulse 基础加载器
+
+定义加载器的抽象基类，提供通用的加载器接口和结构
+
+{!--< tips >!--}
+1. 所有具体加载器应继承自 BaseLoader
+2. 子类需实现 _process_entry_point 方法
+3. 支持启用/禁用配置管理
+{!--< /tips >!--}
+"""
+
+import importlib.metadata
+from abc import ABC, abstractmethod
+from typing import Dict, List, Any, Type, Tuple
+from ..Core.logger import logger
+from ..Core.config import config
+
+class BaseLoader(ABC):
+    """
+    基础加载器抽象类
+    
+    提供通用的加载器接口和配置管理功能
+    
+    {!--< tips >!--}
+    子类需要实现：
+    - _get_entry_point_group: 返回 entry-point 组名
+    - _process_entry_point: 处理单个 entry-point
+    - _should_eager_load: 判断是否立即加载
+    {!--< /tips >!--}
+    
+    {!--< internal-use >!--}
+    此类仅供内部使用，不应直接实例化
+    {!--< /internal-use >!--}
+    """
+    
+    def __init__(self, config_prefix: str):
+        """
+        初始化基础加载器
+        
+        :param config_prefix: 配置前缀（如 "ErisPulse.adapters" 或 "ErisPulse.modules"）
+        """
+        self._config_prefix = config_prefix
+    
+    @abstractmethod
+    def _get_entry_point_group(self) -> str:
+        """
+        获取 entry-point 组名
+        
+        :return: entry-point 组名
+        
+        {!--< internal-use >!--}
+        子类必须实现此方法
+        {!--< /internal-use >!--}
+        """
+        pass
+    
+    @abstractmethod
+    async def _process_entry_point(
+        self,
+        entry_point: Any,
+        objs: Dict[str, Any],
+        enabled_list: List[str],
+        disabled_list: List[str],
+        manager_instance: Any
+    ) -> Tuple[Dict[str, Any], List[str], List[str]]:
+        """
+        处理单个 entry-point
+        
+        :param entry_point: entry-point 对象
+        :param objs: 对象字典
+        :param enabled_list: 启用列表
+        :param disabled_list: 禁用列表
+        :param manager_instance: 管理器实例（用于调用 exists/is_enabled 等方法）
+        :return: (更新后的对象字典, 更新后的启用列表, 更新后的禁用列表)
+        
+        {!--< internal-use >!--}
+        子类必须实现此方法
+        {!--< /internal-use >!--}
+        """
+        pass
+    
+    async def load(self, manager_instance: Any) -> Tuple[Dict[str, Any], List[str], List[str]]:
+        """
+        从 entry-points 加载对象
+        
+        :param manager_instance: 管理器实例
+        :return: 
+            Dict[str, Any]: 对象字典
+            List[str]: 启用列表
+            List[str]: 禁用列表
+            
+        :raises ImportError: 当加载失败时抛出
+        """
+        objs: Dict[str, Any] = {}
+        enabled_list: List[str] = []
+        disabled_list: List[str] = []
+        
+        group_name = self._get_entry_point_group()
+        logger.info(f"正在加载 {group_name} entry-points...")
+        
+        try:
+            # 加载 entry-points
+            entry_points = importlib.metadata.entry_points()
+            if hasattr(entry_points, 'select'):
+                entries = entry_points.select(group=group_name)
+            else:
+                entries = entry_points.get(group_name, [])  # type: ignore[attr-defined]
+            
+            # 处理每个 entry-point
+            for entry_point in entries:
+                objs, enabled_list, disabled_list = await self._process_entry_point(
+                    entry_point, objs, enabled_list, disabled_list, manager_instance
+                )
+            
+            logger.info(f"{group_name} 加载完成")
+            
+        except Exception as e:
+            logger.error(f"加载 {group_name} entry-points 失败: {e}")
+            raise ImportError(f"无法加载 {group_name}: {e}")
+        
+        return objs, enabled_list, disabled_list
+    
+    def _register_config(self, name: str, enabled: bool = False) -> bool:
+        """
+        注册配置项
+        
+        :param name: 名称
+        :param enabled: 是否启用
+        :return: 操作是否成功
+        
+        {!--< internal-use >!--}
+        内部方法，用于注册新的配置项
+        {!--< /internal-use >!--}
+        """
+        config_key = f"{self._config_prefix}.status.{name}"
+        config.setConfig(config_key, enabled)
+        status = "启用" if enabled else "禁用"
+        logger.info(f"{self._config_prefix} {name} 已注册并{status}")
+        return True
+    
+    def _get_config_status(self, name: str) -> bool:
+        """
+        获取配置状态
+        
+        :param name: 名称
+        :return: 是否启用
+        
+        {!--< internal-use >!--}
+        内部方法，用于获取配置状态
+        {!--< /internal-use >!--}
+        """
+        config_key = f"{self._config_prefix}.status.{name}"
+        status = config.getConfig(config_key)
+        
+        if status is None:
+            return False
+        
+        if isinstance(status, str):
+            return status.lower() not in ('false', '0', 'no', 'off')
+        
+        return bool(status)

ErisPulse/loaders/base_loader.pyi

@@ -0,0 +1,23 @@
+# type: ignore
+#
+# Auto-generated type stub for base_loader.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse 基础加载器
+
+定义加载器的抽象基类，提供通用的加载器接口和结构
+
+{!--< tips >!--}
+1. 所有具体加载器应继承自 BaseLoader
+2. 子类需实现 _process_entry_point 方法
+3. 支持启用/禁用配置管理
+{!--< /tips >!--}
+"""
+
+import importlib.metadata
+from abc import ABC, abstractmethod
+from typing import Dict, List, Any, Type, Tuple
+from ..Core.logger import logger
+from ..Core.config import config

ErisPulse/loaders/initializer.py

@@ -0,0 +1,150 @@
+"""
+ErisPulse 初始化协调器
+
+负责协调适配器和模块的加载流程
+
+{!--< tips >!--}
+1. 初始化顺序：适配器 → 模块
+2. 支持并行加载优化
+3. 统一的错误处理和事件提交
+{!--< /tips >!--}
+"""
+
+import asyncio
+from typing import List, Dict, Any, Optional
+from .adapter_loader import AdapterLoader
+from .module_loader import ModuleLoader
+from ..Core.logger import logger
+from ..Core.lifecycle import lifecycle
+
+
+class Initializer:
+    """
+    初始化协调器
+    
+    协调适配器和模块的加载流程，提供统一的初始化接口
+    
+    {!--< tips >!--}
+    使用方式：
+    >>> initializer = Initializer(sdk_instance)
+    >>> success = await initializer.init()
+    {!--< /tips >!--}
+    """
+    
+    def __init__(self, sdk_instance: Any):
+        """
+        初始化协调器
+        
+        :param sdk_instance: SDK 实例
+        """
+        self._sdk = sdk_instance
+        self._adapter_loader = AdapterLoader()
+        self._module_loader = ModuleLoader()
+    
+    async def init(self) -> bool:
+        """
+        初始化所有模块和适配器
+        
+        执行步骤:
+        1. 从 PyPI 包加载适配器
+        2. 从 PyPI 包加载模块
+        3. 注册适配器
+        4. 注册模块
+        5. 初始化模块
+        
+        :return: bool 初始化是否成功
+        
+        :raises ImportError: 当加载失败时抛出
+        """
+        logger.info("[Init] SDK 正在初始化...")
+        lifecycle.start_timer("core.init")
+        
+        try:
+            # 1. 并行加载适配器和模块
+            adapter_manager = self._sdk.adapter
+            module_manager = self._sdk.module
+            
+            (adapter_result, module_result) = await asyncio.gather(
+                self._adapter_loader.load(adapter_manager),
+                self._module_loader.load(module_manager),
+                return_exceptions=True
+            )
+            
+            # 检查是否有异常
+            if isinstance(adapter_result, Exception):
+                logger.error(f"[Init] 适配器加载失败: {adapter_result}")
+                return False
+                
+            if isinstance(module_result, Exception):
+                logger.error(f"[Init] 模块加载失败: {module_result}")
+                return False
+            
+            # 解包结果
+            adapter_objs, enabled_adapters, disabled_adapters = adapter_result  # type: ignore
+            module_objs, enabled_modules, disabled_modules = module_result      # type: ignore
+            
+            logger.info(f"[Init] 加载了 {len(enabled_adapters)} 个适配器, {len(disabled_adapters)} 个适配器被禁用")
+            logger.info(f"[Init] 加载了 {len(enabled_modules)} 个模块, {len(disabled_modules)} 个模块被禁用")
+            
+            # 2. 注册适配器
+            logger.debug("[Init] 正在注册适配器...")
+            if not await self._adapter_loader.register_to_manager(
+                enabled_adapters, adapter_objs, adapter_manager
+            ):
+                return False
+            
+            # 3. 注册模块
+            logger.debug("[Init] 正在注册模块...")
+            if not await self._module_loader.register_to_manager(
+                enabled_modules, module_objs, module_manager
+            ):
+                return False
+            
+            # 4. 初始化模块（创建实例并挂载到 SDK）
+            logger.debug("[Init] 正在初始化模块...")
+            success = await self._module_loader.initialize_modules(
+                enabled_modules, module_objs, module_manager, self._sdk
+            )
+            
+            if success:
+                logger.info("[Init] SDK初始化成功")
+            else:
+                logger.error("[Init] SDK初始化失败")
+            
+            load_duration = lifecycle.stop_timer("core.init")
+            await lifecycle.submit_event(
+                "core.init.complete",
+                msg="模块初始化完成" if success else "模块初始化失败",
+                data={
+                    "duration": load_duration,
+                    "success": success,
+                    "adapters": {
+                        "enabled": enabled_adapters,
+                        "disabled": disabled_adapters
+                    },
+                    "modules": {
+                        "enabled": enabled_modules,
+                        "disabled": disabled_modules
+                    }
+                }
+            )
+            return success
+            
+        except Exception as e:
+            load_duration = lifecycle.stop_timer("core.init")
+            await lifecycle.submit_event(
+                "core.init.complete",
+                msg="模块初始化失败",
+                data={
+                    "duration": load_duration,
+                    "success": False,
+                    "error": str(e)
+                }
+            )
+            logger.critical(f"SDK初始化严重错误: {e}")
+            return False
+
+
+__all__ = [
+    "Initializer"
+]

ErisPulse/loaders/initializer.pyi

@@ -0,0 +1,60 @@
+# type: ignore
+#
+# Auto-generated type stub for initializer.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse 初始化协调器
+
+负责协调适配器和模块的加载流程
+
+{!--< tips >!--}
+1. 初始化顺序：适配器 → 模块
+2. 支持并行加载优化
+3. 统一的错误处理和事件提交
+{!--< /tips >!--}
+"""
+
+import asyncio
+from typing import List, Dict, Any, Optional
+from .adapter_loader import AdapterLoader
+from .module_loader import ModuleLoader
+from ..Core.logger import logger
+from ..Core.lifecycle import lifecycle
+
+class Initializer:
+    """
+    初始化协调器
+    
+    协调适配器和模块的加载流程，提供统一的初始化接口
+    
+    {!--< tips >!--}
+    使用方式：
+    >>> initializer = Initializer(sdk_instance)
+    >>> success = await initializer.init()
+    {!--< /tips >!--}
+    """
+    def __init__(self: None, sdk_instance: Any) -> ...:
+        """
+        初始化协调器
+        
+        :param sdk_instance: SDK 实例
+        """
+        ...
+    async def init(self: object) -> bool:
+        """
+        初始化所有模块和适配器
+        
+        执行步骤:
+        1. 从 PyPI 包加载适配器
+        2. 从 PyPI 包加载模块
+        3. 注册适配器
+        4. 注册模块
+        5. 初始化模块
+        
+        :return: bool 初始化是否成功
+        
+        :raises ImportError: 当加载失败时抛出
+        """
+        ...