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

Fix defect where wild would break if document had non string keys.

Browse source 
modified:   README.rst
	modified:   nested_lookup/nested_lookup.py
	modified:   setup.py
	modified:   test_nested_lookup.py
This commit is contained in:
Russell Ballestrini 2019-06-25 14:50:00 -07:00
parent 9348af1790
commit b6acaca980
4 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
nested_lookup/nested_lookup.py
View file

@ -13,6 +13,11 @@ def nested_lookup(key, document, wild=False, with_keys=False):
return list(_nested_lookup(key, document, wild=wild, with_keys=with_keys))
def _is_case_insensitive_substring(a, b):
"""return True if `a` is a case insensitive substring of `b`, else False"""
return isinstance(a, str) and isinstance(b, str) and a.lower() in b.lower()
def _nested_lookup(key, document, wild=False, with_keys=False):
"""Lookup a key in a nested document, yield a value"""
if isinstance(document, list):
@ -22,7 +27,7 @@ def _nested_lookup(key, document, wild=False, with_keys=False):
if isinstance(document, dict):
for k, v in iteritems(document):
if key == k or (wild and key.lower() in k.lower()):
if key == k or (wild and _is_case_insensitive_substring(key, k)):
if with_keys:
yield k, v
else: