File size: 1,004 Bytes
bb99156 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import os
import sys
from pathlib import Path
def ensure_boltz_importable() -> None:
candidates: list[Path] = []
if "BOLTZ_SRC_DIR" in os.environ:
env_override = os.environ["BOLTZ_SRC_DIR"]
if len(env_override.strip()) > 0:
candidates.append(Path(env_override.strip()))
module_dir = Path(__file__).resolve().parent
for parent in [module_dir, *module_dir.parents]:
candidates.append(parent / "boltz" / "src")
cwd_dir = Path.cwd().resolve()
for parent in [cwd_dir, *cwd_dir.parents]:
candidates.append(parent / "boltz" / "src")
for candidate in candidates:
if candidate.exists():
candidate_str = str(candidate)
if candidate_str not in sys.path:
sys.path.insert(0, candidate_str)
return
assert False, (
"Could not locate cloned Boltz source directory ('boltz/src'). "
"Set BOLTZ_SRC_DIR explicitly if needed."
)
|