o qoi, @s6dZddlZddlZddlmZmZmZmZmZddl m Z ddl m Z ddlmZddlmZddlmZdd lmZejZejZd ed ee jd efd dZejddGdddZde jd ee jfddZded efddZejGdddZ d$dede deded df d d!Z!de jd ej"fd"d#Z#dS)%z "foo[3].bar". Args: base_var: Base variable name. path: Attribute path on `base_var` to assign to. Returns: Python code string for the LHS of an assignment. Raises: TypeError: If the first path element is not a string, or if any path element is not a string or an int. css|]}|jVqdSN)code).0xrV/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/codegen/legacy_codegen.py 3sz"assignment_path..)join)rrrrrassignment_path"srT)frozenc@s"eZdZUdZeed<ddZdS) _VarReferencez6Helper class whose repr() is just the provided string.namecCs|jSr)r)selfrrr__repr__;sz_VarReference.__repr__N)__name__ __module__ __qualname____doc__str__annotations__rrrrrr6s  r buildablecsFtidtjffdd }tj||fddDS)zGFinds any sub-buildable nodes which are referenced by multiple parents.statecsDt|tjrt|d7<|t|<|j|ddD]}qdS)NT ignore_leaves) isinstance config_lib Buildableidyield_map_child_values)valuer'_children_by_idto_countrrtraverseFs  z(_get_shared_buildables..traversecs g|] \}}|dkr|qS)r(r)r child_hashcount)r3rr Os z*_get_shared_buildables..) collectionsCounterrStateBasicTraversalrunitems)r&r5rr2r_get_shared_buildables?s r?r0cs*ddtjffdd }tj||S)z2Returns whether a value has any nested buildables.Fr'cs4pt|tjs||r||dSdSdSr)r+r,r-is_traversableflattened_map_children) sub_valuer'resultrrr5Zsz'_has_child_buildables..traverse)rr;r<r=)r0r5rrCr_has_child_buildablesVsrEc@seZdZUdZeed<eed<eje dZ e e j ed<ejedZeeeejeffed<dejfdd Zd edejd e j d d fddZdedeejded e j fddZd S)SharedBuildableManagerz4Helper class to manage shared configuration objects.r r )default_factory instancesinstances_and_names_by_idr&cCst||jvSr)r.rI)rr&rrr __contains__psz#SharedBuildableManager.__contains__rdeclrNcCs"|j|||f|jt|<dS)zAdds a shared instance. Args: name: Variable name for code representing this instance. buildable: Actual Python Buildable object represented by the code being generated. decl: Code declaration for this instance. N)rHappendrIr.)rrr&rKrrraddss zSharedBuildableManager.addlhs_varlhs_path attr_valuecsd|rt|dtjrtSddfdd t||}t|t|}r0t|dS|S)aReturns an assignment for a Python value. When this value references shared builder objects, then those objects are replaced with name references to the shared objects. Args: lhs_var: Variable name of the left-hand side assignment. lhs_path: Attribute path on `lhs_var` to assign to. attr_value: Python value representing the right-hand side of the expression. Returns: Codegen node representing the assignment. FNcsj|ptj|}|vrtjt|dSt|tjr$dtdS| |r.| |St |j S)Nr(TNotImplemented)rr<beginrrIr.r+r,r-r@ map_childrenr transform_py_valuer )childr'rr5used_not_implementedrrr5s   z/SharedBuildableManager.assign..traversez0fdl.Config sub-nodes will replace NotImplementedr) r+rIndexr Noopr AssignmentreprTrailingComment)rrNrOrPlhs assignmentrrWrassigns zSharedBuildableManager.assign)r r!r"r# Namespacer% ImportManager dataclassesfieldlistrHrr CodegenNodedictrIrintrr,r-r$rJrMrr PathElementrr`rrrrrFds(     rFshared_shared_objectsshared_managerr variable_name_prefixcs@dtjffdd }tj||jd}|||dS)aGenerates configuration for a shared object. The objects configured will be added to `shared_manager`. Args: shared_objects: List of shared objects, or any other nested structure compatible with daglish. Its dependencies can form a DAG, and any leaf nodes will be added first. shared_manager: Shared object manager. import_manager: Import manager. variable_name_prefix: Prefix for any variables introduced. r'c s|j|ddD]}qt|tjr]jjt|jd}t|}|j }t ||d|dg}|j D]\}}t|g} ||| |q<||t |dSdS)z%Generates code for a shared instance.Tr))prefix()N)r/r+r,r-r get_new_name get_callabler rM __class__r r[ __arguments__r>r BuildableAttrrLr`ImmediateAttrsBlock) rVr'r1rrelnamebuildable_subclass_strnodeskeyr0rr rlrmrrr5s    z+_configure_shared_objects..traverse)memoN)rr;MemoizedTraversalrIcopy initial_state)rkrlr rmr5 traverserrr{r_configure_shared_objectssrcst}t|t|dtt|dgdtjdttffdd |g t dt  t jgS)aDGenerates code, preferring nested dot-attribute assignment when possible. Example code output (abbreviated and with additional comments): # Shared instances are on top, since this format supports DAGs. However # choosing names is hard. shared_foo = fdl.Config(Foo) shared_foo.a = 1 # Subsequent blocks set up a tree, doing attribute dot-assignment. root = fdl.Config(Baz) root.foo = shared_foo root.bar = fdl.Config(Bar) root.bar.foo2 = shared_foo Args: buildable: Config or Partial instance. Returns: Codegen node representing output code. r )rlr rVrcst|}|j}ttd|d|dggdtjffdd }tj || t D] \}}||q>dS)z8Configures a tree node for the main configuration block.rootrorpr'cs|js ||dSg|j}t|jdkrt|snt|tjr/|vr/||fn d||| |rKt|tjsM||dSdSdS)z;Inner handler for traverse_with_path, assigning attributes.Nr(r) current_pathrAlenrEr+r,r-rLr`r@)r0r' full_path)deferredryrrlrrhandle_child_attrs" zPcodegen_dot_syntax..configure_main_tree_block..handle_child_attrN) rMr,rrrsr r[rrr;r<r=rLrv)rVrrwrxr sub_childsub_pathconfigure_main_tree_blockr main_tree_blocksrl)rryrrr s   z5codegen_dot_syntax..configure_main_tree_blockr)rarbrFrr?r,r-rrrLr ReturnStmt ConfigBuildersorted_import_linesSharedThenResultAssignmentrH)r&r rrrcodegen_dot_syntaxs&" ( r)rj)$r#r9rctypingrrrrrfiddler fiddle._srcrr,fiddle._src.codegenr import_manager_libr r namespace_libr rarbr$rir dataclassrr-r?boolrErFrrfrrrrrsH        Q 0