from collections.abc import Iterable, Iterator from typing import Any from pip._vendor.dependency_groups import DependencyGroupResolver from pip._internal.exceptions import InstallationError from pip._internal.utils.compat import tomllib 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