def build_dependency_graph(
working_set: WorkingSet,
marker_env: dict[str, str] | None = None) -> DirectedGraph:
"""Build a dependency graph from locked result."""
graph: DirectedGraph[Package | None] = DirectedGraph()
graph.add(None) # sentinel parent of top nodes.
node_with_extras: set[str] = set()
def add_package(key: str, dist: Distribution | None) -> Package:
name, extras = strip_extras(key)
extras = extras or ()
reqs: dict[str, Requirement] = {}
if dist:
requirements = (parse_requirement(r)
for r in filter_requirements_with_extras(
cast(str, dist.metadata["Name"]),
dist.requires or [],
extras,
include_default=True,
))
for req in requirements:
if not req.marker or req.marker.evaluate(marker_env):
reqs[req.identify()] = req
version: str | None = dist.version
else:
version = None
node = Package(key, version, reqs)
if node not in graph:
if extras:
node_with_extras.add(name)
graph.add(node)
for k in reqs:
child = add_package(k, working_set.get(strip_extras(k)[0]))
graph.connect(node, child)
return node
for k, dist in working_set.items():
add_package(k, dist)
for node in list(graph):
if node is not None and not list(graph.iter_parents(node)):
# Top requirements
if node.name in node_with_extras:
# Already included in package[extra], no need to keep the top level
# non-extra package.
graph.remove(node)
else:
graph.connect(None, node)
return graph
def build_dependency_graph(working_set: WorkingSet) -> DirectedGraph:
"""Build a dependency graph from locked result."""
graph: DirectedGraph[Package | None] = DirectedGraph()
graph.add(None) # sentinel parent of top nodes.
node_with_extras = set()
def add_package(key: str, dist: Distribution) -> Package:
name, extras = strip_extras(key)
extras = extras or ()
reqs: dict[str, Requirement] = {}
if dist:
requirements = (
Requirement.from_pkg_requirement(r)
for r in dist.requires(extras) # type: ignore
)
for req in requirements:
reqs[req.identify()] = req
version = dist.version
else:
version = None
node = Package(key, version, reqs)
if node not in graph:
if extras:
node_with_extras.add(name)
graph.add(node)
for k in reqs:
child = add_package(
k, cast(Distribution, working_set.get(strip_extras(k)[0])))
graph.connect(node, child)
return node
for k, dist in working_set.items():
add_package(k, dist)
for node in list(graph):
if node is not None and not list(graph.iter_parents(node)):
# Top requirements
if node.name in node_with_extras:
# Already included in package[extra], no need to keep the top level
# non-extra package.
graph.remove(node)
else:
graph.connect(None, node)
return graph
def graph():
return DirectedGraph()