clearml-server/server/utilities/dicts.py
2019-12-24 18:01:48 +02:00

36 lines
1.2 KiB
Python

from typing import Sequence, Tuple, Any
def flatten_nested_items(
dictionary: dict, nesting: int = None, include_leaves=None, prefix=None
) -> Sequence[Tuple[Tuple[str, ...], Any]]:
"""
iterate through dictionary and return with nested keys flattened into a tuple
"""
next_nesting = None if nesting is None else (nesting - 1)
prefix = prefix or ()
for key, value in dictionary.items():
path = prefix + (key,)
if isinstance(value, dict) and nesting != 0:
yield from flatten_nested_items(
value, next_nesting, include_leaves, prefix=path
)
elif include_leaves is None or key in include_leaves:
yield path, value
def deep_merge(source: dict, override: dict) -> dict:
"""
Merge the override dict into the source in-place
Contrary to the dpath.merge the sequences are not expanded
If override contains the sequence with the same name as source
then the whole sequence in the source is overridden
"""
for key, value in override.items():
if key in source and isinstance(source[key], dict) and isinstance(value, dict):
deep_merge(source[key], value)
else:
source[key] = value
return source