a d(@sdZddlZddlZddlmZddlmZddlmZddlmZddlm Z e gdZ Gd d d e Z d d Zd dZddZejZejd=ddZddZddZddZd2ddZdZddZd3d d!Zd"d#Zd$d%Zd&d'Zd(d)Z d*d+Z!d,d-Z"d.d/Z#d0d1Z$dS)4apytree-related utilities. This module collects various utilities related to the parse trees produced by the lib2to3 library. NodeName(): produces a string name for pytree nodes. ParseCodeToTree(): convenience wrapper around lib2to3 interfaces to parse a given string with code to a pytree. InsertNodeBefore(): insert a node before another in a pytree. InsertNodeAfter(): insert a node after another in a pytree. {Get,Set}NodeAnnotation(): manage custom annotations on pytree nodes. N)pygram)pytree)driver)parse)token)DEDENTINDENTNEWLINE ENDMARKERc@s$eZdZdZdZdZdZdZdZdS) Annotationz)Annotation names associated with pytrees.Z child_indentnewlinesZ must_splitZ split_penaltysubtypeN) __name__ __module__ __qualname____doc__ CHILD_INDENTZNEWLINESZ MUST_SPLIT SPLIT_PENALTYSUBTYPErra/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapf/pytree/pytree_utils.pyr +s r cCs(|jdkrtj|jStjj|jSdS)zProduce a string name for a given node. For a Leaf this is the token name, and for a Node this is the type. Arguments: node: a tree node Returns: Name as a string. N)typertok_namerZpython_grammarZ number2symbolnoderrrNodeName4s  rcCst|tjr|St|jdS)Nr) isinstancerLeaf FirstLeafNodechildrenrrrrrFs rcCst|tjr|St|jdS)N)rrr LastLeafNoder rrrrr"Ls r"execcCsb|tjs|tj7}z"tjttjd}|j|dd}Wn t j yXt |Yn0t |S)aParse the given code to a lib2to3 pytree. Arguments: code: a string with the code to parse. Raises: SyntaxError if the code is invalid syntax. parse.ParseError if some other parsing failure. Returns: The root node of the parsed tree. )convertF)debug) endswithoslineseprZDriver_PYTHON_GRAMMARrr$ parse_stringr ParseErrorast_WrapEndMarker)codeZ parser_drivertreerrrParseCodeToTree]s   r0cCs.t|tjr*|jtjkr*ttjj |gS|S)aSWrap a single ENDMARKER token in a "file_input" node. Arguments: tree: (pytree.Node) The root node of the parsed tree. Returns: The root node of the parsed tree. If the tree is a single ENDMARKER node, then that node is wrapped in a "file_input" node. That will ensure we don't skip comments attached to that node. ) rrrrrr NoderZpython_symbolsZ file_input)r/rrrr-ys r-cCs|D]}t||ddqdS)aTInsert new_nodes before the given target location in the tree. Arguments: new_nodes: a sequence of new nodes to insert (the nodes should not be in the tree). target: the target node before which the new node node will be inserted. Raises: RuntimeError: if the tree is corrupted, or the insertion would corrupt it. FafterN) _InsertNodeAtZ new_nodestargetrrrrInsertNodesBefores r7cCs t|D]}t||ddqdS)aRInsert new_nodes after the given target location in the tree. Arguments: new_nodes: a sequence of new nodes to insert (the nodes should not be in the tree). target: the target node after which the new node node will be inserted. Raises: RuntimeError: if the tree is corrupted, or the insertion would corrupt it. Tr2N)reversedr4r5rrrInsertNodesAfters r9FcCs|jdurtd||jf|j}|dur4td|ft|jD]2\}}||ur>|rZ|dn|}|||dSq>td|fdS)a|Underlying implementation for node insertion. Arguments: new_node: a new node to insert (this node should not be in the tree). target: the target node. after: if True, new_node is inserted after target. Otherwise, it's inserted before target. Returns: nothing Raises: RuntimeError: if the tree is corrupted, or the insertion would corrupt it. Nz)inserting node which already has a parentz%expected target node to have a parentz.unable to find insertion point for target node)parent RuntimeError enumerater Z insert_child)new_noder6r3Zparent_of_targetichildZinsertion_indexrrrr4s   r4Z_yapf_annotation_cCs0t|D]"}|trt||t||dqdS)zCopy all YAPF annotations from the source node to the destination node. Arguments: src: the source node. dst: the destination node. N)dir startswith_NODE_ANNOTATION_PREFIXsetattrgetattr)srcdst annotationrrrCopyYapfAnnotationss  rIcCst|t||S)aCGet annotation value from a node. Arguments: node: the node. annotation: annotation name - a string. default: the default value to return if there's no annotation. Returns: Value of the annotation in the given node. If the node doesn't have this particular annotation name yet, returns default. rErC)rrHdefaultrrrGetNodeAnnotations rLcCst|t||dS)zSet annotation value on a node. Arguments: node: the node. annotation: annotation name - a string. value: annotation value to set. NrDrC)rrHvaluerrrSetNodeAnnotationsrOcCs(t||t}||t|||dS)zAppends an annotation value to a list of annotations on the node. Arguments: node: the node. annotation: annotation name - a string. value: annotation value to set. N)rLsetaddrO)rrHrNattrrrrAppendNodeAnnotations rScCs4t|tj}|r0||vr0||t|tj|dS)zRemoves an annotation value from the subtype annotations on the node. Arguments: node: the node. value: annotation value to remove. N)rLr rremoverO)rrNrRrrrRemoveSubtypeAnnotations   rUcCst|tddS)zGet opening bracket value from a node. Arguments: node: the node. Returns: The opening bracket node or None if it couldn't find one. container_bracketNrJrrrrGetOpeningBrackets rWcCst|td|dS)zoSet opening bracket value for a node. Arguments: node: the node. bracket: opening bracket to set. rVNrM)rZbracketrrrSetOpeningBracketsrXc Cslt|tjrBd}|jt|t||j|jt|j t |t j ddSd}|jt|t |jt |t jdSdS)zDump a string representation of the given node. For debugging. Arguments: node: the node. Returns: The string representation. zV{name}({value}) [lineno={lineno}, column={column}, prefix={prefix}, penalty={penalty}]N)namerNlinenocolumnprefixZpenaltyz1{node} [{len} children] [child_indent="{indent}"])rlenindent)rrrformatr_PytreeNodeReprrZr[reprr\rLr rr]r r)rfmtrrrDumpNodeToString(s   rccCsVt|tjr.d|jjt|dd|jDfSt|tjrRd|jjt||jfSdS)zCLike pytree.Node.__repr__, but names instead of numbers for tokens.z %s(%s, %r)cSsg|] }t|qSr)r`).0crrr Gz#_PytreeNodeRepr..N) rrr1 __class__rrr rrNrrrrr`Cs   r`cCst|dko|jdjtjkS)NZ simple_stmtr)rr rrCOMMENTrrrrIsCommentStatementLs rj)F)N)%rr,r'Zyapf_third_party._ylib2to3rrZ yapf_third_party._ylib2to3.pgen2rrr frozensetZNONSEMANTIC_TOKENSobjectr rrr"Z!python_grammar_no_print_statementcopyr)keywordsr0r-r7r9r4rCrIrLrOrSrUrWrXrcr`rjrrrrs<          (