Sorry, it my mistake

Author: STYLE-STARS

pip模块支持/__init__.py

@@ -1,8 +1,12 @@
-import sys
 import importlib
+import shutil
 import subprocess
-from tooldelta import Plugin, utils, fmts, plugin_entry
-from importlib.metadata import version, PackageNotFoundError
+import sys
+from importlib.metadata import PackageNotFoundError, version
+
+from tooldelta import Plugin, fmts, plugin_entry, utils
+
+PYPI_INDEX_URL = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
 
 
 def decode_bytes(data: bytes) -> str:
@@ -30,7 +34,7 @@ def __init__(self, frame):
         )
 
     # -------------------------  API  -----------------------------
-    def install(self, packages: list[str], upgrade=False):
+    def install(self, packages: list[str], upgrade: bool = False):
         if not packages:
             return
         pending: list[str] = []
@@ -48,33 +52,7 @@ def install(self, packages: list[str], upgrade=False):
             return
         packages = pending
 
-        pyexec = sys.executable
-        if "py" not in pyexec:
-            # 这不是 Python, 是 ToolDelta
-            install_opts = [
-                "pip",
-                "install",
-                "-i",
-                "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple",
-                "--target",
-                self.data_path,
-                *packages,
-            ]
-        else:
-            install_opts = [
-                sys.executable,
-                "-m",
-                "pip",
-                "install",
-                "-i",
-                "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple",
-                "--target",
-                self.data_path,
-                *packages,
-            ]
-        if upgrade:
-            install_opts.append("--upgrade")
-
+        install_opts = self._build_install_command(packages, upgrade)
         proc = subprocess.Popen(
             install_opts,
             stdout=subprocess.PIPE,
@@ -85,11 +63,73 @@ def install(self, packages: list[str], upgrade=False):
         self._readline_stderr(proc)
         returncode = proc.wait()
         if returncode != 0:
-            fmts.print_err("pip安装模块时出现错误")
+            fmts.print_err("模块安装时出现错误")
             raise SystemExit
         importlib.invalidate_caches()
         fmts.print_suc(f"模块 {', '.join(packages)} 安装成功")
 
+    def _build_install_command(
+        self, packages: list[str], upgrade: bool = False
+    ) -> list[str]:
+        """Build the command used to install Python packages."""
+        install_args = [
+            "install",
+            "--index-url",
+            PYPI_INDEX_URL,
+            "--target",
+            str(self.data_path),
+        ]
+        if upgrade:
+            install_args.append("--upgrade")
+        install_args.extend(packages)
+
+        if self._current_python_has_pip():
+            return [sys.executable, "-m", "pip", *install_args]
+
+        uv_exec = shutil.which("uv")
+        if uv_exec:
+            uv_args = ["pip", *install_args]
+            if self._can_run_current_python():
+                uv_args[2:2] = ["--python", sys.executable]
+            return [uv_exec, *uv_args]
+
+        pip_exec = shutil.which("pip")
+        if pip_exec:
+            return [pip_exec, *install_args]
+
+        fmts.print_err("未找到可用的 pip 或 uv 命令")
+        raise SystemExit
+
+    @staticmethod
+    def _can_run_current_python() -> bool:
+        """Return whether the current Python executable can be started."""
+        try:
+            proc = subprocess.run(
+                [sys.executable, "-c", "import sys"],
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                check=False,
+            )
+        except OSError:
+            return False
+        return proc.returncode == 0
+
+    @classmethod
+    def _current_python_has_pip(cls) -> bool:
+        """Return whether the current Python executable can run pip."""
+        if not cls._can_run_current_python():
+            return False
+        try:
+            proc = subprocess.run(
+                [sys.executable, "-m", "pip", "--version"],
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                check=False,
+            )
+        except OSError:
+            return False
+        return proc.returncode == 0
+
     def require(self, module_pip_name_and_imp_name: dict[str, str] | str | list[str]):
         """
         显式需求一个 Python 模块。