o ,wi_ @sdZddlZddlmZmZmZddlmZddlmZddZ dd Z d d Z d d Z ej e e e dZGdddejZeddZ ddededefddZdS)z4Utilities for traversing and converting dataclasses.N)AnyOptionalType)config)daglishcCsddt|DS)NcSsg|]}|jqS)name.0fieldrra/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fiddle/_src/experimental/dataclasses.py sz_names..) dataclassesfields) value_or_typerrr _namessrcs&t}fdd|Dt|ffS)Ncsg|]}t|qSr)getattrr rvaluerr r z&_flatten_dataclass..)rtype)rnamesrrr _flatten_dataclasssrcCs>|\}}t|t|ksJddt||D}|di|S)NcSsi|]\}}||qSrr)r rrrrr %rz(_unflatten_dataclass..r)lenzip)valuesmetadatatypr values_dictrrr _unflatten_dataclass"sr!cCstddt|DS)Ncss|]}t|VqdSN)rAttrrrrr *sz$_path_elements_fn..)tuplerrrrr _path_elements_fn)sr&)flatten unflatten path_elementsc@s(eZdZdeedeejfddZdS)DataclassTraverserRegistry node_typereturncCs&|j|}|durt|rt}|Sr")fallback_registryfind_node_traverserr is_dataclassdataclass_traverser)selfr+ traverserrrr r.6s z.DataclassTraverserRegistry.find_node_traverserN) __name__ __module__ __qualname__rrrr NodeTraverserr.rrrr r*4s r*T) use_fallbackFrootallow_post_initr,cs0dtjffdd }t||t}|||S)aConverts a dataclass or nested structure of dataclasses to Fiddle configs. Args: root: Root dataclass instance or structure (lists, tuples, dicts, etc.) of dataclasses. allow_post_init: Allow conversion of dataclasses with a __post_init__ method. Returns: Fiddle config that, when built, should return an object equal to `root` (presuming `root` does not already contain Fiddle configs). statecsbtr,tts,ttdrstdtjtfifddtD| S)N __post_init__zVDataclasses with __post_init__ are not allowed, unless allow_post_init is set to True.cs"i|] }|jr|jt|jqSr)initrrr rrr r[s zDconvert_dataclasses_to_configs..traverse..) rr/ isinstancerhasattr ValueErrorrConfigr map_children)rr:r9rr traverseRs  z0convert_dataclasses_to_configs..traverse)rStateMemoizedTraversaldaglish_dataclass_registry initial_state)r8r9rCr:rrBr convert_dataclasses_to_configsBs rH)F)__doc__rtypingrrr fiddle._srcrrrrr!r&r6r0NodeTraverserRegistryr*rFboolrHrrrr s0