ErisPulse 2.3.2__py3-none-any.whl → 2.3.3__py3-none-any.whl

ErisPulse/Core/module.pyi

@@ -0,0 +1,178 @@
+# type: ignore
+#
+# Auto-generated type stub for module.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse 模块系统
+
+提供标准化的模块注册、加载和管理功能，与适配器系统保持一致的设计模式
+"""
+
+import asyncio
+from typing import Any, Dict, List, Type, Optional
+from .logger import logger
+from .config import config
+from .Bases import BaseModule
+from .lifecycle import lifecycle
+
+class ModuleManager:
+    """
+    模块管理器
+    
+    提供标准化的模块注册、加载和管理功能，模仿适配器管理器的模式
+    
+    {!--< tips >!--}
+    1. 使用register方法注册模块类
+    2. 使用load/unload方法加载/卸载模块
+    3. 通过get方法获取模块实例
+    {!--< /tips >!--}
+    """
+    def __init__(self: None) -> ...:
+        ...
+    def register(self: object, module_name: str, module_class: Type, module_info: Optional[Dict] = ...) -> bool:
+        """
+        注册模块类
+        
+        :param module_name: 模块名称
+        :param module_class: 模块类
+        :param module_info: 模块信息
+        :return: 是否注册成功
+        
+        :raises TypeError: 当模块类无效时抛出
+            
+        :example:
+        >>> module.register("MyModule", MyModuleClass)
+        """
+        ...
+    async def load(self: object, module_name: str) -> bool:
+        """
+        加载指定模块（标准化加载逻辑）
+        
+        :param module_name: 模块名称
+        :return: 是否加载成功
+            
+        :example:
+        >>> await module.load("MyModule")
+        """
+        ...
+    async def unload(self: object, module_name: str = ...) -> bool:
+        """
+        卸载指定模块或所有模块
+        
+        :param module_name: 模块名称，如果为None则卸载所有模块
+        :return: 是否卸载成功
+            
+        :example:
+        >>> await module.unload("MyModule")
+        >>> await module.unload()  # 卸载所有模块
+        """
+        ...
+    def get(self: object, module_name: str) -> Any:
+        """
+        获取模块实例
+        
+        :param module_name: 模块名称
+        :return: 模块实例或None
+            
+        :example:
+        >>> my_module = module.get("MyModule")
+        """
+        ...
+    def exists(self: object, module_name: str) -> bool:
+        """
+        检查模块是否存在（在配置中注册）
+        
+        :param module_name: [str] 模块名称
+        :return: [bool] 模块是否存在
+        """
+        ...
+    def is_loaded(self: object, module_name: str) -> bool:
+        """
+        检查模块是否已加载
+        
+        :param module_name: 模块名称
+        :return: 模块是否已加载
+            
+        :example:
+        >>> if module.is_loaded("MyModule"): ...
+        """
+        ...
+    def list_registered(self: object) -> List[str]:
+        """
+        列出所有已注册的模块
+        
+        :return: 模块名称列表
+            
+        :example:
+        >>> registered = module.list_registered()
+        """
+        ...
+    def list_loaded(self: object) -> List[str]:
+        """
+        列出所有已加载的模块
+        
+        :return: 模块名称列表
+            
+        :example:
+        >>> loaded = module.list_loaded()
+        """
+        ...
+    def _config_register(self: object, module_name: str, enabled: bool = ...) -> bool:
+        """
+        注册新模块信息
+        
+        :param module_name: [str] 模块名称
+        :param enabled: [bool] 是否启用模块
+        :return: [bool] 操作是否成功
+        """
+        ...
+    def is_enabled(self: object, module_name: str) -> bool:
+        """
+        检查模块是否启用
+        
+        :param module_name: [str] 模块名称
+        :return: [bool] 模块是否启用
+        """
+        ...
+    def enable(self: object, module_name: str) -> bool:
+        """
+        启用模块
+        
+        :param module_name: [str] 模块名称
+        :return: [bool] 操作是否成功
+        """
+        ...
+    def disable(self: object, module_name: str) -> bool:
+        """
+        禁用模块
+        
+        :param module_name: [str] 模块名称
+        :return: [bool] 操作是否成功
+        """
+        ...
+    def list_modules(self: object) -> Dict[(str, bool)]:
+        """
+        列出所有模块状态
+        
+        :return: [Dict[str, bool]] 模块状态字典
+        """
+        ...
+    def __getattr__(self: object, module_name: str) -> Any:
+        """
+        通过属性访问获取模块实例
+        
+        :param module_name: [str] 模块名称
+        :return: [Any] 模块实例
+        :raises AttributeError: 当模块不存在或未启用时
+        """
+        ...
+    def __contains__(self: object, module_name: str) -> bool:
+        """
+        检查模块是否存在且处于启用状态
+        
+        :param module_name: [str] 模块名称
+        :return: [bool] 模块是否存在且启用
+        """
+        ...

ErisPulse/Core/router.pyi

@@ -0,0 +1,120 @@
+# type: ignore
+#
+# Auto-generated type stub for router.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse 路由系统
+
+提供统一的HTTP和WebSocket路由管理，支持多适配器路由注册和生命周期管理。
+
+{!--< tips >!--}
+1. 适配器只需注册路由，无需自行管理服务器
+2. WebSocket支持自定义认证逻辑
+{!--< /tips >!--}
+"""
+
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+from fastapi.routing import APIRoute
+from typing import Dict, List, Optional, Callable, Any, Awaitable, Tuple
+from collections import defaultdict
+from .logger import logger
+from .lifecycle import lifecycle
+import asyncio
+from hypercorn.config import Config
+from hypercorn.asyncio import serve
+
+class RouterManager:
+    """
+    路由管理器
+    
+    {!--< tips >!--}
+    核心功能：
+    - HTTP/WebSocket路由注册
+    - 生命周期管理
+    - 统一错误处理
+    {!--< /tips >!--}
+    """
+    def __init__(self: None) -> ...:
+        """
+        初始化路由管理器
+        
+        {!--< tips >!--}
+        会自动创建FastAPI实例并设置核心路由
+        {!--< /tips >!--}
+        """
+        ...
+    def register_http_route(self: object, module_name: str, path: str, handler: Callable, methods: List[str] = ...) -> None:
+        """
+        注册HTTP路由
+        
+        :param module_name: str 模块名称
+        :param path: str 路由路径
+        :param handler: Callable 处理函数
+        :param methods: List[str] HTTP方法列表(默认["POST"])
+        
+        :raises ValueError: 当路径已注册时抛出
+        """
+        ...
+    def register_webhook(self: object, *args: ..., **kwargs: ...) -> None:
+        """
+        兼容性方法：注册HTTP路由（适配器旧接口）
+        """
+        ...
+    def unregister_http_route(self: object, module_name: str, path: str) -> bool:
+        """
+        取消注册HTTP路由
+        
+        :param module_name: 模块名称
+        :param path: 路由路径
+        
+        :return: Bool
+        """
+        ...
+    def register_websocket(self: object, module_name: str, path: str, handler: Callable[([WebSocket], Awaitable[Any])], auth_handler: Optional[Callable[([WebSocket], Awaitable[bool])]] = ...) -> None:
+        """
+        注册WebSocket路由
+        
+        :param module_name: str 模块名称
+        :param path: str WebSocket路径
+        :param handler: Callable[[WebSocket], Awaitable[Any]] 主处理函数
+        :param auth_handler: Optional[Callable[[WebSocket], Awaitable[bool]]] 认证函数
+        
+        :raises ValueError: 当路径已注册时抛出
+        """
+        ...
+    def unregister_websocket(self: object, module_name: str, path: str) -> bool:
+        ...
+    def get_app(self: object) -> FastAPI:
+        """
+        获取FastAPI应用实例
+        
+        :return: FastAPI应用实例
+        """
+        ...
+    async def start(self: object, host: str = ..., port: int = ..., ssl_certfile: Optional[str] = ..., ssl_keyfile: Optional[str] = ...) -> None:
+        """
+        启动路由服务器
+        
+        :param host: str 监听地址(默认"0.0.0.0")
+        :param port: int 监听端口(默认8000)
+        :param ssl_certfile: Optional[str] SSL证书路径
+        :param ssl_keyfile: Optional[str] SSL密钥路径
+        
+        :raises RuntimeError: 当服务器已在运行时抛出
+        """
+        ...
+    async def stop(self: object) -> None:
+        """
+        停止服务器
+        """
+        ...
+    def _format_display_url(self: object, url: str) -> str:
+        """
+        格式化URL显示，将回环地址转换为更友好的格式
+        
+        :param url: 原始URL
+        :return: 格式化后的URL
+        """
+        ...

