python-benedict/benedict/core/search.py

from benedict.core.traverse import traverse
from benedict.utils import type_util


def _get_term(value, case_sensitive):
    v_is_str = type_util.is_string(value)
    v = value.lower() if (v_is_str and not case_sensitive) else value
    return (v, v_is_str)


def _get_match(query, value, exact, case_sensitive):
    q, q_is_str = _get_term(query, case_sensitive)
    v, v_is_str = _get_term(value, case_sensitive)
    # TODO: add regex support
    if q_is_str and v_is_str and not exact:
        return q in v
    return q == v


def search(d, query, in_keys=True, in_values=True, exact=False, case_sensitive=True):
    items = []

    def _search_item(item_dict, item_key, item_value):
        match_key = in_keys and _get_match(query, item_key, exact, case_sensitive)
        match_val = in_values and _get_match(query, item_value, exact, case_sensitive)
        if any([match_key, match_val]):
            items.append((item_dict, item_key, item_value))

    traverse(d, _search_item)
    return items