You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.6 KiB
79 lines
2.6 KiB
import sys
|
|
from typing import Any, Dict, Iterable, Iterator, List, Tuple
|
|
|
|
if sys.version_info >= (3, 11):
|
|
import tomllib
|
|
else:
|
|
from pip._vendor import tomli as tomllib
|
|
|
|
from pip._vendor.dependency_groups import DependencyGroupResolver
|
|
|
|
from pip._internal.exceptions import InstallationError
|
|
|
|
|
|
def parse_dependency_groups(groups: List[Tuple[str, str]]) -> List[str]:
|
|
"""
|
|
Parse dependency groups data as provided via the CLI, in a `[path:]group` syntax.
|
|
|
|
Raises InstallationErrors if anything goes wrong.
|
|
"""
|
|
resolvers = _build_resolvers(path for (path, _) in groups)
|
|
return list(_resolve_all_groups(resolvers, groups))
|
|
|
|
|
|
def _resolve_all_groups(
|
|
resolvers: Dict[str, DependencyGroupResolver], groups: List[Tuple[str, str]]
|
|
) -> Iterator[str]:
|
|
"""
|
|
Run all resolution, converting any error from `DependencyGroupResolver` into
|
|
an InstallationError.
|
|
"""
|
|
for path, groupname in groups:
|
|
resolver = resolvers[path]
|
|
try:
|
|
yield from (str(req) for req in resolver.resolve(groupname))
|
|
except (ValueError, TypeError, LookupError) as e:
|
|
raise InstallationError(
|
|
f"[dependency-groups] resolution failed for '{groupname}' "
|
|
f"from '{path}': {e}"
|
|
) from e
|
|
|
|
|
|
def _build_resolvers(paths: Iterable[str]) -> Dict[str, Any]:
|
|
resolvers = {}
|
|
for path in paths:
|
|
if path in resolvers:
|
|
continue
|
|
|
|
pyproject = _load_pyproject(path)
|
|
if "dependency-groups" not in pyproject:
|
|
raise InstallationError(
|
|
f"[dependency-groups] table was missing from '{path}'. "
|
|
"Cannot resolve '--group' option."
|
|
)
|
|
raw_dependency_groups = pyproject["dependency-groups"]
|
|
if not isinstance(raw_dependency_groups, dict):
|
|
raise InstallationError(
|
|
f"[dependency-groups] table was malformed in {path}. "
|
|
"Cannot resolve '--group' option."
|
|
)
|
|
|
|
resolvers[path] = DependencyGroupResolver(raw_dependency_groups)
|
|
return resolvers
|
|
|
|
|
|
def _load_pyproject(path: str) -> Dict[str, Any]:
|
|
"""
|
|
This helper loads a pyproject.toml as TOML.
|
|
|
|
It raises an InstallationError if the operation fails.
|
|
"""
|
|
try:
|
|
with open(path, "rb") as fp:
|
|
return tomllib.load(fp)
|
|
except FileNotFoundError:
|
|
raise InstallationError(f"{path} not found. Cannot resolve '--group' option.")
|
|
except tomllib.TOMLDecodeError as e:
|
|
raise InstallationError(f"Error parsing {path}: {e}") from e
|
|
except OSError as e:
|
|
raise InstallationError(f"Error reading {path}: {e}") from e
|
|
|