Restructuring GitHub GraphQL responses with Glom

glom_graphql_restructure.py

from glom import glom, Coalesce, Val
from rich.pretty import pprint

nodes = [
    {
        "owner": {"login": "python"},
        "name": "cpython",
        "stargazerCount": 48591,
        "createdAt": "2017-02-10T19:23:51Z",
        "updatedAt": "2022-11-03T04:49:20Z",
        "defaultBranchRef": {
            "name": "main",
            "target": {"history": {"totalCount": 115096}},
        },
    },
    {
        "owner": {"login": "aosp-mirror"},
        "name": "kernel_common",
        "stargazerCount": 435,
        "createdAt": "2008-10-21T18:19:54Z",
        "updatedAt": "2022-10-30T10:39:56Z",
        "defaultBranchRef": None,
    },
]

spec = [
    {
        "owner": "owner.login",
        "repo": "name",
        "stars": "stargazerCount",
        "created": "createdAt",
        "updated": "updatedAt",
        "commits": Coalesce("defaultBranchRef.target.history.totalCount", Val(None)),
    }
]

print("Original node data:")
pprint(nodes, indent_guides=False, expand_all=True)

print("\nRestructured with glom:")
pprint(glom(nodes, spec), indent_guides=False, expand_all=True)

restructured_output.txt

[
    {
        'owner': 'python',
        'repo': 'cpython',
        'stars': 48591,
        'created': '2017-02-10T19:23:51Z',
        'updated': '2022-11-03T04:49:20Z',
        'commits': 115096
    },
    {
        'owner': 'aosp-mirror',
        'repo': 'kernel_common',
        'stars': 435,
        'created': '2008-10-21T18:19:54Z',
        'updated': '2022-10-30T10:39:56Z',
        'commits': None
    }
]