benedict.utils.keypath_util

Complexity

Total Complexity

17

Size/Duplication

Total Lines	48
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	join_keys()	0	2	1
A	all_keys()	0	8	4
B	split_keys()	0	18	6
B	check_keys()	0	8	6

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

from six import string_types


def all_keys(d):
    keys = []
    for key, val in d.items():
        if key not in keys:
            keys.append(key)
        if isinstance(val, dict):
            keys += all_keys(val)
    return keys


def check_keys(keys, separator):
    if separator:
        for key in keys:
            if key and isinstance(key, string_types):
                if separator in key:
                    raise ValueError(
                        'keys should not contain keypath separator '
                        '\'{}\'.'.format(separator))


def join_keys(keys, separator):
    return separator.join(keys)


def split_keys(key, separator):
    if separator:
        if isinstance(key, string_types):
            keypath = key
            if separator in keypath:
                keys = list(keypath.split(separator))
                return keys
            else:
                return [key]
        elif isinstance(key, (list, tuple, )):
            keys = []
            for key_item in key:
                keys += split_keys(key_item, separator)
            return keys
        else:
            return [key]
    else:
        return [key]