py-wlcli 0.8.0__py3-none-any.whl

wl_cli/commands/self/services/installation_manager.py

@@ -0,0 +1,187 @@
+"""
+Installation manager for self update command.
+"""
+
+import os
+import shutil
+import subprocess
+import sys
+
+from ....utils.platform_adapter import PlatformAdapter
+
+
+class InstallationManager:
+    """Manager for handling installation-related functionality."""
+
+    def prepare_environment(self) -> tuple[str, dict]:
+        """
+        Prepare environment variables and find uv executable.
+        """
+
+        env = os.environ.copy()
+
+        if PlatformAdapter.is_windows():
+            env["PYTHONIOENCODING"] = "utf-8"
+            env["PYTHONLEGACYWINDOWSFSENCODING"] = "1"
+
+        uv_path = shutil.which("uv")
+        if uv_path is None:
+            raise FileNotFoundError("uv command not found in PATH")
+
+        return uv_path, env
+
+    def uninstall_existing(self, uv_path: str, env: dict) -> None:
+        """
+        Uninstall existing wl_cli (ignore failure if not installed).
+        """
+        from wl_cli.utils.logging import log_info_pair
+
+        log_info_pair("Uninstalling existing wl_cli...", "正在卸载现有wl_cli...")
+
+        uninstall_result = subprocess.run(
+            [uv_path, "tool", "uninstall", "py_wlcli"],
+            check=False,
+            capture_output=True,
+            text=True,
+            encoding="utf-8" if PlatformAdapter.is_windows() else None,
+            env=env,
+        )
+
+        if uninstall_result.returncode != 0:
+            log_info_pair(
+                "Note: wl_cli was not installed or failed to uninstall. Continuing with installation.",
+                "注意：wl_cli未安装或卸载失败。继续安装。",
+            )
+        else:
+            log_info_pair(
+                "Successfully uninstalled existing wl_cli.", "成功卸载现有wl_cli。"
+            )
+
+    def is_file_in_use_error(self, error_msg: str) -> bool:
+        """
+        Check if the error message indicates a file in use error on Windows.
+        """
+        return any(
+            keyword in error_msg
+            for keyword in [
+                "Access is denied",
+                "无法访问文件",
+                "The process cannot access the file",
+            ]
+        )
+
+    def detect_installation_method(self) -> str:
+        """
+        Detect how the current wl command is installed.
+
+        Returns:
+            str: Installation method - "uv_tool", "uv_pip", "pip", or "unknown"
+        """
+        from wl_cli.utils.logging import log_info_pair
+        from wl_cli.utils.uv_tool import is_running_in_uv_tool
+
+        if is_running_in_uv_tool():
+            log_info_pair(
+                "Detected installation method: uv tool", "检测到安装方式: uv tool"
+            )
+            return "uv_tool"
+
+        try:
+            import importlib.metadata
+
+            dist = importlib.metadata.distribution("py_wlcli")
+            log_info_pair(f"Detected distribution: {dist}", f"检测到发行版: {dist}")
+
+            log_info_pair(
+                "Detected installation method: uv pip", "检测到安装方式: uv pip"
+            )
+            return "uv_pip"
+        except (importlib.metadata.PackageNotFoundError, Exception) as e:
+            log_info_pair(
+                f"Warning: Could not detect installation method: {e}",
+                f"警告: 无法检测安装方式: {e}",
+            )
+            return "unknown"
+
+    def uninstall(self, uv_path: str, env: dict) -> None:
+        """
+        Uninstall wl command using the appropriate method.
+
+        Args:
+            uv_path (str): Path to the uv executable.
+            env (dict): Environment variables to use for uninstallation.
+        """
+        from wl_cli.utils.logging import log_info, log_info_pair
+
+        install_method = self.detect_installation_method()
+
+        try:
+            if install_method == "uv_tool":
+                cmd = [uv_path, "tool", "uninstall", "py_wlcli"]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "uv_pip":
+                cmd = [uv_path, "pip", "uninstall", "py_wlcli"]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "pip":
+                cmd = [sys.executable, "-m", "pip", "uninstall", "-y", "py_wlcli"]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            else:
+                # Try multiple uninstall methods as fallback
+                log_info_pair(
+                    "Trying multiple uninstall methods as fallback...",
+                    "尝试多种卸载方法作为回退...",
+                )
+
+                # Try uv tool uninstall
+                tool_result = subprocess.run(
+                    [uv_path, "tool", "uninstall", "py_wlcli"],
+                    check=False,
+                    capture_output=True,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+
+                # Try pip uninstall
+                pip_result = subprocess.run(
+                    [uv_path, "pip", "uninstall", "py_wlcli"],
+                    check=False,
+                    capture_output=True,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+
+                if tool_result.returncode != 0 and pip_result.returncode != 0:
+                    raise Exception(
+                        f"Failed to uninstall: uv tool error: {tool_result.stderr}, pip error: {pip_result.stderr}"
+                    )
+
+            log_info_pair("Successfully uninstalled wl command!", "成功卸载wl命令！")
+        except subprocess.CalledProcessError as e:
+            log_info_pair(f"Failed to uninstall: {e}", f"卸载失败: {e}")
+            raise

wl_cli/commands/self/services/local_installer.py

@@ -0,0 +1,165 @@
+"""
+Local installer for self update command.
+"""
+
+import subprocess
+
+from wl_cli.utils.logging import log_info, log_info_pair
+
+from ....utils.platform_adapter import PlatformAdapter
+from .installation_manager import InstallationManager
+from .windows_update import WindowsUpdate
+
+
+class LocalInstaller:
+    """Installer for local installation of the wl command."""
+
+    def install(self, uv_path: str, env: dict) -> None:
+        """
+        Install the current local code (replicates reinstall.bat functionality).
+
+        Args:
+            uv_path (str): Path to the uv executable.
+            env (dict): Environment variables to use for the installation.
+        """
+        log_info_pair("Installing local wl_cli...", "正在安装本地wl_cli...")
+
+        installation_manager = InstallationManager()
+        installation_manager.uninstall_existing(uv_path, env)
+
+        log_info_pair(
+            "Installing current directory in editable mode...",
+            "正在以可编辑模式安装当前目录...",
+        )
+
+        install_cmd = [uv_path, "tool", "install", "--editable", "."]
+        log_info(f"Running: {' '.join(install_cmd)}")
+
+        try:
+            subprocess.run(
+                install_cmd,
+                check=True,
+                capture_output=False,
+                text=True,
+                encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                env=env,
+            )
+            log_info_pair(
+                "Local installation completed successfully!", "本地安装成功完成！"
+            )
+        except subprocess.CalledProcessError as e:
+            log_info_pair(f"Local installation failed: {e}", f"本地安装失败: {e}")
+            self._try_alternative_installation(uv_path, env)
+
+    def _try_alternative_installation(self, uv_path: str, env: dict) -> None:
+        """
+        Try alternative installation method if primary fails.
+
+        Args:
+            uv_path (str): Path to the uv executable.
+            env (dict): Environment variables to use for the installation.
+        """
+        log_info_pair(
+            "Trying alternative installation method...", "尝试备选安装方法..."
+        )
+
+        alternative_cmd = [
+            uv_path,
+            "tool",
+            "install",
+            "--editable",
+            ".",
+            "--python-preference",
+            "only-system",
+        ]
+        log_info(f"Running: {' '.join(alternative_cmd)}")
+
+        try:
+            subprocess.run(
+                alternative_cmd,
+                check=True,
+                capture_output=False,
+                text=True,
+                encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                env=env,
+            )
+            log_info_pair(
+                "Local installation completed successfully with alternative method!",
+                "使用备选方法本地安装成功完成！",
+            )
+        except subprocess.CalledProcessError as alt_e:
+            log_info_pair(
+                f"Alternative installation also failed: {alt_e}",
+                f"备选安装方法也失败: {alt_e}",
+            )
+            raise
+
+    def run_installation_attempt(
+        self,
+        uv_path: str,
+        env: dict,
+        timeout_duration: int,
+        use_alternative: bool = False,
+    ) -> bool:
+        """
+        Run a single installation attempt (primary or alternative).
+        Returns True if installation succeeded, False otherwise.
+        """
+        if use_alternative:
+            cmd = [
+                uv_path,
+                "tool",
+                "install",
+                "--editable",
+                ".",
+                "--python-preference",
+                "only-system",
+            ]
+            log_info_pair(
+                "Trying alternative installation method...", "尝试备选安装方法..."
+            )
+        else:
+            cmd = [uv_path, "tool", "install", "--editable", "."]
+
+        log_info(f"Running: {' '.join(cmd)}")
+        try:
+            subprocess.run(
+                cmd,
+                check=True,
+                capture_output=False,
+                text=True,
+                encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                env=env,
+                timeout=timeout_duration,
+            )
+            return True
+        except subprocess.TimeoutExpired:
+            log_info_pair(
+                f"{'Alternative' if use_alternative else 'Primary'} installation timed out after {timeout_duration} seconds.",
+                f"{'备选' if use_alternative else '主'}安装方法在{timeout_duration}秒后超时。",
+            )
+            return False
+        except subprocess.CalledProcessError as e:
+            log_info_pair(
+                f"{'Alternative' if use_alternative else 'Primary'} installation failed: {e}",
+                f"{'备选' if use_alternative else '主'}安装方法失败: {e}",
+            )
+            installation_manager = InstallationManager()
+            if (
+                PlatformAdapter.is_windows()
+                and installation_manager.is_file_in_use_error(str(e))
+            ):
+                windows_update = WindowsUpdate()
+                windows_update.handle_access_error(uv_path)
+                return False
+            if use_alternative:
+                raise
+            return False
+        except Exception as e:
+            log_info_pair(
+                f"Unexpected error during {'alternative' if use_alternative else 'primary'} installation: {e}",
+                f"{'备选' if use_alternative else '主'}安装过程中发生意外错误: {e}",
+            )
+            if use_alternative:
+                raise
+            return False

wl_cli/commands/self/services/pypi_updater.py

@@ -0,0 +1,278 @@
+"""
+PyPI updater for self update command.
+"""
+
+import subprocess
+import sys
+
+from wl_cli.commands.self.common.version_cache import VersionCache
+from wl_cli.commands.self.services.installation_manager import (
+    InstallationManager,
+)
+
+from ....utils.platform_adapter import PlatformAdapter
+
+
+class PypiUpdater:
+    """Updater for updating the wl command from PyPI."""
+
+    def __init__(self) -> None:
+        """Initialize PypiUpdater with cache."""
+        self._cache_timeout: int = 300
+        # When _cache_timeout is changed from outside, the VersionCache
+        # still uses the original default, but that's OK since cache entry
+        # expiration is checked per entry when accessed
+        self._version_cache = VersionCache(default_timeout=self._cache_timeout)
+        # Allow overriding cache timeout from outside for configuration
+
+    def update(self, uv_path: str, env: dict) -> None:
+        """
+        Update wl command from PyPI using the same installation method as current.
+
+        Args:
+            uv_path (str): Path to the uv executable.
+            env (dict): Environment variables to use for the update.
+        """
+
+        from wl_cli.utils.logging import log_info, log_info_pair
+        from wl_cli.utils.version.pypi_version_checker import PyPIVersionChecker
+        from wl_cli.utils.version.version_comparator import VersionComparator
+        from wl_cli.utils.version.version_detectors import VersionDetector
+
+        installation_manager = InstallationManager()
+        install_method = installation_manager.detect_installation_method()
+
+        detector = VersionDetector()
+        checker = PyPIVersionChecker()
+        comparator = VersionComparator()
+
+        current_version = detector.get_current_version(comparator)
+        log_info_pair(
+            f"Current local version: {current_version}",
+            f"当前本地版本: {current_version}",
+        )
+
+        # Get PyPI version with caching
+        def fetch_pypi_version():
+            return checker._get_pypi_version()
+
+        pypi_version = self._version_cache.get_or_set(
+            "pypi_version", fetch_pypi_version
+        )
+        if pypi_version is None:
+            log_info_pair(
+                "Failed to get PyPI version. Skipping update.",
+                "无法获取PyPI版本。跳过更新。",
+            )
+            return
+        else:
+            if self._version_cache.get("pypi_version") is not None:
+                log_info_pair(
+                    "Using cached PyPI version information", "使用缓存的PyPI版本信息"
+                )
+
+        if pypi_version is None:
+            log_info_pair(
+                "Failed to get PyPI version. Skipping update.",
+                "无法获取PyPI版本。跳过更新。",
+            )
+            return
+
+        log_info_pair(
+            f"Latest PyPI version: {pypi_version}", f"最新PyPI版本: {pypi_version}"
+        )
+
+        if not comparator.is_version_greater(pypi_version, current_version):
+            log_info_pair(
+                f"Local version {current_version} is already up to date with PyPI version {pypi_version}.",
+                f"本地版本 {current_version} 已经与PyPI版本 {pypi_version} 保持最新。",
+            )
+            return
+
+        log_info_pair(
+            f"Updating from version {current_version} to {pypi_version}...",
+            f"正在从版本 {current_version} 更新到 {pypi_version}...",
+        )
+
+        try:
+            if install_method == "uv_tool":
+                cmd = [uv_path, "tool", "install", "py_wlcli"]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "uv_pip":
+                cmd = [
+                    uv_path,
+                    "pip",
+                    "install",
+                    "--upgrade",
+                    "--system",
+                    "py_wlcli",
+                ]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "pip":
+                cmd = [
+                    sys.executable,
+                    "-m",
+                    "pip",
+                    "install",
+                    "--upgrade",
+                    "py_wlcli",
+                ]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            else:
+                log_info_pair(
+                    f"Unknown installation method: {install_method}, falling back to uv tool install.",
+                    f"未知安装方法: {install_method}，回退到uv tool安装。",
+                )
+                cmd = [uv_path, "tool", "install", "py_wlcli"]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+
+            log_info_pair("Successfully updated from PyPI!", "成功从PyPI更新！")
+        except subprocess.CalledProcessError as e:
+            log_info_pair(f"Failed to update from PyPI: {e}", f"从PyPI更新失败: {e}")
+            raise
+
+    def install_version(
+        self, uv_path: str, env: dict, version: str | None = None
+    ) -> None:
+        """
+        Install a specific version of wl command.
+
+        Args:
+            uv_path (str): Path to the uv executable.
+            env (dict): Environment variables to use for the installation.
+            version (str): Version to install (e.g., "0.2.48" or "latest").
+        """
+        from wl_cli.utils.logging import log_info, log_info_pair
+
+        installation_manager = InstallationManager()
+        install_method = installation_manager.detect_installation_method()
+
+        try:
+            if version == "latest" or version is None:
+                # Install latest version
+                package_spec = "py_wlcli"
+            else:
+                # Install specific version
+                package_spec = f"py_wlcli=={version}"
+
+            if install_method == "uv_tool":
+                cmd = [uv_path, "tool", "install", package_spec]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "uv_pip":
+                cmd = [
+                    uv_path,
+                    "pip",
+                    "install",
+                    "--system",
+                    package_spec,
+                ]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            elif install_method == "pip":
+                cmd = [
+                    sys.executable,
+                    "-m",
+                    "pip",
+                    "install",
+                    package_spec,
+                ]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+            else:
+                log_info_pair(
+                    f"Unknown installation method: {install_method}, falling back to uv tool install.",
+                    f"未知安装方法: {install_method}，回退到uv tool安装。",
+                )
+                cmd = [uv_path, "tool", "install", package_spec]
+                log_info(f"Running: {' '.join(cmd)}")
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    capture_output=False,
+                    text=True,
+                    encoding="utf-8" if PlatformAdapter.is_windows() else None,
+                    env=env,
+                )
+
+            log_info_pair(
+                f"Successfully installed {package_spec}!", f"成功安装 {package_spec}！"
+            )
+        except subprocess.CalledProcessError as e:
+            log_info_pair(f"Failed to install version: {e}", f"安装版本失败: {e}")
+            raise
+
+    def get_latest_version(self) -> str:
+        """
+        Get the latest version from PyPI.
+
+        Returns:
+            str: Latest version string from PyPI.
+        """
+        from wl_cli.utils.version.pypi_version_checker import PyPIVersionChecker
+
+        # Get PyPI version with caching
+        def fetch_pypi_version():
+            checker = PyPIVersionChecker()
+            return checker._get_pypi_version()
+
+        version: str | None = self._version_cache.get_or_set(
+            "pypi_version", fetch_pypi_version
+        )
+
+        if version is None:
+            raise Exception("Failed to get latest version from PyPI")
+        return version