o ,wi4 @sdZddlZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddl mZddl mZddl mZddlmZejd d Gd d d Zd2d edefddZdejdefddZd ededefddZejd d GdddZejd d GdddZde jdejde e efddZddZ d3d e jd ede jfd!d"Z d d#d$de jd%ededefd&d'Z!d#d(dededefd)d*Z"dededeefd+d,Z#dejd-e ej$dedefd.d/Z%de jdeeeffd0d1Z&dS)4z8Functions to output representations of `fdl.Buildable`s.N)AnyDictIteratorListOptionalType)config)daglish)history)tagging)formatting_utilitiesT)frozenc@s.eZdZUdZdZejed<defddZ dS) _UnsetValuez*A wrapper class indicating an unset value.) parameterrreturncCs0|jj|jjur dSt|jjdd}d|dS)N <[unset]>Fraw_value_reprz<[unset; default: z]>)rdefaultempty _format_value)self default_valuerQ/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fiddle/_src/printing.py__repr__%s  z_UnsetValue.__repr__N) __name__ __module__ __qualname____doc__ __slots__inspect Parameter__annotations__strrrrrrrs  rvaluercCs6|ptjt|}t|tjp||ot| |SN) r MemoizedTraversalbegin_has_nested_builder isinstancer Buildableis_traversableanyyield_map_child_values)r%staterrrr).s r)pathcCs.t|}|rt|dtjr|ddS|S)aFormats path in a way customized to this file. In the future, we may wish to consider a format that is readable for printing more arbitrary collections. Args: path: A path, which typically starts with an attribute. Returns: String representation of the path. rN)r path_strr*Attr)r0r2rrr _path_str5s r4rcCs(|rt|St|trt|St|Sr&)reprr*r$r pretty_print)r%rrrrrFs   rc@s(eZdZUdZdZejed<ddZdS)_TaggedValueWrapperz?Customizes representation for TaggedValues in flattened output.)wrappedr8cCsJ|jjtjur d}nt|jj}dtdd|jjD}|d|S)Nr css|]}t|VqdSr&)r$).0trrr [sz/_TaggedValueWrapper.__repr__..)r8r%r NO_VALUEr5joinsortedtags)r value_reprtag_strrrrrVs  z_TaggedValueWrapper.__repr__N) rrrrr r TaggedValueClsr#rrrrrr7Ps   r7c@s4eZdZUdZejed<eee ed<e ed<dS) _LeafSettingz(Represents a leaf configuration setting.r0 annotationr%N) rrrrr Pathr#rrrrrrrrD_s   rDcfgcCs|r t|dtjs dS|}|ddD]}||}qt|tjraz |jj|dj}Wn(t yU|jj D]}|j |j krQ|j |jurJdn|j YSq:YdSw|j |jur^dS|j SdS)z8Gets the type annotation associated with a Daglish path.N)r*r r3followrr+__signature_info__ parametersnameKeyErrorvalueskind VAR_KEYWORDrEr)rGr0r% path_elementparamrrr_get_annotationgs"    rScCsN|r%|d}t|tjr#t||dd}t|tjr#|j|jSdSdS)zEReturns the tags for a given Daglish path, or None if there are none.rHN) r*r r3 follow_pathrr+__argument_tags__getrL)rGr0childparentrrr _get_tags{s  rYinsert_unset_sentinelscCst|}t|j}i}|jjD]"\}}|j|j|jhvr q||vr,| |||<q|r4t |||<q| |t |d||S)aReturns a copy of a Buildable with normalized arguments. This normalizes arguments by re-creating the __arguments__ dictionary in the order of the configured function or class' signature. It also inserts "unset" sentinels for values in the signature that don't have a value set. Args: value: Buildable to copy and normalize. insert_unset_sentinels: If true, insert unset sentinels to arguments as the default values. Returns: Copy of `value` with arguments normalized. __arguments__)copydictr[rJrKitemsrOrPVAR_POSITIONALpoprupdateobject __setattr__)r%rZ old_arguments new_arguments param_namerRrrr_rearrange_buildable_argss    rgF) include_typesrrhcs\d dttffdd tddi}ttd|dtffd d }d t|S) acReturns a string of cfg's paths and values, one pair per line. Some automated tools (e.g. grep, sort, diff, ...) handle data line-by-line. This output format contains the full path to a value and a string representation of said value on a single line. Args: cfg: A buildable to generate a string representation for. include_types: If true, include type annotations (where available) in the string representation of each leaf value. raw_value_repr: If true, use `repr` on values, otherwise, a custom pretty-printed format is used. Returns: a string representation of `cfg`. Nrc3s|p tj|}t|j}|rtj||d}t|tj r#t |}t|tj r=t |}t |j}t|j||VdSt|sQt |j}t|j||VdS||sXJ||D]}|EdHq]dS)N)r@r)r BasicTraversalr(rY current_pathr TaggedValuer*rr+rgrCr7rSrDr)r,r.)r%r/r@rE sub_result)rGgeneraterrrms$      z"as_str_flattened..generater GenericAliaslinecspd}r%|jdur%t|j}t|jtr|sd|jj}nd|j}t|jd}t|j|d|S)Nrnz: r = )rEr*typerrr%r4r0)rptype_annotationis_parameterized_genericr%) generic_aliasrhrrr format_lines  z%as_str_flattened..format_line r&)rrDrrgetattrtypesr>map)rGrhr dummy_typervr)rGrmrurhrras_str_flatteneds   r|rcCsdt||dS)anReturns a string representing the history of cfg's leaf params. Because Buildable's are designed to be mutated, tracking down when and where a particular parameter's value is changed can be difficult. This function returns a string representation of each parameter and (in reverse-chronological order) its current and past values. This representation elides the "DAG"-construction aspects of constructing the `cfg`, and instead only prints the history of the "outer-most" parameters (ones that don't contain a reference to another Buildable). Args: cfg: A buildable, or collection containing buildables, to generate a history for. For non-Buildable collections, either (a) the entire collection will be printed as a history element or (b) only nested Buildable elements will be printed, since we can't store assignment history for normal Python collections like lists/dicts. raw_value_repr: If true, use `repr` when string-ifying values, otherwise use a customized pretty-printing routine. Returns: A string representation of `cfg`'s history, organized by param name. rwr)r>"_make_per_leaf_histories_recursiverGrrrrhistory_per_leaf_parameters rcs"ddttffdd |S)zFRecursively traverses `cfg` and generates per-param history summaries.Nrc3s|p tj|}t|tjrm|jD],\}}t||}t |r/| |t |EdHqg|j t |R}t ||Vqtt|jjt|j}|D]}g|j t |R}t|dVqUdS||r||D]}|EdHqwdSt|j d|VdS)Nz = <[unset]>rq)r rir(r*rr+__argument_history__r^rxr)callr3rj_make_per_leaf_history_textr?setrJrKkeysr4r,r.)r%r/rL param_history sub_valuer0 unset_fieldsrlrtraverserrr s.     z4_make_per_leaf_histories_recursive..traverser&)rr$r~rrrr}sr}rcs|sJddtjdtffdd dd|D}t|dkr8fd dt|d d D}d |}d |}nd }t|d jd}|d|d j}t |d||S)afReturns a string representing a parameter's history. Args: path: The path to the parameter. param_history: The parameter's history. raw_value_repr: If True, use `repr` unmodified, otherwise, use a custom pretty-printing routine. Returns: A string representation of the parameter's history that can be displayed to the user. z$param_history should never be empty.entryrcs t|jd}d|d|jS)Nrz - previously:  @ )r new_valuelocation)rr%rrrmake_previous_text<sz7_make_per_leaf_history_text..make_previous_textcSsg|] }|jtjjkr|qSr)rOr ChangeKind NEW_VALUEr:rrrr As z/_make_per_leaf_history_text..r1csg|]}|qSrrr)rrrrFsNrHrwrnrrrq) r HistoryEntryr$lenreversedr>rrrr4)r0rr value_history value_updatespast current_valuecurrentr)rrrr+s"      rcs>ddttffdd i}|D] }|j|t|j<q|S)a Returns a flattened dict of cfg's paths (dot syntax) and values. Default values and tags won't be included in the flattened dict. Args: cfg: A buildable to generate a flattened dict representation for. Returns: A flattened Dict representation of `cfg`. Nrc3st|p tj|}t|tjrt|dd}t|s$t|j d|VdS| |s+J| |D]}|EdHq0dS)NF)rZ) r rir(r*rr+rgr)rDrjr,r.)r%r/rl dict_generaterrr^s   z(as_dict_flattened..dict_generater&)rrDr%r4r0)rG args_dictleafrrras_dict_flattenedSs   rr&)T)'rr\ dataclassesr!rytypingrrrrrr fiddle._srcrr r r fiddle._src.codegenr dataclassrboolr)rFr$r4rr7rDr+rSrYrgr|rr}rrrrrrrs             $  C   $ (