-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathlocal_deps.py
More file actions
67 lines (54 loc) · 1.99 KB
/
local_deps.py
File metadata and controls
67 lines (54 loc) · 1.99 KB
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#!/usr/bin/env python3
import importlib.metadata
import re
import sys
import tomllib
from pathlib import Path
def bootstrap_local_dependency(repo_name: str) -> None:
repo_root = Path(__file__).resolve().parent
candidate = repo_root.parent / repo_name
if not candidate.is_dir():
return
nested_src = candidate / repo_name
if (nested_src / repo_name / "__init__.py").is_file():
candidate_path = str(nested_src)
else:
candidate_path = str(candidate)
if candidate_path not in sys.path:
sys.path.insert(0, candidate_path)
def _version_key(version: str) -> tuple:
parts = []
for token in re.split(r"[.+-]", str(version or "").strip()):
if not token:
continue
if token.isdigit():
parts.append((0, int(token)))
else:
parts.append((1, token.lower()))
return tuple(parts)
def _local_repo_version(repo_name: str) -> str | None:
repo_root = Path(__file__).resolve().parent
pyproject_path = repo_root.parent / repo_name / "pyproject.toml"
if not pyproject_path.is_file():
return None
try:
with pyproject_path.open("rb") as handle:
payload = tomllib.load(handle)
return str(payload.get("project", {}).get("version") or "").strip() or None
except Exception:
return None
def dependency_version(repo_name: str) -> str | None:
local_version = _local_repo_version(repo_name)
if local_version:
return local_version
try:
return importlib.metadata.version(repo_name)
except importlib.metadata.PackageNotFoundError:
return None
def ensure_dependency_version(repo_name: str, minimum_version: str) -> str:
version = dependency_version(repo_name)
if version is None:
raise ModuleNotFoundError(f"{repo_name} is not installed")
if _version_key(version) < _version_key(minimum_version):
raise RuntimeError(f"{repo_name} {minimum_version}+ required, found {version}")
return version