ErisPulse/Core/storage.pyi

@@ -0,0 +1,273 @@
+# type: ignore
+#
+# Auto-generated type stub for storage.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse 存储管理模块
+
+提供键值存储、事务支持、快照和恢复功能，用于管理框架运行时数据。
+基于SQLite实现持久化存储，支持复杂数据类型和原子操作。
+
+支持两种数据库模式：
+1. 项目数据库（默认）：位于项目目录下的 config/config.db
+2. 全局数据库：位于包内的 ../data/config.db
+
+用户可通过在 config.toml 中配置以下选项来选择使用全局数据库：
+```toml
+[ErisPulse.storage]
+use_global_db = true
+```
+
+{!--< tips >!--}
+1. 支持JSON序列化存储复杂数据类型
+2. 提供事务支持确保数据一致性
+3. 自动快照功能防止数据丢失
+{!--< /tips >!--}
+"""
+
+import os
+import json
+import sqlite3
+import shutil
+import time
+from datetime import datetime
+from typing import List, Dict, Optional, Any, Tuple, Type
+
+class StorageManager:
+    """
+    存储管理器
+    
+    单例模式实现，提供键值存储的增删改查、事务和快照管理
+    
+    支持两种数据库模式：
+    1. 项目数据库（默认）：位于项目目录下的 config/config.db
+    2. 全局数据库：位于包内的 ../data/config.db
+    
+    用户可通过在 config.toml 中配置以下选项来选择使用全局数据库：
+    ```toml
+    [ErisPulse.storage]
+    use_global_db = true
+    ```
+    
+    {!--< tips >!--}
+    1. 使用get/set方法操作存储项
+    2. 使用transaction上下文管理事务
+    3. 使用snapshot/restore管理数据快照
+    {!--< /tips >!--}
+    """
+    def __new__(cls: object, *args: ..., **kwargs: ...) -> ...:
+        ...
+    def __init__(self: None) -> ...:
+        ...
+    def _ensure_directories(self: object) -> None:
+        """
+        确保必要的目录存在
+        """
+        ...
+    def get(self: object, key: str, default: Any = ...) -> Any:
+        """
+        获取存储项的值
+        
+        :param key: 存储项键名
+        :param default: 默认值(当键不存在时返回)
+        :return: 存储项的值
+        
+        :example:
+        >>> timeout = storage.get("network.timeout", 30)
+        >>> user_settings = storage.get("user.settings", {})
+        """
+        ...
+    def get_all_keys(self: object) -> List[str]:
+        """
+        获取所有存储项的键名
+        
+        :return: 键名列表
+        
+        :example:
+        >>> all_keys = storage.get_all_keys()
+        >>> print(f"共有 {len(all_keys)} 个存储项")
+        """
+        ...
+    def set(self: object, key: str, value: Any) -> bool:
+        """
+        设置存储项的值
+        
+        :param key: 存储项键名
+        :param value: 存储项的值
+        :return: 操作是否成功
+        
+        :example:
+        >>> storage.set("app.name", "MyApp")
+        >>> storage.set("user.settings", {"theme": "dark"})
+        """
+        ...
+    def set_multi(self: object, items: Dict[(str, Any)]) -> bool:
+        """
+        批量设置多个存储项
+        
+        :param items: 键值对字典
+        :return: 操作是否成功
+        
+        :example:
+        >>> storage.set_multi({
+        >>>     "app.name": "MyApp",
+        >>>     "app.version": "1.0.0",
+        >>>     "app.debug": True
+        >>> })
+        """
+        ...
+    def getConfig(self: object, key: str, default: Any = ...) -> Any:
+        """
+        获取模块/适配器配置项（委托给config模块）
+        :param key: 配置项的键(支持点分隔符如"module.sub.key")
+        :param default: 默认值
+        :return: 配置项的值
+        """
+        ...
+    def setConfig(self: object, key: str, value: Any) -> bool:
+        """
+        设置模块/适配器配置（委托给config模块）
+        :param key: 配置项键名(支持点分隔符如"module.sub.key")
+        :param value: 配置项值
+        :return: 操作是否成功
+        """
+        ...
+    def delete(self: object, key: str) -> bool:
+        """
+        删除存储项
+        
+        :param key: 存储项键名
+        :return: 操作是否成功
+        
+        :example:
+        >>> storage.delete("temp.session")
+        """
+        ...
+    def delete_multi(self: object, keys: List[str]) -> bool:
+        """
+        批量删除多个存储项
+        
+        :param keys: 键名列表
+        :return: 操作是否成功
+        
+        :example:
+        >>> storage.delete_multi(["temp.key1", "temp.key2"])
+        """
+        ...
+    def get_multi(self: object, keys: List[str]) -> Dict[(str, Any)]:
+        """
+        批量获取多个存储项的值
+        
+        :param keys: 键名列表
+        :return: 键值对字典
+        
+        :example:
+        >>> settings = storage.get_multi(["app.name", "app.version"])
+        """
+        ...
+    def transaction(self: object) -> StorageManager._Transaction:
+        """
+        创建事务上下文
+        
+        :return: 事务上下文管理器
+        
+        :example:
+        >>> with storage.transaction():
+        >>>     storage.set("key1", "value1")
+        >>>     storage.set("key2", "value2")
+        """
+        ...
+    def set_snapshot_interval(self: object, seconds: int) -> None:
+        """
+        设置自动快照间隔
+        
+        :param seconds: 间隔秒数
+        
+        :example:
+        >>> # 每30分钟自动快照
+        >>> storage.set_snapshot_interval(1800)
+        """
+        ...
+    def clear(self: object) -> bool:
+        """
+        清空所有存储项
+        
+        :return: 操作是否成功
+        
+        :example:
+        >>> storage.clear()  # 清空所有存储
+        """
+        ...
+    def __getattr__(self: object, key: str) -> Any:
+        """
+        通过属性访问存储项
+        
+        :param key: 存储项键名
+        :return: 存储项的值
+        
+        :raises AttributeError: 当存储项不存在时抛出
+            
+        :example:
+        >>> app_name = storage.app_name
+        """
+        ...
+    def __setattr__(self: object, key: str, value: Any) -> None:
+        """
+        通过属性设置存储项
+        
+        :param key: 存储项键名
+        :param value: 存储项的值
+            
+        :example:
+        >>> storage.app_name = "MyApp"
+        """
+        ...
+    def snapshot(self: object, name: Optional[str] = ...) -> str:
+        """
+        创建数据库快照
+        
+        :param name: 快照名称(可选)
+        :return: 快照文件路径
+        
+        :example:
+        >>> # 创建命名快照
+        >>> snapshot_path = storage.snapshot("before_update")
+        >>> # 创建时间戳快照
+        >>> snapshot_path = storage.snapshot()
+        """
+        ...
+    def restore(self: object, snapshot_name: str) -> bool:
+        """
+        从快照恢复数据库
+        
+        :param snapshot_name: 快照名称或路径
+        :return: 恢复是否成功
+        
+        :example:
+        >>> storage.restore("before_update")
+        """
+        ...
+    def list_snapshots(self: object) -> List[Tuple[(str, datetime, int)]]:
+        """
+        列出所有可用的快照
+        
+        :return: 快照信息列表(名称, 创建时间, 大小)
+        
+        :example:
+        >>> for name, date, size in storage.list_snapshots():
+        >>>     print(f"{name} - {date} ({size} bytes)")
+        """
+        ...
+    def delete_snapshot(self: object, snapshot_name: str) -> bool:
+        """
+        删除指定的快照
+        
+        :param snapshot_name: 快照名称
+        :return: 删除是否成功
+        
+        :example:
+        >>> storage.delete_snapshot("old_backup")
+        """
+        ...

ErisPulse/Core/ux.pyi

@@ -0,0 +1,94 @@
+# type: ignore
+#
+# Auto-generated type stub for ux.py
+# DO NOT EDIT MANUALLY - Generated by generate-type-stubs.py
+#
+
+"""
+ErisPulse UX优化模块
+
+提供更友好的初始化和API接口，简化常用操作
+"""
+
+import json
+import asyncio
+from typing import List, Dict
+from pathlib import Path
+from rich.console import Console
+from rich.table import Table
+from rich.panel import Panel
+from rich.prompt import Confirm
+
+class UXManager:
+    """
+    UX优化管理器
+    
+    提供用户友好的界面和简化操作
+    """
+    def __init__(self: None) -> ...:
+        ...
+    async def _fetch_available_adapters(self: object) -> Dict[(str, str)]:
+        """
+        从云端获取可用适配器列表
+        
+        :return: 适配器名称到描述的映射
+        """
+        ...
+    def welcome(self: object, version: str = ...) -> None:
+        """
+        显示欢迎信息
+        
+        :param version: 框架版本号
+        """
+        ...
+    def show_status(self: object) -> None:
+        """
+        显示系统状态概览
+        """
+        ...
+    def list_modules(self: object, detailed: bool = ...) -> None:
+        """
+        列出所有模块状态
+        
+        :param detailed: 是否显示详细信息
+        """
+        ...
+    def list_adapters(self: object, detailed: bool = ...) -> None:
+        """
+        列出所有适配器状态
+        
+        :param detailed: 是否显示详细信息
+        """
+        ...
+    def init_project(self: object, project_name: str, adapter_list: List[str] = ...) -> bool:
+        """
+        初始化新项目
+        
+        :param project_name: 项目名称
+        :param adapter_list: 需要初始化的适配器列表
+        :return: 是否初始化成功
+        """
+        ...
+    def interactive_init(self: object, project_name: str = ..., force: bool = ...) -> bool:
+        """
+        交互式初始化项目，包括项目创建和配置设置
+        
+        :param project_name: 项目名称，可为None
+        :param force: 是否强制覆盖现有配置
+        :return: 是否初始化成功
+        """
+        ...
+    def _configure_adapters_interactive_sync(self: object, project_path: str = ...) -> None:
+        """
+        交互式配置适配器的同步版本，从云端获取适配器列表
+        
+        :param project_path: 项目路径，用于加载项目特定的配置
+        """
+        ...
+    async def _configure_adapters_interactive(self: object, project_path: str = ...) -> None:
+        """
+        交互式配置适配器，从云端获取适配器列表
+        
+        :param project_path: 项目路径，用于加载项目特定的配置
+        """
+        ...

ErisPulse/__init__.py

@@ -17,9 +17,13 @@ import importlib
 import asyncio
 import inspect
 import importlib.metadata
-from typing import Dict, List, Tuple, Type, Any
+from typing import Dict, List, Tuple, Type, Any, TYPE_CHECKING
 from pathlib import Path
 
+# 类型检查时导入 Protocol，避免运行时循环导入
+if TYPE_CHECKING:
+    from .sdk_protocol import SDKProtocol
+
 # BaseModules: SDK核心模块
 # 事件处理模块
 from .Core import Event
@@ -37,7 +41,7 @@ from .Core import router, adapter_server
 from .Core import ux, UXManager
 
 # SDK统一对外接口
-sdk = types.ModuleType('sdk')
+sdk: 'SDKProtocol' = types.ModuleType('sdk')  # type: ignore[assignment]
 
 try:
     __version__ = importlib.metadata.version('ErisPulse')
@@ -1182,7 +1186,7 @@ async def restart() -> bool:
     logger.info("[Reload] 重新加载完成")
     return True
 
-async def run() -> None:
+async def run(keep_running: bool = True) -> None:
     """
     无头模式运行ErisPulse
     
@@ -1197,8 +1201,9 @@ async def run() -> None:
         
         await adapter.startup()
         
-        # 保持程序运行
-        await asyncio.Event().wait()
+        if keep_running:
+            # 保持程序运行
+            await asyncio.Event().wait()
     except Exception as e:
         logger.error(e)
     finally: