ErisPulse 2.5.1__tar.gz → 2.5.2.dev0__tar.gz

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ErisPulse
-Version: 2.5.1
+Version: 2.5.2.dev0
 Summary: ErisPulse 是一个模块化、可扩展的异步 Python SDK 框架，主要用于构建高效、可维护的机器人应用程序。
 Project-URL: Homepage, https://github.com/ErisPulse/ErisPulse
 Project-URL: Documentation, https://www.erisdev.com#docs

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ErisPulse"
-version = "2.5.1"
+version = "2.5.2-dev.0"
 description = "ErisPulse 是一个模块化、可扩展的异步 Python SDK 框架，主要用于构建高效、可维护的机器人应用程序。"
 readme = "README.md"
 requires-python = ">=3.10"

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/src/ErisPulse/CLI/commands/run.py

@@ -168,11 +168,28 @@ class RunCommand(Command):
             "asyncio.run(sdk.run(keep_running=True))",
         ]
 
+        import signal
+
         try:
             while True:
                 process = subprocess.Popen(cmd)
+
+                # 转发 SIGTERM/SIGINT 到子进程，使 SDK 能优雅关闭
+                def _forward_signal(signum, frame):
+                    try:
+                        process.send_signal(signum)
+                    except Exception:
+                        pass
+
+                old_term = signal.signal(signal.SIGTERM, _forward_signal)
+                old_int = signal.signal(signal.SIGINT, _forward_signal)
+
                 process.wait()
 
+                # 恢复默认信号处理
+                signal.signal(signal.SIGTERM, old_term)
+                signal.signal(signal.SIGINT, old_int)
+
                 if process.returncode == self._RESTART_EXIT_CODE:
                     console.print(f"[info]{i18n.t('cli.run.restart_request')}[/]")
                     time.sleep(0.5)
@@ -193,6 +210,9 @@ class RunCommand(Command):
         if reload_mode:
             self._run_script_with_reload(script_path_abs)
         else:
+            script_dir = os.path.dirname(script_path_abs)
+            if script_dir not in sys.path:
+                sys.path.insert(0, script_dir)
             try:
                 runpy.run_path(script_path_abs, run_name="__main__")
             except SystemExit:

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/src/ErisPulse/CLI/utils/package_manager.py

@@ -673,18 +673,59 @@ class PackageManager:
            并将目标 Python 解析为当前虚拟环境的解释器，
            避免安装到全局环境。
 
+        操作成功后自动保存包快照到持久化目录，
+        供 Docker 容器重建后恢复使用。
+
         :param args: [List[str]] pip 子命令与参数，如 ["install", "--upgrade", pkg]
         :param description: [str] 展示给用户的操作描述
         :return: [bool] 执行成功返回 True
         """
         uv_cmd = self._get_uv_command()
+        success = False
         if uv_cmd:
             if self._execute_backend(uv_cmd + ["pip"], args, description, "uv"):
-                return True
-            console.print(f"[warning]{i18n.t('cli.package.uv_fallback_to_pip')}[/]")
+                success = True
+            else:
+                console.print(f"[warning]{i18n.t('cli.package.uv_fallback_to_pip')}[/]")
+
+        if not success:
+            pip_cmd = [self._get_target_python(), "-m", "pip"]
+            success = self._execute_backend(pip_cmd, args, description, "pip")
+
+        # 操作成功后保存包快照（非关键，失败不影响主流程）
+        if success:
+            self._save_package_snapshot()
+
+        return success
+
+    def _save_package_snapshot(self) -> None:
+        """
+        {!--< internal-use >!--}
+        将当前已安装的包列表保存到持久化目录。
+
+        快照文件用于 Docker 容器重建后的包恢复：
+        entrypoint 创建新 venv 时会读取 .pip-snapshot.txt
+        并尝试重新安装这些包。
 
