82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import pathlib
|
|
import shlex
|
|
from typing import Iterable, Optional, Sequence, Tuple
|
|
|
|
|
|
def package_root(pkg: str) -> pathlib.Path:
|
|
spec = importlib.util.find_spec(pkg)
|
|
if spec is None:
|
|
raise RuntimeError(f"package not found: {pkg}")
|
|
if not spec.submodule_search_locations:
|
|
raise RuntimeError(f"package has no package root: {pkg}")
|
|
return pathlib.Path(next(iter(spec.submodule_search_locations))).resolve()
|
|
|
|
|
|
def ensure_file(path: pathlib.Path) -> pathlib.Path:
|
|
if not path.is_file():
|
|
raise FileNotFoundError(str(path))
|
|
return path
|
|
|
|
|
|
def ensure_dir(path: pathlib.Path) -> pathlib.Path:
|
|
if not path.is_dir():
|
|
raise FileNotFoundError(str(path))
|
|
return path
|
|
|
|
|
|
def replace_once(path: pathlib.Path,
|
|
old: str,
|
|
new: str,
|
|
*,
|
|
required: bool = True,
|
|
already_contains: Optional[str] = None) -> bool:
|
|
path = ensure_file(path)
|
|
text = path.read_text()
|
|
marker = already_contains if already_contains is not None else new
|
|
if marker in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
if old not in text:
|
|
msg = f"anchor not found in {path}: {old[:120]!r}"
|
|
if required:
|
|
raise RuntimeError(msg)
|
|
print(f"[warn] {msg}")
|
|
return False
|
|
path.write_text(text.replace(old, new, 1))
|
|
print(f"[ok] patched: {path}")
|
|
return True
|
|
|
|
|
|
def replace_one_of(path: pathlib.Path,
|
|
replacements: Sequence[Tuple[str, str]],
|
|
*,
|
|
required: bool = True,
|
|
already_contains: Optional[str] = None) -> bool:
|
|
path = ensure_file(path)
|
|
text = path.read_text()
|
|
if already_contains is not None and already_contains in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
for _, new in replacements:
|
|
if new in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
for old, new in replacements:
|
|
if old in text:
|
|
path.write_text(text.replace(old, new, 1))
|
|
print(f"[ok] patched: {path}")
|
|
return True
|
|
anchors = ", ".join(repr(old[:80]) for old, _ in replacements)
|
|
msg = f"anchor not found in {path}; tried: {anchors}"
|
|
if required:
|
|
raise RuntimeError(msg)
|
|
print(f"[warn] {msg}")
|
|
return False
|
|
|
|
|
|
def shell_env_line(name: str, value: pathlib.Path) -> str:
|
|
return f"{name}={shlex.quote(str(value))}"
|