benedict.utils.keypath_util

Complexity

Total Complexity

12

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
B	list_keys()	0	14	6
B	check_keys()	0	13	6

# -*- coding: utf-8 -*-

from benedict.utils import dict_util

from six import string_types


def check_keys(d, separator):
    """
    Check if dict keys contain keypath separator.
    """
    if not isinstance(d, dict) or not separator:
        return

    def check_key(parent, key, value):
        if key and isinstance(key, string_types) and separator in key:
            raise ValueError(
                'keys should not contain keypath separator '
                '\'{}\', found: \'{}\'.'.format(separator, key))
    dict_util.traverse(d, check_key)


def list_keys(keypath, separator):
    """
    List keys splitting a keypath using the given separator.
    """
    if isinstance(keypath, string_types):
        if separator and separator in keypath:
            return list(keypath.split(separator))
        return [keypath]
    elif isinstance(keypath, (list, tuple, )):
        keys = []
        for key in keypath:
            keys += list_keys(key, separator)
        return keys
    return [keypath]