-        pip_cmd = [self._get_target_python(), "-m", "pip"]
-        return self._execute_backend(pip_cmd, args, description, "pip")
+        仅在 config/ 目录存在时保存（即项目环境或 Docker 环境）。
+        """
+        try:
+            config_dir = os.path.join(os.getcwd(), "config")
+            if not os.path.isdir(config_dir):
+                return
+
+            result = subprocess.run(
+                [sys.executable, "-m", "pip", "freeze"],
+                capture_output=True,
+                text=True,
+                timeout=30,
+            )
+            if result.returncode == 0:
+                snapshot_path = os.path.join(config_dir, ".pip-snapshot.txt")
+                with open(snapshot_path, "w", encoding="utf-8") as f:
+                    f.write(result.stdout)
+        except Exception:
+            pass
 
     def _version_key(self, version: str) -> tuple:
         """

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/src/ErisPulse/CLI/utils/package_manager.pyi

@@ -243,6 +243,9 @@ class PackageManager:
            并将目标 Python 解析为当前虚拟环境的解释器，
            避免安装到全局环境。
         
+        操作成功后自动保存包快照到持久化目录，
+        供 Docker 容器重建后恢复使用。
+        
         :param args: [List[str]] pip 子命令与参数，如 ["install", "--upgrade", pkg]
         :param description: [str] 展示给用户的操作描述
         :return: [bool] 执行成功返回 True

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/src/ErisPulse/sdk.py

@@ -898,12 +898,37 @@ class SDK:
         """
         无头模式运行 ErisPulse
 
+        监听 SIGTERM/SIGINT 信号以实现优雅关闭：
+        收到信号后触发 shutdown_event，使 finally 块中的 uninit() 正常执行，
+        确保适配器断开、模块卸载、数据落盘等清理工作完成。
+
         :param keep_running: bool 是否保持运行
 
         :example:
         >>> await sdk.run(keep_running=True)
         """
+        import signal
+
+        shutdown_event = asyncio.Event()
+        loop = asyncio.get_running_loop()
+        _registered_signals: list[int] = []
+
+        def _request_shutdown() -> None:
+            """信号回调：设置关闭事件，使 run() 退出等待并进入清理流程"""
+            if not shutdown_event.is_set():
+                self.logger.info(i18n.t("core.sdk.run.shutdown_signal"))
+                shutdown_event.set()
+
         try:
+            # 注册信号处理器，使 Docker SIGTERM / Ctrl+C 能触发优雅关闭
+            for sig in (signal.SIGTERM, signal.SIGINT):
+                try:
+                    loop.add_signal_handler(sig, _request_shutdown)
+                    _registered_signals.append(sig)
+                except (NotImplementedError, RuntimeError):
+                    # Windows 不支持 add_signal_handler，跳过
+                    pass
+
             isInit = await self.init()
 
             if not isInit:
@@ -911,13 +936,19 @@ class SDK:
                 return
 
             if keep_running:
-                shutdown_event = asyncio.Event()
                 await shutdown_event.wait()
         except asyncio.CancelledError:
             self.logger.info(i18n.t("core.sdk.run.shutdown_signal"))
         except Exception as e:
             self.logger.error(e)
         finally:
+            # 清理信号处理器，避免重复触发
+            for sig in _registered_signals:
+                try:
+                    loop.remove_signal_handler(sig)
+                except Exception:
+                    pass
+
             if keep_running:
                 try:
                     await self.uninit()

{erispulse-2.5.1 → erispulse-2.5.2.dev0}/src/ErisPulse/sdk.pyi

@@ -142,6 +142,10 @@ class SDK:
         """
         无头模式运行 ErisPulse
         
+        监听 SIGTERM/SIGINT 信号以实现优雅关闭：
+        收到信号后触发 shutdown_event，使 finally 块中的 uninit() 正常执行，
+        确保适配器断开、模块卸载、数据落盘等清理工作完成。
+        
         :param keep_running: bool 是否保持运行
         
         :example: