find stuff in overly informative API responses

json_searching_utils.py

def walk(obj, path):
    last = obj
    for key in path:
        if isinstance(last, list):
            last = last[key]
        else:
            last = last.get(key, {})


def prune(obj, keys=["Name", "DisplayMessage", "Type", "Time"]):
    if isinstance(obj, (list, tuple)):
        return [
            maybe_child for item in obj if (maybe_child := prune(item, keys))
        ]
    elif isinstance(obj, dict):
        result = {}
        for k, v in obj.items():
            if k in keys:
                result[k] = v
            if isinstance(v, (dict, list, tuple)):
                maybe_child = prune(v, keys)
                if maybe_child:
                    result[k] = maybe_child
        return result