python/nested-lookup
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity Actions

Updated README.rst and set list as default output

Browse source
This commit is contained in:
Guillaume 2018-04-24 10:30:29 -07:00
parent dfc2483e3e
commit 3b1f2ced67
2 changed files with 67 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

10
nested_lookup/nested_lookup.py
View file

@ -1,15 +1,13 @@
from six import iteritems
def nested_lookup(key, document, wild=False, output='dict'):
def nested_lookup(key, document, wild=False, output='list'):
"""Lookup a key in a nested document, return a list of values"""
if output == 'dict':
return dict(_nested_lookup(key, document, wild=wild, output=output))
elif output == 'list':
return list(_nested_lookup(key, document, wild=wild, output=output))
else:
raise ValueError('Output is of unknown type '+output)
return list(_nested_lookup(key, document, wild=wild, output=output))
def _nested_lookup(key, document, wild=False, output='dict'):
def _nested_lookup(key, document, wild=False, output='list'):
"""Lookup a key in a nested document, yield a value"""
if isinstance(document, list):
for d in document:
@ -21,7 +19,7 @@ def _nested_lookup(key, document, wild=False, output='dict'):
if key == k or (wild and key.lower() in k.lower()):
if output == 'dict':
yield k, v
elif output == 'list':
else:
yield v
elif isinstance(v, dict):
for result in _nested_lookup(key, v, wild=wild, output=output):