?dY*dZddlZddlZddlmZddlmZddlmZddlmZddlm Z e gdZ Gd d e Z d Zd Zd Zej"j%Zej(d=ej*j%Zej(d=dZdZdZdZd dZdZdZd!dZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&y)"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 ENDMARKERc$eZdZdZdZdZdZdZdZy) Annotationz)Annotation names associated with pytrees. child_indentnewlines must_split split_penaltysubtypeN) __name__ __module__ __qualname____doc__ CHILD_INDENTNEWLINES MUST_SPLIT SPLIT_PENALTYSUBTYPE:/usr/lib/python3/dist-packages/yapf/pytree/pytree_utils.pyr r +s1, (*!- 'rr c|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 rNodeNamer'4s= YY_ >>$)) $$  . .tyy 99rcjt|tjr|St|jdS)Nr) isinstancerLeaf FirstLeafNodechildrenr%s rr+r+Fs)fkk" K t}}Q' ((rcjt|tjr|St|jdS)N)r)rr* LastLeafNoder,r%s rr/r/Ls)fkk" K dmmB' ((rexecnonlocalc:|jtjs|tjz } tjt t j}|j|d}t|S#tj$r tjtt j}|j|d}nA#tj$r+ tj|#t$r}|d}~wwxYwwxYwYt|SwxYw)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)endswithoslineseprDriver_GRAMMAR_FOR_PY3rr3 parse_stringr ParseError_GRAMMAR_FOR_PY2ast SyntaxError_WrapEndMarker)code parser_drivertreees rParseCodeToTreerD^s rzz "BJJDMM"2FNNKM  % %d% % 8D       mm$4fnnMm  ' 'E ' :d    $      sM JJ,556 ha !&AAo##OX>  EY  r_yapf_annotation_c t|D]0}|jtst||t ||d2y)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 rCopyYapfAnnotationsris;H?j45 c:wsJ=>?rc*t|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. rerc)r&rhdefaults rGetNodeAnnotationrms .;W EErc,t|t|z|y)zSet annotation value on a node. Arguments: node: the node. annotation: annotation name - a string. value: annotation value to set. Nrdrc)r&rhvalues rSetNodeAnnotationrqs $'*4eC :: d^   z'>'>?  AArc lt|tjrI|jjdt |d|j Dcgc] }t|c}dSt|tjr2|jjdt |d|jdSycc}w)zCLike pytree.Node.__repr__, but names instead of numbers for tokens.(z, )N) r)rrF __class__rr'r,rr*rp)r&cs rrrPs~fkk">>22HTN8< F1OA.F HHfkk">>22HTNDJJ OO#Gs B1 cvt|dk(xr*|jdjtjk(S)N simple_stmtr)r'r,r!rCOMMENTr%s rIsCommentStatementrYs3 4.M ) 1 --   5== 02r)F)N)'rr=r6lib2to3rr lib2to3.pgen2rrr frozensetNONSEMANTIC_TOKENSobjectr r'r+r/!python_grammar_no_print_statementcopyr9keywordsr#r<rDr?rPrSrLrcrirmrqrvryr|rrrrrrrrs    KL:$) );;@@Bf%((--/j)%P  - ,! P. ? F= , 6 LHA6P2r