o qoin@s8dZddlZddlZddlmZmZddlZGdddejdZ ej ddGd d d e Z ej ddGd d d e Z ej ddGd dde Z ej ddGddde Zdeeeeee fdeedeefddZej ddGddde Zej ddGddde Zej ddGddde ZdS)zVery minimal AST library. We are transitioning to LibCST, so please don't use this library. It is only for supporting older code. N)ListUnionc@s(eZdZdZejdeefddZdS) CodegenNodezSuper-lightweight codegen node.returncCsdS)zReturns lines of code for this element. If the final lines should be indented, any outer nodes should modify the output of this method. Returns: Lines of code. NselfrrP/home/ubuntu/.local/lib/python3.10/site-packages/fiddle/_src/codegen/mini_ast.pylines s zCodegenNode.linesN) __name__ __module__ __qualname____doc__abcabstractmethodrstrr rrrr rsr) metaclassT)frozenc@s"eZdZdZdeefddZdS)Noopz2No-op node, when emitting something is convenient.rcCsgSNrrrrr r 1sz Noop.linesN)r r r rrrr rrrr r-src@s4eZdZUdZeed<eed<deefddZdS) Assignmentz5Assigns an expression to a left-hand-side expression.lhsexprrcCs|jd|jgS)Nz = )rrrrrr r ;zAssignment.linesNr r r rr__annotations__rr rrrr r5 rc@s,eZdZUdZeed<deefddZdS) ReturnStmtzReturns an expression.rrcCsd|jgS)Nzreturn )rrrrr r DszReturnStmt.linesNrrrrr r?s rc@s4eZdZUdZeed<eed<deefddZdS)TrailingCommentz=Attaches a trailing comment to the last line of another node.childcommentrcCsBd|j}|j}|rg|dd|dd|S|gS)Nz#  )r rr )rr r rrr r Ns  ,zTrailingComment.linesN) r r r rrrrrr rrrr rHrrsub_nodes_or_lines separatorrcCstd|}g}t|D]?\}}|dkr||t|tr#||q t|tr7tdd|Dr7||q t|trD|| q t d||S)aHelper to generate lines for a block of code. Args: sub_nodes_or_lines: Items to join, which are either string constants or codegen nodes. separator: Lines to insert between items to join. Returns: List of lines, taken from `sub_nodes_or_lines`. Nrcss|]}t|tVqdSr) isinstancer).0xrrr iszblock..z'Unsupported item in argument to block: ) filter enumerateextendr%rappendlistallrr TypeError)r#r$resultiitemrrr blockTs      r3c@s0eZdZUdZeeed<deefddZdS)ImmediateAttrsBlockzGBlock of code setting up immediate objects (as opposed to nested ones).nodesrcCs t|jgSr)r3r5rrrr r ws zImmediateAttrsBlock.linesN r r r rrrrrr rrrr r4rs  r4c@s<eZdZUdZeeed<eeed<deefddZdS)SharedThenResultAssignmentz4Block of code setting shared instances, then a tree.shared_instances tree_blocksrcCst|j|jdgS)N)r3r8r9rrrr r rz SharedThenResultAssignment.linesNr6rrrr r7{s   r7c@s>eZdZUdZeejed<eeed<dee fddZ dS) ConfigBuilderzqTop-level node for imports, followed by builder code. By default, this generates a `build_config()` method. imports builder_bodyrcCsLtj|jdj}ddt|jdgD}tt|ggdg|gddgS)N)bodycSsg|]}d|qS)r"r)r&linerrr sz'ConfigBuilder.lines..r:zdef build_config():)cstModuler<code splitlinesr3r=)r import_linesfn_bodyrrr r s zConfigBuilder.linesN) r r r rrrASimpleStatementLinerrrr rrrr r;s  r;)rr dataclassestypingrrlibcstrAABCMetar dataclassrrrrrr3r4r7r;rrrr s2