o Yi@szddlmZWneyddlmZYnwddlZddlmZmZmZm Z ddl m Z m Z m Z mZeeee dZe e e edZ    dd d Zd d ZdddZdS))MappingN) tuple_reducer path_reducer dot_reducerunderscore_reducer)tuple_splitter path_splitter dot_splitterunderscore_splitter)tuplepathdot underscorer Fc sttft|stdt|fdur$dkr$tdttr-tidfdd |ddS) aFlatten `Mapping` object. Parameters ---------- d : dict-like object The dict that will be flattened. reducer : {'tuple', 'path', 'underscore', 'dot', Callable} The key joining method. If a `Callable` is given, the `Callable` will be used to reduce. 'tuple': The resulting key will be tuple of the original keys. 'path': Use `os.path.join` to join keys. 'underscore': Use underscores to join keys. 'dot': Use dots to join keys. inverse : bool Whether you want invert the resulting key and value. max_flatten_depth : Optional[int] Maximum depth to merge. enumerate_types : Sequence[type] Flatten these types using `enumerate`. For example, if we set `enumerate_types` to ``(list,)``, `list` indices become keys: ``{'a': ['b', 'c']}`` -> ``{('a', 0): 'b', ('a', 1): 'c'}``. keep_empty_types : Sequence[type] By default, ``flatten({1: 2, 3: {}})`` will give you ``{(1,): 2}``, that is, the key ``3`` will disappear. This is also applied for the types in `enumerate_types`, that is, ``flatten({1: 2, 3: []}, enumerate_types=(list,))`` will give you ``{(1,): 2}``. If you want to keep those empty values, you can specify the types in `keep_empty_types`: >>> flatten({1: 2, 3: {}}, keep_empty_types=(dict,)) {(1,): 2, (3,): {}} Returns ------- flat_dict : dict z3argument type %s is not in the flattenalbe types %sNrz,max_flatten_depth should not be less than 1.c st|r t|nt|}d}|D]?\}}d}||}t|r;dus*|kr;||d|d}|s:t|s;qrB||}}|vrMtd|||<q|S)NFTr)depthparentduplicated key '{}') isinstance enumeratesix viewitems ValueErrorformat) _drrkey_value_iterablehas_itemkeyvalueflat_key has_child_flattenenumerate_types flat_dictflattenable_typesinversekeep_empty_typesmax_flatten_depthreducerrM/home/ubuntu/.local/lib/python3.10/site-packages/flatten_dict/flatten_dict.pyr"Vs"     zflatten.._flatten)r)N)r rrrtypestr REDUCER_DICT)dr)r&r(r#r'rr!r*flattens +     r/cCsb|sJ|d}t|dkr||vrtd||||<dS||i}t||dd|dS)zSet a value to a sequence of nested keys. Parameters ---------- d : Mapping keys : Sequence[str] value : Any rrrN)lenrr setdefaultnested_set_dict)r.keysrrrrr*r2us   r2cCsPt|tr t|}i}t|D]\}}|r||}}||}t|||q|S)aUnflatten dict-like object. Parameters ---------- d : dict-like object The dict that will be unflattened. splitter : {'tuple', 'path', 'underscore', 'dot', Callable} The key splitting method. If a Callable is given, the Callable will be used to split `d`. 'tuple': Use each element in the tuple key as the key of the unflattened dict. 'path': Use `pathlib.Path.parts` to split keys. 'underscore': Use underscores to split keys. 'dot': Use underscores to split keys. inverse : bool Whether you want to invert the key and value before flattening. Returns ------- unflattened_dict : dict )rr, SPLITTER_DICTrrr2)r.splitterr&unflattened_dictrr key_tuplerrr* unflattens  r8)r FNrr)r F)collections.abcr ImportError collectionsrreducersrrrr splittersrr r r r-r4r/r2r8rrrr*s4   Z