U A`g; @sdZddlmZddlmZmZddlmZddl m Z ddZ dd Z d d Z d d ZdWddZddZddZddZe e fddZdXddZddZddZdYdd Zd!d"ZdZd#d$Zd[d%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2d3d4d5d6d7d8d9d:h Z d;d<Z!d=a"d>a#d?a$d@a%dAdBZ&dCdDZ'dEdFZ(dGdHZ)dIdJZ*dKdLZ+dMdNZ,dOdPZ-ej.ej/hZ0d\dQdRZ1ej/ej.ej2hZ3dSdTZ4d]dUdVZ5dS)^z1Utility functions, node construction macros, etc.)token)LeafNode)python_symbols)patcompcCsttj|ttjd|gS)N=)rsymsZargumentrrEQUAL)keywordvaluer (/usr/lib/python3.8/lib2to3/fixer_util.py KeywordArgsrcCs ttjdS)N()rrLPARr r r r LParensrcCs ttjdS)N))rrRPARr r r r RParensrcCsHt|ts|g}t|ts&d|_|g}ttj|ttjdddg|S)zBuild an assignment statement rprefix) isinstancelistrrratomrrr )targetsourcer r r Assigns  rNcCsttj||dS)zReturn a NAME leafr)rrNAME)namerr r r Name$sr cCs|ttjt|ggS)zA node tuple for obj.attr)rrtrailerDot)objattrr r r Attr(sr%cCs ttjdS)z A comma leaf,)rrCOMMAr r r r Comma,sr(cCs ttjdS)zA period (.) leaf.)rrDOTr r r r r"0sr"cCs4ttj||g}|r0|dttj||S)z-A parenthesised argument list, used by Call()r)rrr!clone insert_childarglist)argsZlparenZrparennoder r r ArgList4sr0cCs&ttj|t|g}|dk r"||_|S)zA function callN)rrpowerr0r)Z func_namer.rr/r r r Call;sr2cCs ttjdS)zA newline literal rrNEWLINEr r r r NewlineBsr6cCs ttjdS)z A blank liner4r r r r BlankLineFsr8cCsttj||dS)Nr)rrNUMBER)nrr r r NumberJsr;cCs"ttjttjd|ttjdgS)zA numeric or string subscript[])rrr!rrLBRACERBRACE)Z index_noder r r SubscriptMs r@cCsttj||dS)z A string leafr)rrSTRING)stringrr r r StringSsrCc Csd|_d|_d|_ttjd}d|_ttjd}d|_||||g}|rtd|_ttjd}d|_|ttj||gttj|ttj |g}ttj ttj d|ttj dgS)zuA list comprehension of the form [xp for fp in it if test]. If test is None, the "if test" part is omitted. r7rforinifr<r=) rrrrappendrrZcomp_ifZ listmakerZcomp_forrr>r?) ZxpfpitZtestZfor_leafZin_leafZ inner_argsZif_leafinnerr r r ListCompWs(      rKcCsV|D] }|qttjdttj|ddttjdddttj|g}ttj|}|S)zO Return an import statement in the form: from package import name_leafsfromrrimport)removerrrrrimport_as_names import_from)Z package_nameZ name_leafsZleafchildrenimpr r r FromImportos    rSc Cs|d}|jtjkr"|}nttj|g}|d}|rNdd|D}ttjtt|dt|dttj|d||dgg|}|j |_ |S) zfReturns an import statement and calls a method of the module: import module module.name()r#aftercSsg|] }|qSr )r+).0r:r r r sz!ImportAndCall..rZlparZrpar) r+typerr-rr1r%r r!r)r/resultsnamesr#Z newarglistrTnewr r r ImportAndCalls*     r\cCst|tr |jttgkr dSt|tot|jdkot|jdtot|jdtot|jdto|jdjdko|jdjdkS)z(Does the node represent a tuple literal?TrWrrr)rrrQrrlenrr r/r r r is_tuples  racCsXt|toVt|jdkoVt|jdtoVt|jdtoV|jdjdkoV|jdjdkS)z'Does the node represent a list literal?rrWr<r=)rrr_rQrr r`r r r is_lists  rccCsttjt|tgSN)rrrrrr`r r r parenthesizesresortedrsetanyalltuplesumminmax enumerateccs$t||}|r |Vt||}q dS)alFollow an attribute chain. If you have a chain of objects where a.foo -> b, b.foo-> c, etc, use this to iterate over all objects in the chain. Iteration is terminated by getattr(x, attr) is None. Args: obj: the starting object attr: the name of the chaining attribute Yields: Each successive object in the chain. N)getattr)r#r$nextr r r attr_chains rqzefor_stmt< 'for' any 'in' node=any ':' any* > | comp_for< 'for' any 'in' node=any any* > z power< ( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' | 'any' | 'all' | 'enumerate' | (any* trailer< '.' 'join' >) ) trailer< '(' node=any ')' > any* > z` power< ( 'sorted' | 'enumerate' ) trailer< '(' arglist ')' > any* > FcCspts&ttattattadatttg}t|t|dD]*\}}i}|||r@|d|kr@dSq@dS)a Returns true if node is in an environment where all that is required of it is being iterable (ie, it doesn't matter if it returns a list or an iterator). See test_map_nochange in test_fixers.py for some examples and tests. Tparentr/F) pats_builtrZcompile_patternp0p1p2ziprqmatch)r/ZpatternspatternrrrYr r r in_special_contexts    rzcCs|j}|dk r|jtjkrdS|j}|jtjtjfkr:dS|jtjkrX|j d|krXdS|jtj ks|jtj kr|dk r|jtj ks|j d|krdSdS)zG Check that something isn't an attribute or function name etc. NFrWT) Z prev_siblingrXrr*rrrfuncdefclassdef expr_stmtrQZ parametersZ typedargslistr')r/prevrrr r r is_probably_builtins&    rcCsJ|dk rF|jtjkr>t|jdkr>|jd}|jtjkr>|jS|j}qdS)zFind the indentation of *node*.Nr^rr7) rXrsuiter_rQrINDENTr rr)r/indentr r r find_indentations  rcCs>|jtjkr|S|}|jd}|_ttj|g}||_|Srd)rXrrr+rrr)r/rrrr r r make_suites rcCs$|jtjkr |j}|stdq|S)zFind the top level namespace.z,root found before file_input node was found.)rXrZ file_inputrr ValueErrorr`r r r find_root&s   rcCst|t||}t|S)z Returns true if name is imported from package at the top level of the tree which node belongs to. To cover the case of an import like 'import foo', use None for the package and 'foo' for the name. ) find_bindingrbool)packagerr/Zbindingr r r does_tree_import/srcCs|jtjtjfkS)z0Returns true if the node is an import statement.)rXr import_namerPr`r r r is_import7src Cs.dd}t|}t|||r dSd}}t|jD]F\}}||sDq2t|j|dD]\}}||sVqlqV||}qzq2|dkrt|jD]8\}}|jtjkr|jr|jdjtjkr|d}qq|dkrt tj t tj dt tj |ddg} nt |t tj |ddg} | tg} ||t tj| dS) z\ Works like `does_tree_import` but adds an import statement if it was not imported. cSs |jtjko|jot|jdS)NrW)rXr simple_stmtrQrr`r r r is_import_stmt>s z$touch_import..is_import_stmtNrWrrMrr)rrrnrQrXrrrrArrrrrSr6r,) rrr/rrootZ insert_posoffsetidxZnode2import_rQr r r touch_import;s8   rcCs|jD]}d}|jtjkrVt||jdr4|St|t|jd|}|rR|}n0|jtjtjfkrt|t|jd|}|r|}n|jtj krt|t|jd|}|r|}nTt |jddD]@\}}|jt j kr|j dkrt|t|j|d|}|r|}qnx|jtkr2|jdj |kr2|}nTt|||rF|}n@|jtjkrbt|||}n$|jtjkrt||jdr|}|r|s|St|r|SqdS) z Returns the node which binds variable name, otherwise None. If optional argument package is supplied, only imports will be returned. See test cases for examples.Nrrbr^r]:rW)rQrXrZfor_stmt_findrrZif_stmtZ while_stmtZtry_stmtrnrCOLONr _def_syms_is_import_bindingrr}r)rr/rchildZretr:iZkidr r r risH   rcCsT|g}|rP|}|jdkr4|jtkr4||jq|jtjkr|j|kr|SqdS)N)poprX _block_symsextendrQrrr )rr/Znodesr r r rsrcCs|jtjkr|s|jd}|jtjkrx|jD]H}|jtjkrV|jdj|krt|Sq,|jtjkr,|j|kr,|Sq,nL|jtjkr|jd}|jtjkr|j|kr|Sn|jtjkr|j|kr|Sn|jtj kr|rt |jd |krdS|jd}|rt d|rdS|jtj kr0t ||r0|S|jtjkrh|jd}|jtjkr|j|kr|Sn6|jtjkr|j|kr|S|r|jtjkr|SdS)z Will reuturn node if node will import name, or node will import * from package. None is returned otherwise. See test cases for examples. rr^rbNr]as)rXrrrQZdotted_as_namesZdotted_as_namer rrrPstrstriprrOZimport_as_nameSTAR)r/rrrRrZlastr:r r r rs@          r)N)NN)N)N)N)N)N)6__doc__Zpgen2rZpytreerrZpygramrrr7rrrrrr r%r(r"r0r2r6r8r;r@rCrKrSr\rarcreZconsuming_callsrqrtrurvrsrzrrrrrrrr|r{rrr!rrrr r r r s`           -  *