`N`p*dZddlZddlmZddlmZddlmZddlmZddlmZe gdZ e hd Z e hd Z Gd d e Zd ZdZdZejZejd=ejZejd=dZdZdZdZd#dZdZdZd$dZdZ dZ!dZ"dZ#dZ$d Z%d!Z&d"Z'dS)%apytree-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 ENDMARKER>([{>)]}c&eZdZdZdZdZdZdZdZdS) Annotationz)Annotation names associated with pytrees. child_indentnewlines must_split split_penaltysubtypeN) __name__ __module__ __qualname____doc__ CHILD_INDENTNEWLINES MUST_SPLIT SPLIT_PENALTYSUBTYPE9lib/python3.11/site-packages/yapf/yapflib/pytree_utils.pyrr-s-11, (*!- '''r#rc~|jdkrtj|jStjj|jS)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. )typertok_namerpython_grammar number2symbolnodes r$NodeNamer-6s1 Y__ >$) $$  .ty 99r#cnt|tjr|St|jdS)Nr) isinstancerLeaf FirstLeafNodechildrenr+s r$r1r1Hs0fk"" K t}Q' ( ((r#cnt|tjr|St|jdS)N)r/rr0 LastLeafNoder2r+s r$r5r5Ns0fk"" K dmB' ( ((r#execnonlocalc tjttj}||d}n#t j$r~ tjttj}||d}n=#t j$r+ tj|#t$r}|d}~wwxYwwxYwYnwxYwt|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)debugN) rDriver_GRAMMAR_FOR_PY3rr9 parse_stringr ParseError_GRAMMAR_FOR_PY2ast SyntaxError_WrapEndMarker)code parser_drivertreees r$ParseCodeToTreerG`sM"2FNKKKM  % %d% % 8 8DD      m$4fnMMMm  ' 'E ' : :dd   $   d    sK J JJ,566  ha !&-AAo##OX>>> ff EY   r#_yapf_annotation_c t|D]<}|tr t||t ||d=dS)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 annotations r$CopyYapfAnnotationsrlsYHH??j455? c:wsJ==>>>??r#c4t|t|z|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. rhrf)r,rkdefaults r$GetNodeAnnotationrps .;W E EEr#c8t|t|z|dS)zSet annotation value on a node. Arguments: node: the node. annotation: annotation name - a string. value: annotation value to set. Nrgrf)r,rkvalues r$SetNodeAnnotationrts" $'*4e<<<<C :: d^^    z'>??  A AAr#ct|tjr1|jjdt |dd|jDdSt|tjr'|jjdt |d|jdSdS)zCLike pytree.Node.__repr__, but names instead of numbers for tokens.r z, c,g|]}t|Sr")r).0cs r$ z#_PytreeNodeRepr..Ss FFF1OA..FFFr#rN) r/rrI __class__rr-r2r0rsr+s r$rrOsfk""H>222HTNNNNFF FFFFF HHfk""P>222HTNNNNDJJJ OOPPr#cht|dko|jdjtjkS)N simple_stmtr)r-r2r'rCOMMENTr+s r$IsCommentStatementrXs. 4..M ) 1 -  5= 02r#)F)N)(rr@lib2to3rr lib2to3.pgen2rrr frozensetNONSEMANTIC_TOKENSOPENING_BRACKETSCLOSING_BRACKETSobjectrr-r1r5!python_grammar_no_print_statementcopyr<keywordsr)r?rGrBrSrVrOrfrlrprtryr|rrrrrr"r#r$rs]     YKKKLL9___--9___--:::$))) )));@@BBf%(--//j)"""J    - - - , , ,! ! ! ! P. ? ? ? F F F F=== , , , 6 6 6 L L LHHHAAA6PPP22222r#