benedict.core.traverse

Complexity

Total Complexity

9

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	traverse()	0	4	2
A	_traverse_list()	0	5	2
A	_traverse_collection()	0	5	3
A	_traverse_dict()	0	6	2

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

from benedict.utils import type_util


def _traverse_collection(d, callback):
    if type_util.is_dict(d):
        _traverse_dict(d, callback)
    elif type_util.is_list_or_tuple(d):
        _traverse_list(d, callback)


def _traverse_dict(d, callback):
    keys = list(d.keys())
    for key in keys:
        value = d.get(key, None)
        callback(d, key, value)
        _traverse_collection(value, callback)


def _traverse_list(l, callback):
    for index in range(len(l)):
        value = l[index]
        callback(l, index, value)
        _traverse_collection(value, callback)


def traverse(d, callback):
    if not callable(callback):
        raise ValueError('callback argument must be a callable.')
    _traverse_collection(d, callback)