repositories.repo_graph_model()

Complexity

Conditions

9

Size

Total Lines	99
Code Lines	76

Duplication

Lines	0
Ratio	0 %

Importance

Changes

0

import itertools
def repo_graph_model():  # noqa: C901
    """
    Create a graph model for GitHub repositories.

    Returns:
        GraphModel: A graph model representing GitHub repositories with nodes and edges.
    """

    graph_model = graphinate.model(name='GitHub Repository Graph')

    @graph_model.edge()
    def github(user_id: str | None = None,
               repository_id: str | None = None,
               commit_id: str | None = None,
               file_id: str | None = None,
               **kwargs):
        user = github_user(user_id)
        for repo in github_repositories(user_id, repository_id):
            yield {'source': (user.login,), 'target': (user.login, repo.name)}
            for commit in github_commits(repo, commit_id):
                yield {
                    'source': (user.login, repo.name),
                    'target': (user.login, repo.name, commit.sha)
                }
                for file in github_files(commit, file_id):
                    yield {
                        'source': (user.login, repo.name, commit.sha),
                        'target': (user.login, repo.name, commit.sha, file.filename)
                    }

    user_node = graph_model.node(key=operator.attrgetter('login'),
                                 value=operator.attrgetter('raw_data'),
                                 label=operator.itemgetter('name'))

    repository_node = graph_model.node(parent_type='user',
                                       key=operator.attrgetter('name'),
                                       value=operator.attrgetter('raw_data'),
                                       label=operator.itemgetter('name'))

    def commit_label(commit):
        return commit['sha'][-7:]

    commit_node = graph_model.node(parent_type='repository',
                                   key=operator.attrgetter('sha'),
                                   value=operator.attrgetter('raw_data'),
                                   label=commit_label)

    file_node = graph_model.node(parent_type='commit',
                                 unique=True,
                                 key=operator.attrgetter('filename'),
                                 value=operator.attrgetter('raw_data'),
                                 label=operator.itemgetter('filename'))

    @user_node
    def user(user_id: str | None = None, **kwargs):
        yield github_user(user_id)

    @repository_node
    def repository(user_id: str | None = None,
                   repository_id: str | None = None,
                   **kwargs):
        repos = github_repositories(user_id, repository_id)
        yield from repos

    @commit_node
    def commit(user_id: str | None = None,
               repository_id: str | None = None,
               commit_id: str | None = None,
               **kwargs):
        for repo in github_repositories(user_id, repository_id):
            yield from github_commits(repo, commit_id)

    def file_type(user_id: str | None = None,
                  repository_id: str | None = None,
                  commit_id: str | None = None,
                  file_type_id: str | None = None,
                  **kwargs):
        def group_key(file):
            return pathlib.PurePath(file).suffix

        for repo in github_repositories(user_id, repository_id):
            for commit in github_commits(repo, commit_id):
                yield from ((k, list(g)) for k, g in
                            itertools.groupby(
                                sorted(github_files(commit),
                                       key=group_key), group_key
                            ))

    @file_node
    def file(user_id: str | None = None,
             repository_id: str | None = None,
             commit_id: str | None = None,
             file_id: str | None = None,
             **kwargs):
        for repo in github_repositories(user_id, repository_id):
            for commit in github_commits(repo, commit_id):
                yield from github_files(commit, file_id)

    return graph_model