python_dependencies

Complexity

Total Complexity

3

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	dependency_graph_model()	0	22	3

from pipdeptree._cli import get_options
from pipdeptree._discovery import get_installed_distributions
from pipdeptree._models import PackageDAG

import graphinate


def dependency_graph_model():
    """
    Generate a dependency graph model.

    Returns:
        GraphModel: A graph model representing the dependency graph.
    """

    options = get_options(args=None)

    pkgs = get_installed_distributions(local_only=options.local_only, user_only=options.user_only)
    tree = PackageDAG.from_pkgs(pkgs)

    graph_model = graphinate.model(name="Dependency Graph")

    @graph_model.edge()
    def dependency():
        for p, d in tree.items():
            for c in d:
                yield {'source': p.project_name, 'target': c.project_name}

    return graph_model


if __name__ == '__main__':
    dependency_model = dependency_graph_model()
    schema = graphinate.builders.GraphQLBuilder(dependency_model).build()
    graphinate.graphql.server(schema)