o bcf;@sdZddlmZddlmZmZddlmZddl m Z ddZ dd Z d d Z d d ZdNddZddZddZddZe e fddZdOddZddZddZdNdd Zd!d"ZdNd#d$ZdNd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zhd1Z d2d3Z!d4a"d5a#d6a$d7a%d8d9Z&d:d;Z'dd?Z)d@dAZ*dBdCZ+dDdEZ,dFdGZ-ej.ej/hZ0dNdHdIZ1ej/ej.ej2hZ3dJdKZ4dNdLdMZ5dS)Pz1Utility functions, node construction macros, etc.)token)LeafNode)python_symbols)patcompcCsttj|ttjd|gS)N=)rsymsargumentrrEQUAL)keywordvaluer )/usr/lib/python3.10/lib2to3/fixer_util.py KeywordArgsrcC ttjdS)N()rrLPARr r r rLParen rcCr)N))rrRPARr r r rRParenrrcCsHt|ts|g}t|tsd|_|g}ttj|ttjdddg|S)zBuild an assignment statement rprefix) isinstancelistrrratomrrr )targetsourcer r rAssigns  r NcCttj||dS)zReturn a NAME leafr)rrNAME)namerr r rName$r$cCs|ttjt|ggS)zA node tuple for obj.attr)rrtrailerDot)objattrr r rAttr(sr*cCr)z A comma leaf,)rrCOMMAr r r rComma, r-cCr)zA period (.) leaf.)rrDOTr r r rr'0r.r'cCs4ttj||g}|r|dttj||S)z-A parenthesised argument list, used by Call()r)rrr&clone insert_childarglist)argslparenrparennoder r rArgList4sr8cCs&ttj|t|g}|dur||_|S)zA function callN)rrpowerr8r) func_namer4rr7r r rCall;sr;cCr)zA newline literal rrNEWLINEr r r rNewlineBr.r?cCr)z A blank liner=r r r r BlankLineFr.rAcCr!)Nr)rrNUMBER)nrr r rNumberJsrDcCs"ttjttjd|ttjdgS)zA numeric or string subscript[])rrr&rrLBRACERBRACE) index_noder r r SubscriptMs rJcCr!)z A string leafr)rrSTRING)stringrr r rStringSr%rMc Csd|_d|_d|_ttjd}d|_ttjd}d|_||||g}|r:d|_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. r@rforinifrErF) rrrr"appendrrcomp_if listmakercomp_forrrGrH) xpfpittestfor_leafin_leaf inner_argsif_leafinnerr r rListCompWs(      r^cCsV|D]}|qttjdttj|ddttjdddttj|g}ttj|}|S)zO Return an import statement in the form: from package import name_leafsfromrrimport)removerrr"rrimport_as_names import_from) package_name name_leafsleafchildrenimpr r r FromImportos    ric Cs|d}|jtjkr|}n ttj|g}|d}|r'dd|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 )r1).0rCr r r sz!ImportAndCall..rlparrpar) r1typerr3rr9r*r$r&r)r7resultsnamesr( newarglistrjnewr r r ImportAndCalls*     rucCst|tr|jttgkrdSt|toCt|jdkoCt|jdtoCt|jdtoCt|jdtoC|jdjdkoC|jdjdkS)z(Does the node represent a tuple literal?Trmrrr)rrrgrrlenrr r7r r ris_tuples  rzcCsXt|to+t|jdko+t|jdto+t|jdto+|jdjdko+|jdjdkS)z'Does the node represent a list literal?rrmrErF)rrrxrgrr ryr r ris_lists  r|cCsttjt|tgSN)rrrrrryr r r parenthesizesr~> allanymaxminsetsumrtuplesorted enumerateccs,t||}|r|Vt||}|sdSdS)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    rzefor_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* > FcCsptsttattattadatttg}t|t|dD]\}}i}|||r5|d|ur5dSq 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. Tparentr7F) pats_builtrcompile_patternp0p1p2ziprmatch)r7patternspatternrrqr r rin_special_contexts    rcCs|j}|dur|jtjkrdS|j}|jtjtjfvrdS|jtjkr,|j d|ur,dS|jtj ksI|jtj krK|durB|jtj ksI|j d|urKdSdS)zG Check that something isn't an attribute or function name etc. NFrmT) prev_siblingrprr0rrfuncdefclassdef expr_stmtrg parameters typedargslistr,)r7prevrr r ris_probably_builtins  rcCsP|dur&|jtjkrt|jdkr|jd}|jtjkr|jS|j}|dusdS)zFind the indentation of *node*.Nrwrr@) rprsuiterxrgrINDENTr r)r7indentr r rfind_indentations  rcCs>|jtjkr|S|}|jd}|_ttj|g}||_|Sr})rprrr1rr)r7rrr r r make_suites rcCs.|jtjkr|j}|std|jtjks|S)zFind the top level namespace.z,root found before file_input node was found.)rpr file_inputr ValueErrorryr 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)packager#r7bindingr r rdoes_tree_import/srcCs|jtjtjfvS)z0Returns true if the node is an import statement.)rpr import_namercryr r r is_import7src Cs*dd}t|}t|||rdSd}}t|jD]!\}}||s"qt|j|dD] \}}||s5nq+||}|dkrat|jD]\}}|jtjkr`|jr`|jdjtjkr`|d}nqD|durxt tj t tj dt tj |ddg} n t |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)Nrm)rpr simple_stmtrgrryr r ris_import_stmt>s z$touch_import..is_import_stmtNrmrr`rr)rrrrgrprrrrKrrrr"rir?r2) rr#r7rroot insert_posoffsetidxnode2import_rgr r r touch_import;s:   rcCs|jD]}d}|jtjkr)t||jdr|St|t|jd|}|r(|}n|jtjtjfvrBt|t|jd|}|rA|}nw|jtj krt|t|jd|}|rX|}nat |jddD] \}}|jt j kr|j dkrt|t|j|d|}|r|}qan6|jtvr|jdj |kr|}n&t|||r|}n|jtjkrt|||}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.Nrr{rwrv:rm)rgrprfor_stmt_findrrif_stmt while_stmttry_stmtrrCOLONr _def_syms_is_import_bindingrrr)r#r7rchildretrCikidr r rrisL      rcCsV|g}|r)|}|jdkr|jtvr||jn |jtjkr'|j|kr'|S|sdS)N)poprp _block_symsextendrgrr"r )r#r7nodesr r rrsrcCs|jtjkrf|sf|jd}|jtjkr=|jD]$}|jtjkr+|jdj|kr*|Sq|jtjkr:|j|kr:|SqdS|jtjkrW|jd}|jtjkrU|j|krU|SdS|jtjkrd|j|krd|SdS|jtj kr|r{t |jd |kr{dS|jd}|rt d|rdS|jtj krt ||r|S|jtjkr|jd}|jtjkr|j|kr|SdS|jtjkr|j|kr|S|r|jtjkr|SdS)z Will return node if node will import name, or node will import * from package. None is returned otherwise. See test cases for examples. rrwr{Nrvas)rprrrgdotted_as_namesdotted_as_namer rr"rcstrstriprrbimport_as_nameSTAR)r7r#rrhrlastrCr r rrsL         rr})NN)6__doc__pgen2rpytreerrpygramrrr@rrrrr r$r*r-r'r8r;r?rArDrJrMr^rirurzr|r~consuming_callsrrrrrrrrrrrrrrrrrr&rrrr r r rsZ            - *