Russell Ballestrini
756ad31763
modified: README.rst modified: nested_lookup/nested_lookup.py modified: test_nested_loopkup.py
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
from six import iteritems
|
|
|
|
from collections import defaultdict
|
|
|
|
def nested_lookup(key, document, wild=False, with_keys=False):
|
|
"""Lookup a key in a nested document, return a list of values"""
|
|
if with_keys:
|
|
d = defaultdict(list)
|
|
for k, v in _nested_lookup(key, document, wild=wild, with_keys=with_keys):
|
|
d[k].append(v)
|
|
return d
|
|
return list(_nested_lookup(key, document, wild=wild, with_keys=with_keys))
|
|
|
|
def _nested_lookup(key, document, wild=False, with_keys=False):
|
|
"""Lookup a key in a nested document, yield a value"""
|
|
if isinstance(document, list):
|
|
for d in document:
|
|
for result in _nested_lookup(key, d, wild=wild, with_keys=with_keys):
|
|
yield result
|
|
|
|
if isinstance(document, dict):
|
|
for k, v in iteritems(document):
|
|
if key == k or (wild and key.lower() in k.lower()):
|
|
if with_keys:
|
|
yield k, v
|
|
else:
|
|
yield v
|
|
elif isinstance(v, dict):
|
|
for result in _nested_lookup(key, v, wild=wild, with_keys=with_keys):
|
|
yield result
|
|
elif isinstance(v, list):
|
|
for d in v:
|
|
for result in _nested_lookup(key, d, wild=wild, with_keys=with_keys):
|
|
yield result
|