a d@sVdZddlZddlmZddlmZGdddeZejfddZ Gd d d eZ dS) aGeneric visitor pattern for pytrees. The lib2to3 parser produces a "pytree" - syntax tree consisting of Node and Leaf types. This module implements a visitor pattern for such trees. It also exports a basic "dumping" visitor that dumps a textual representation of a pytree into a stream. PyTreeVisitor: a generic visitor pattern for pytrees. PyTreeDumper: a configurable "dumper" for displaying pytrees. DumpPyTree(): a convenience function to dump a pytree. N)pytree) pytree_utilsc@s(eZdZdZddZddZddZdS) PyTreeVisitoraVisitor pattern for pytree trees. Methods named Visit_XXX will be invoked when a node with type XXX is encountered in the tree. The type is either a token type (for Leaf nodes) or grammar symbols (for Node nodes). The return value of Visit_XXX methods is ignored by the visitor. Visitors can modify node contents but must not change the tree structure (e.g. add/remove children and move nodes around). This is a very common visitor pattern in Python code; it's also used in the Python standard library ast module for providing AST visitors. Note: this makes names that aren't style conformant, so such visitor methods need to be marked with # pylint: disable=invalid-name We don't have a choice here, because lib2to3 nodes have under_separated names. For more complex behavior, the visit, DefaultNodeVisit and DefaultLeafVisit methods can be overridden. Don't forget to invoke DefaultNodeVisit for nodes that may have children - otherwise the children will not be visited. cCsPdt|}t||r*t|||n"t|tjrB||n | |dS)z Visit a node.z Visit_{0}N) formatrZNodeNamehasattrgetattr isinstancerZLeafDefaultLeafVisitDefaultNodeVisit)selfnodemethodrc/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapf/pytree/pytree_visitor.pyVisit9s    zPyTreeVisitor.VisitcCs|jD]}||qdS)zDefault visitor for Node: visits the node's children depth-first. This method is invoked when no specific visitor for the node is defined. Arguments: node: the node to visit N)childrenr)r r childrrrr Es zPyTreeVisitor.DefaultNodeVisitcCsdS)zDefault visitor for Leaf: no-op. This method is invoked when no specific visitor for the leaf is defined. Arguments: leaf: the leaf to visit Nrr Zleafrrrr PszPyTreeVisitor.DefaultLeafVisitN)__name__ __module__ __qualname____doc__rr r rrrrr"s  rcCst|}||dS)azConvenience function for dumping a given pytree. This function presents a very minimal interface. For more configurability (for example, controlling how specific node types are displayed), use PyTreeDumper directly. Arguments: tree: the tree to dump. target_stream: the stream to dump the tree to. A file-like object. By default will dump into stdout. N) PyTreeDumperr)tree target_streamZdumperrrr DumpPyTree[s rcs>eZdZdZejfddZddZfddZdd Z Z S) rzVVisitor that dumps the tree to a stream. Implements the PyTreeVisitor interface. cCs||_d|_dS)zCreate a tree dumper. Arguments: target_stream: the stream to dump the tree to. A file-like object. By default will dump into stdout. rN)_target_stream_current_indent)r rrrr__init__qszPyTreeDumper.__init__cCs|jdd|j|dS)Nz{0}{1}  )rwriterr)r srrr _DumpString{szPyTreeDumper._DumpStringcs@|t||jd7_tt|||jd8_dS)N)r"rDumpNodeToStringrsuperrr )r r  __class__rrr ~szPyTreeDumper.DefaultNodeVisitcCs|t|dS)N)r"rr$rrrrr szPyTreeDumper.DefaultLeafVisit) rrrrsysstdoutrr"r r __classcell__rrr&rrks   r) rr(Zyapf_third_party._ylib2to3rZ yapf.pytreerobjectrr)rrrrrrs    9