Find all occurrences of a key in nested dictionaries and lists

search_for_keys.py

# Source: https://stackoverflow.com/questions/9807634/find-all-occurrences-of-a-key-in-nested-dictionaries-and-lists
# Python dict.iteritems became just dict.items.
# Easily coherse the generator into a list with: `list(gen_dict_extract("KeytoFind", MyDict))`

def gen_dict_extract(key, var):
    if hasattr(var, 'items'):
        for k, v in var.items():
            if k == key:
                yield v
            if isinstance(v, dict):
                for result in gen_dict_extract(key, v):
                    yield result
            elif isinstance(v, list):
                for d in v:
                    for result in gen_dict_extract(key, d):
                        yield result