remap enter to skip some items about boltons HOT 4 CLOSED

def enter(path, key, value):
    # I'd like to skip `baz's`
    if 'baz' in value and isinstance(value, dict):
        value.pop('baz')
    return default_enter(path, key, value)

print(remap({'foo': 'bar', 'baz': 'qux', 'data': ['foo', 'bar', 'baz']}, enter=enter))

>>> {'foo': 'bar', 'data': ['foo', 'bar', 'baz']}

import copy

d = {
    'name': 'ROOT',
    'properties': ['foo'],
    'includes': [
        {
            'name': 'nested-1',
            'properties-override': ['bar', 'baz'],
            'includes': [
                {'id': '1'},
                {'id': '2', 'properties': ['quz']},
                {'id': '3'},
            ]
        }
    ]
}


def traverse(data, parents=None):
    result = list()

    if parents is None:
        parents = list()

    if 'id' in data:
        assert 'includes' not in data
        data['parents'] = copy.deepcopy(parents)
        return [data]
    else:
        includes = data.pop('includes', list())
        parents.append(data)

        if not isinstance(includes, list):
            includes = [includes]

        for inc in includes:
            child = traverse(inc, parents)
            result.extend(child)

        parents.pop()
    return result


if __name__ == '__main__':
    for item in traverse(d):
        print(item)

>>> {'id': '1', 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}
>>> {'id': '2', 'properties': ['quz'], 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}
>>> {'id': '3', 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}

import copy
from boltons.iterutils import remap, default_enter, default_exit

d = {
    'name': 'ROOT',
    'properties': ['foo'],
    'includes': [
        {
            'name': 'nested-1',
            'properties-override': ['bar', 'baz'],
            'includes': [
                {'id': '1'},
                {'id': '2', 'properties': ['quz']},
                {'id': '3'},
            ]
        }
    ]
}


def traverse(data=None):
    if not data:
        data = d

    result = list()
    parents = list()

    def enter(path, key, value):
        if 'includes' in value:
            includes = value.pop('includes')
            parents.append(value)
            value = includes

        res = default_enter(path, key, value)
        return res

    def visit(path, key, value):
        if 'id' in value and isinstance(value, dict):
            item = copy.deepcopy(value)
            item['parents'] = copy.deepcopy(parents)
            result.append(item)

        return key, value

    def exit(path, key, old_parent, new_parent, new_items):
        if old_parent == parents[-1]:
            parents.pop()
        res = default_exit(path, key, old_parent, new_parent, new_items)
        return res

    remap(data, visit=visit, enter=enter, exit=exit)
    return result


if __name__ == '__main__':
    for item in traverse(d):
        print(item)

>>> {'id': '1', 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}
>>> {'id': '2', 'properties': ['quz'], 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}
>>> {'id': '3', 'parents': [{'name': 'ROOT', 'properties': ['foo']}, {'properties-override': ['bar', 'baz'], 'name': 'nested-1'}]}