a d;@sdZddlmZddlmZddlmZddlmZddZdd d Z e gd Z d d Z e dge BZ ddZddZddZdS)apComment splicer for lib2to3 trees. The lib2to3 syntax tree produced by the parser holds comments and whitespace in prefix attributes of nodes, rather than nodes themselves. This module provides functionality to splice comments out of prefixes and into nodes of their own, making them easier to process. SpliceComments(): the main function exported by this module. )pygram)pytree)token) pytree_utilscs(dgt|fdd|dS)a4Given a pytree, splice comments into nodes of their own right. Extract comments from the prefixes where they are housed after parsing. The prefixes that previously housed the comments become empty. Args: tree: a pytree.Node - the tree to work on. The tree is modified by this function. Nc s|jddD]}t|tjr*|q|jdr|j}|j|d}|j }|jd}|d| d}d|vr|| ddd}d|_|j t jkr|t|8}tt|||dddn|j t jkrg}d}|dD]t}| d} | dkr$|durX|d7}qn4|dus8| |krX| }|d|} ||| gf|d d |q|D]^\}} } t|| } | j t jkrtj} ntj} | td| d||d d| |t| 7}qpnt|}|D]}|j t jkrqqt|t|krxt|}t|jd vrV||dd}||jdkrV|j}tt||dd d|qq|djkr0| }|d}|dr0dj }|tdj!7}|t|dt|d7}tj"t j#|dd||ffd }t|gdd|dd}|d7}d|vrBdn|$dd}t||dt||d}t|||dd}t||qq|d<qdS) zclassdeffuncdeftypevaluecontext)%children isinstancerNodeprefixlstrip startswithlinenocountcolumnfindrfindrrNEWLINElenrZInsertNodesAfter_CreateCommentsFromPrefixDEDENTsplitappend_FindAncestorAtIndentZInsertNodesBeforejoin_FindStmtParentZleavesid!_FindNodeWithStandaloneLineParentNodeNameparentrstrip splitlinesrLeafCOMMENTrindex)nodechildcomment_prefixcomment_linenocomment_columnZ child_prefixZ prefix_indentZcomment_groupsZcmtcolZcomment_indentZ comment_groupZancestor_at_indentZ InsertNodesZ stmt_parentZleaf_in_parentZnode_with_line_parentZ comment_endZ comment_linesr comment_leafr.comments _VisitNodeRecZ prev_leafd/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/yapf/pytree/comment_splicer.pyr8.s                  z%SpliceComments.._VisitNodeRec)_AnnotateIndents)treer9r7r:SpliceCommentss r=Fc Csg}|d}d}|t|krg}|t|kr\||dr\||||d7}q"|r||d}|d|d<|d|d<tjtj d |d||ffd} |s| nt t j j| g} || |t|kr||s|d7}qq|S)aGCreate pytree nodes to represent the given comment prefix. Args: comment_prefix: (unicode) the text of the comment from the node's prefix. comment_lineno: (int) the line number for the start of the comment. comment_column: (int) the column for the start of the comment. standalone: (bool) determines if the comment is standalone or not. Returns: The simple_stmt nodes if this is a standalone comment, otherwise a list of new COMMENT leafs. The prefix may consist of multiple comment blocks, separated by blank lines. Each block gets its own leaf. rrrrr r r)r!rrrr"striprr,rr-r$rrZpython_symbols simple_stmt) r1r2r3r r6linesindexZ comment_blockZ new_linenor5Z comment_noder9r9r:rs0         r) ZsuiteZif_stmtZ while_stmtZfor_stmtZtry_stmtZ with_stmtr r Z decoratedZ file_inputcCs"t|jtvr|St|jSdS)zFind a node whose parent is a 'standalone line' node. See the comment above _STANDALONE_LINE_NODES for more details. Arguments: node: node to start from Returns: Suitable node that's either the node itself or one of its ancestors. N)rr(r)_STANDALONE_LINE_NODESr'r/r9r9r:r's r'r?cCs t|tvr|St|jSdS)zFind the nearest parent of node that is a statement node. Arguments: node: node to start from Returns: Nearest parent (or node itself, if suitable). N)rr(_STATEMENT_NODESr%r)rCr9r9r:r%'s r%cCsH|jjdur|St|jtjj}|dur8||r8|St|j|SdS)aFind an ancestor of node with the given indentation. Arguments: node: node to start from. This must not be the tree root. indent: indentation string for the ancestor we're looking for. See _AnnotateIndents for more details. Returns: An ancestor node with suitable indentation. If no suitable ancestor is found, the closest ancestor to the tree root is returned. N)r)rGetNodeAnnotation Annotation CHILD_INDENTrr#)r/indentZ parent_indentr9r9r:r#6s  r#cCs|jdurt|tjjd|jD]\}|jtjkrvt |tjj}|durb||j krbt d||ft|tjj|j t |q"dS)aeAnnotate the tree with child_indent annotations. A child_indent annotation on a node specifies the indentation (as a string, like " ") of its children. It is inferred from the INDENT child of a node. Arguments: tree: root of a pytree. The pytree is modified to add annotations to nodes. Raises: RuntimeError: if the tree is malformed. Nr z"inconsistent indentation for child) r)rZSetNodeAnnotationrFrGrrrINDENTrEr RuntimeErrorr;)r<r0Z child_indentr9r9r:r;Us     r;N)F)__doc__Zyapf_third_party._ylib2to3rrZ yapf_third_party._ylib2to3.pgen2rZ yapf.pytreerr=r frozensetrBr'rDr%r#r;r9r9r9r:s    3 ;