refactor(self-mgmt): adapt update/uninstall to system-python editable install

This commit is contained in:
Zhengshou Lai
2026-06-04 00:59:55 +08:00
parent 2d2ff45fca
commit da52571a9c
2 changed files with 44 additions and 71 deletions
+24 -31
View File
@@ -20,43 +20,33 @@ def _run(cmd: list[str], cwd: Path | None = None, check: bool = True) -> None:
raise click.Exit(result.returncode)
def _has_uv() -> bool:
return shutil.which("uv") is not None
def _install(root: Path) -> None:
venv_toolkit = root / ".venv" / "bin" / "mytoolkit"
user_local = Path.home() / ".local" / "bin" / "mytoolkit"
comp_dir = Path.home() / ".local" / "bin" / "completions"
"""Install or update using the current Python's pip (editable)."""
click.secho("Installing mytoolkit (editable)…", fg="cyan")
_run([sys.executable, "-m", "pip", "install", "-e", str(root), "--upgrade"])
# 1. Sync venv / install package
click.secho("Installing package…", fg="cyan")
if _has_uv():
_run(["uv", "sync"], cwd=root)
entry_point = shutil.which("mytoolkit")
if entry_point:
click.secho(f"Entry point: {entry_point}", fg="green")
else:
venv_pip = root / ".venv" / "bin" / "pip"
if not venv_pip.exists():
_run([sys.executable, "-m", "venv", str(root / ".venv")])
_run([str(venv_pip), "install", "-e", str(root)], cwd=root)
click.secho("Warning: mytoolkit not found in PATH after install", fg="yellow")
if not venv_toolkit.exists():
click.secho(f"error: missing {venv_toolkit}", fg="red", err=True)
raise click.Exit(1)
_install_completions()
# 2. Symlink to ~/.local/bin
click.secho(f"Linking {user_local}{venv_toolkit}", fg="cyan")
user_local.parent.mkdir(parents=True, exist_ok=True)
if user_local.exists() or user_local.is_symlink():
user_local.unlink()
user_local.symlink_to(venv_toolkit)
# 3. Install completions
def _install_completions() -> None:
venv_toolkit = shutil.which("mytoolkit")
if not venv_toolkit:
click.secho("Warning: cannot generate completions (mytoolkit not in PATH)", fg="yellow")
return
comp_dir = Path.home() / ".local" / "bin" / "completions"
comp_dir.mkdir(parents=True, exist_ok=True)
shell = os.environ.get("SHELL", "")
if "zsh" in shell:
comp_file = comp_dir / "_mytoolkit"
result = subprocess.run(
[str(venv_toolkit)],
[venv_toolkit],
env={**os.environ, "_MYTOOLKIT_COMPLETE": "zsh_source"},
capture_output=True,
text=True,
@@ -69,7 +59,7 @@ def _install(root: Path) -> None:
elif "bash" in shell:
comp_file = comp_dir / "mytoolkit.bash"
result = subprocess.run(
[str(venv_toolkit)],
[venv_toolkit],
env={**os.environ, "_MYTOOLKIT_COMPLETE": "bash_source"},
capture_output=True,
text=True,
@@ -82,9 +72,14 @@ def _install(root: Path) -> None:
def _uninstall() -> None:
user_local = Path.home() / ".local" / "bin" / "mytoolkit"
click.secho("Uninstalling mytoolkit package…", fg="cyan")
subprocess.run(
[sys.executable, "-m", "pip", "uninstall", "mytoolkit", "-y"],
check=False,
)
comp_dir = Path.home() / ".local" / "bin" / "completions"
for f in (comp_dir / "_mytoolkit", comp_dir / "mytoolkit.bash", user_local):
for f in (comp_dir / "_mytoolkit", comp_dir / "mytoolkit.bash"):
if f.exists() or f.is_symlink():
f.unlink()
click.secho(f"Removed {f}", fg="green")
@@ -105,5 +100,3 @@ def uninstall_cmd():
click.secho("Uninstalling mytoolkit…", fg="cyan")
_uninstall()
click.secho("mytoolkit removed.", fg="green")