; Ðâ"Ic @sXdZddlmZddlmZmZddlmZddl m Z d„Z d„Z d„Z d „Zd0d „Zd „Zd „Zd „Ze ƒe ƒd„Zd0d0d„Zd„Zd„Zd0d„Zd„Zd0d„Zd0d„Zd„Zd„Zd„Zy eZWne k od„ZYnXy e!Z!Wn$e k oddl"m#Z!YnXy e$Z$Wne k od„Z$YnXe!dddd d!d"d#d$d%g ƒZ%d&„Z&d'a'd(a(d)a)d1a+d*„Z,d+„Z-d,„Z.e!ej/ej0gƒZ1d0d-„Z2e!ej0ej/ej3gƒZ4d.„Z5d0d/„Z6d0S(2u1Utility functions, node construction macros, etc.i(utoken(uLeafuNode(upython_symbols(upatcompcCs%ttj|ttjdƒ|gƒS(Nu=(uNodeusymsuargumentuLeafutokenuEQUAL(ukeyworduvalue((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu KeywordArgs cCsttjdƒS(Nu((uLeafutokenuLPAR(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuLParenscCsttjdƒS(Nu)(uLeafutokenuRPAR(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuRParenscCstt|tƒp |g}nt|tƒp|jdƒ|g}nttj|ttjdddƒg|ƒS(uBuild an assignment statementu u=uprefix( u isinstanceulistu set_prefixuNodeusymsuatomuLeafutokenuEQUAL(utargetusource((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuAssigns    cCsttj|d|ƒS(uReturn a NAME leafuprefix(uLeafutokenuNAME(unameuprefix((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuName$scCs|ttjtƒ|gƒgS(uA node tuple for obj.attr(uNodeusymsutraileruDot(uobjuattr((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuAttr(scCsttjdƒS(u A comma leafu,(uLeafutokenuCOMMA(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuComma,scCsttjdƒS(uA period (.) leafu.(uLeafutokenuDOT(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuDot0scCsOttj|jƒ|jƒgƒ}|o |jdttj|ƒƒn|S(u-A parenthesised argument list, used by Call()i(uNodeusymsutrailerucloneu insert_childuarglist(uargsulparenurparenunode((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuArgList4s$ cCs@ttj|t|ƒgƒ}|dk o|j|ƒn|S(uA function callN(uNodeusymsupoweruArgListuNoneu set_prefix(u func_nameuargsuprefixunode((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuCall;s cCsttjdƒS(uA newline literalu (uLeafutokenuNEWLINE(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuNewlineBscCsttjdƒS(u A blank lineu(uLeafutokenuNEWLINE(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu BlankLineFscCsttj|d|ƒS(Nuprefix(uLeafutokenuNUMBER(unuprefix((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuNumberJscCs1ttjttjdƒ|ttjdƒgƒS(uA numeric or string subscriptu[u](uNodeusymsutraileruLeafutokenuLBRACEuRBRACE(u index_node((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu SubscriptMscCsttj|d|ƒS(u A string leafuprefix(uLeafutokenuSTRING(ustringuprefix((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuStringSsc Cs"|jdƒ|jdƒ|jdƒttjdƒ}|jdƒttjdƒ}|jdƒ||||g}|oO|jdƒttjdƒ}|jdƒ|jttj||gƒƒnttj|ttj |ƒgƒ}ttj ttj dƒ|ttj dƒgƒS(uuA list comprehension of the form [xp for fp in it if test]. If test is None, the "if test" part is omitted. uu uforuinuifu[u]( u set_prefixuLeafutokenuNAMEuappenduNodeusymsucomp_ifu listmakerucomp_foruatomuLBRACEuRBRACE( uxpufpuitutestufor_leafuin_leafu inner_argsuif_leafuinner((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuListCompWs$       #$ c Csx|D]}|jƒqWttjdƒttj|ddƒttjdddƒttj|ƒg}ttj|ƒ}|S(uO Return an import statement in the form: from package import name_leafsufromuprefixu uimport(uremoveuLeafutokenuNAMEuNodeusymsuimport_as_namesu import_from(u package_nameu name_leafsuleafuchildrenuimp((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu FromImportoscCsÇt|tƒo!|jtƒtƒgkodSt|tƒo†t|jƒdkopt|jdtƒoYt|jdtƒoBt|jdtƒo+|jdjdko|jdjdkS(u(Does the node represent a tuple literal?iiiiu(u)T( u isinstanceuNodeuchildrenuLParenuRParenuTrueulenuLeafuvalue(unode((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuis_tuple‡s,cCst|tƒoot|jƒdkoYt|jdtƒoBt|jdtƒo+|jdjdko|jdjdkS(u'Does the node represent a list literal?iiiÿÿÿÿu[u](u isinstanceuNodeulenuchildrenuLeafuvalue(unode((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuis_list“s cCs!x|D]}|odSqWdS(NTF(uTrueuFalse(uluo((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuany¤s  i(uSetcCs|ddd…S(Niÿÿÿÿ((ul((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyureversed²susortedulistusetuanyuallutupleusumuminumaxccs6t||ƒ}x |o|Vt||ƒ}qWdS(ulFollow 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(ugetattr(uobjuattrunext((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu attr_chain½s uefor_stmt< 'for' any 'in' node=any ':' any* > | comp_for< 'for' any 'in' node=any any* > u¥ power< ( 'iter' | 'list' | 'tuple' | 'sorted' | 'set' | 'sum' | 'any' | 'all' | (any* trailer< '.' 'join' >) ) trailer< '(' node=any ')' > any* > uN power< 'sorted' trailer< '(' arglist ')' > any* > cCs©tp7tjtƒatjtƒatjtƒadantttg}xUt|t|dƒƒD];\}}i}|j ||ƒo|d|kodSqfWdS(u Returns true if node is in an environment where all that is required of it is being itterable (ie, it doesn't matter if it returns a list or an itterator). See test_map_nochange in test_fixers.py for some examples and tests. uparentunodeTF( u pats_builtupatcompucompile_patternup1up0up2uTrueuzipu attr_chainumatchuFalse(unodeupatternsupatternuparenturesults((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyuin_special_contextãs  $ cCsY|jtjko|S|jƒ}|jd}|_ttj|gƒ}||_|S(N(utypeusymsusuiteucloneuparentuNoneuNode(unodeuparentusuite((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu make_suiteús  cCs@x!|jtjko |j}qWt|||ƒ}t|ƒS(uÞ 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 Null for the package and 'foo' for the name. (utypeusymsu file_inputuparentu find_bindingubool(upackageunameunodeubinding((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyudoes_tree_imports cCs„x}|jD]r}d}|jtjkoPt||jdƒo|St|t|jdƒ|ƒ}|o |}qDnÌ|jtjtj fko4t|t|jdƒ|ƒ}|o |}qDn||jtj ko¹t|t|jdƒ|ƒ}|o |}qDx5t |jdd…ƒD]g\}}|jt j koH|jdko8t|t|j|dƒ|ƒ}|o |}qq&q&Wn°|jtko!|jdj|ko |}nt|||ƒo |}nb|jtjkot|||ƒ}n9|jtjko%t||jdƒo |}qDn|o1|p|S|jtjtjfko|Sq q WdS( uº 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.iiÿÿÿÿiiNu:ii(uchildrenuNoneutypeusymsufor_stmtu_findu find_bindingu make_suiteuif_stmtu while_stmtutry_stmtu enumerateutokenuCOLONuvalueu _def_symsu_is_import_bindingu simple_stmtu expr_stmtu import_nameu import_from(unameunodeupackageuchilduretunuiukid((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu find_bindingsL   ##'   cCs„|g}xt|ol|jƒ}|jdko$|jtko|j|jƒq |jtjko|j|ko|Sq WdS(Ni( upoputypeu _block_symsuextenduchildrenutokenuNAMEuvalueuNone(unameunodeunodes((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu_find=s   # cCsQ|jtjko| o |jd}|jtjkosxä|jD]a}|jtjko |jdj|ko|SqE|jtjko|j|ko|SqEWqM|jtjko9|jd}|jtjko|j|ko|SqM|jtjko|j|ko|Sn(|jtj ko|o%t |jdƒj ƒ|kodS|jd}|ot d|ƒodS|jtjkot ||ƒo|S|jtjko9|jd}|jtjko|j|ko|SqM|jtjko|j|ko|S|o|jtjko|SndS(u› Will reuturn node if node will import name, or node will import * from package. None is returned otherwise. See test cases for examples. iiiÿÿÿÿiuasN(utypeusymsu import_nameuchildrenudotted_as_namesudotted_as_nameuvalueutokenuNAMEu import_fromustrustripuNoneu_finduimport_as_namesuimport_as_nameuSTAR(unodeunameupackageuimpuchildulastun((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyu_is_import_bindingGsB   #  # # ' # # # NF(7u__doc__upgen2utokenupytreeuLeafuNodeupygramupython_symbolsusymsuupatcompu KeywordArguLParenuRParenuAssignuNoneuNameuAttruCommauDotuArgListuCalluNewlineu BlankLineuNumberu SubscriptuStringuListCompu FromImportuis_tupleuis_listuanyu NameErrorusetusetsuSetureverseduconsuming_callsu attr_chainup0up1up2uFalseu pats_builtuin_special_contextu make_suiteudoes_tree_importuclassdefufuncdefu _def_symsu find_bindingutraileru _block_symsu_findu_is_import_binding(((u//mit/python/lib/python3.0/lib2to3/fixer_util.pyusd                      *