Jerry
109 lines
2.9 KiB
Python
Executable File
109 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Bootstrap runtime package in a persistent Python REPL/kernel."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
PACKAGE_SPEC = "simplecadapi==2.0.8"
|
|
CASES_MODULE = "simplecad_self_evolve_cases"
|
|
|
|
|
|
def _script_dir() -> Path:
|
|
return Path(__file__).resolve().parent
|
|
|
|
|
|
def _cases_root() -> Path:
|
|
return _script_dir().parent / "cases"
|
|
|
|
|
|
def _activate_cases_path() -> None:
|
|
cases_root = str(_cases_root())
|
|
os.environ["SIMPLECAD_CASES_ROOT"] = cases_root
|
|
os.environ["SIMPLECAD_CASES_MODULE"] = CASES_MODULE
|
|
|
|
current_pythonpath = os.environ.get("PYTHONPATH", "")
|
|
parts = [item for item in current_pythonpath.split(os.pathsep) if item]
|
|
if cases_root not in parts:
|
|
parts.insert(0, cases_root)
|
|
os.environ["PYTHONPATH"] = os.pathsep.join(parts)
|
|
|
|
if cases_root not in sys.path:
|
|
sys.path.insert(0, cases_root)
|
|
|
|
|
|
def _install(with_jupyter: bool = False) -> None:
|
|
cmd = [str(_script_dir() / "install.sh")]
|
|
if with_jupyter:
|
|
cmd.append("--with-jupyter")
|
|
env = os.environ.copy()
|
|
env.setdefault("PYTHON_BIN", sys.executable)
|
|
subprocess.run(cmd, check=True, env=env)
|
|
|
|
|
|
def _has_simplecadapi() -> bool:
|
|
try:
|
|
import simplecadapi # noqa: F401
|
|
except Exception:
|
|
return False
|
|
return True
|
|
|
|
|
|
def _has_jupyter_deps() -> bool:
|
|
try:
|
|
import jupyterlab # noqa: F401
|
|
import ipykernel # noqa: F401
|
|
except Exception:
|
|
return False
|
|
return True
|
|
|
|
|
|
def activate(check: bool = False, with_jupyter: bool = False) -> int:
|
|
skill_root = _script_dir().parent
|
|
os.environ["SIMPLECAD_SKILL_ROOT"] = str(skill_root)
|
|
_activate_cases_path()
|
|
|
|
if not _has_simplecadapi():
|
|
print(f"Installing runtime package {PACKAGE_SPEC} ...")
|
|
_install(with_jupyter=with_jupyter)
|
|
|
|
if with_jupyter and not _has_jupyter_deps():
|
|
print("Installing Jupyter dependencies ...")
|
|
_install(with_jupyter=True)
|
|
|
|
if check and not _has_simplecadapi():
|
|
print("Runtime install failed: cannot import simplecadapi", file=sys.stderr)
|
|
return 1
|
|
|
|
print(f"Skill runtime activated from site-packages: {PACKAGE_SPEC}")
|
|
print(f"Skill cases module ready: {CASES_MODULE}.evolve")
|
|
print("You can now run: import simplecadapi as scad")
|
|
return 0
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(
|
|
description="Bootstrap SimpleCAD runtime in persistent Python process"
|
|
)
|
|
parser.add_argument(
|
|
"--check",
|
|
action="store_true",
|
|
help="Validate that import simplecadapi succeeds after bootstrap",
|
|
)
|
|
parser.add_argument(
|
|
"--with-jupyter",
|
|
action="store_true",
|
|
help="Also ensure jupyterlab and ipykernel are installed",
|
|
)
|
|
args = parser.parse_args()
|
|
raise SystemExit(activate(check=args.check, with_jupyter=args.with_jupyter))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|