python-benedict/benedict/core/flatten.py

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

from benedict.core import clone
from benedict.utils import type_util


def _flatten_key(base_key, key, separator):
    if base_key and separator:
        return '{}{}{}'.format(base_key, separator, key)
    return key


def flatten(d, separator='_', **kwargs):
    new_dict = clone(d, empty=True)
    keys = list(d.keys())
    base_key = kwargs.pop('base_key', '')
    for key in keys:
        value = d.get(key, None)
        new_key = _flatten_key(base_key, key, separator)
        if type_util.is_dict(value):
            new_value = flatten(value, separator=separator, base_key=new_key)
            new_value.update(new_dict)
            new_dict.update(new_value)
            continue
        new_dict[new_key] = value
    return new_dict