32 lines
978 B
Python
32 lines
978 B
Python
from benedict.core import clone
|
|
from benedict.utils import type_util
|
|
|
|
|
|
def _flatten_key(base_key, key, separator):
|
|
if base_key and separator:
|
|
return f"{base_key}{separator}{key}"
|
|
return key
|
|
|
|
|
|
def _flatten_item(d, base_dict, base_key, separator):
|
|
new_dict = base_dict
|
|
keys = list(d.keys())
|
|
for key in keys:
|
|
new_key = _flatten_key(base_key, key, separator)
|
|
value = d.get(key, None)
|
|
if type_util.is_dict(value):
|
|
new_value = _flatten_item(
|
|
value, base_dict=new_dict, base_key=new_key, separator=separator
|
|
)
|
|
new_dict.update(new_value)
|
|
continue
|
|
if new_key in new_dict:
|
|
raise KeyError(f"Invalid key: {new_key!r}, key already in flatten dict.")
|
|
new_dict[new_key] = value
|
|
return new_dict
|
|
|
|
|
|
def flatten(d, separator="_"):
|
|
new_dict = clone(d, empty=True)
|
|
return _flatten_item(d, base_dict=new_dict, base_key="", separator=separator)
|