ó &^Ê[c@sÂdZddlmZmZmZmZmZmZddlm Z m Z ddl m Z mZddl mZddlm Z mZmZm Z mZmZddlZd„Zdd „Zdd „Zdd „Zd „Zd „Zd„Zd„Zddd„Ze je je j e j!e j"fZ#e j$e j%fZ&d„Z'dd„Z(d„Z)d„Z*d„Z+d„Z,d„Z-d„Z.d„Z/dZ0dZ1d„Z2d„Z3dd„Z4dS(så Utility functions from 2to3, 3to2 and python-modernize (and some home-grown ones). Licences: 2to3: PSF License v2 3to2: Apache Software License (from 3to2/setup.py) python-modernize licence: BSD (from python-modernize/LICENSE) iÿÿÿÿ(t FromImporttNewlinet is_importt find_roottdoes_tree_importtComma(tLeaftNode(tpython_symbolstpython_grammar(ttoken(RtCalltNametsymsRtNumberNcCsÂd|kr|S|jdƒr,|d}ng|D]$}|jdj|ƒƒr3|^q3}t|ƒdkr•tddjd„|Dƒƒƒ‚n!t|ƒd kr¶td ƒ‚n|d Sd S( sl Examples: >>> canonical_fix_name('fix_wrap_text_literals') 'libfuturize.fixes.fix_wrap_text_literals' >>> canonical_fix_name('wrap_text_literals') 'libfuturize.fixes.fix_wrap_text_literals' >>> canonical_fix_name('wrap_te') ValueError("unknown fixer name") >>> canonical_fix_name('wrap') ValueError("ambiguous fixer name") s.fix_tfix_isfix_{0}isOAmbiguous fixer name. Choose a fully qualified module name instead from these: s css|]}d|VqdS(s N((t.0tmyf((s5lib/python2.7/site-packages/libfuturize/fixer_util.pys -sis1Unknown fixer. Use --list-fixes or -l for a list.N(t startswithtendswithtformattlent ValueErrortjoin(tfixt avail_fixestftfound((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytcanonical_fix_names   $ cCsttjdd|ƒS(Nu*tprefix(RR tSTAR(R((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytStar6scCsttjdd|ƒS(Nu**R(RR t DOUBLESTAR(R((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt DoubleStar9scCsttjdd|ƒS(Nu-R(RR tMINUS(R((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytMinus<scCs?g}x+|D]#}|j|ƒ|jtƒƒq W|d=|S(s{ Accepts/turns: (Name, Name, ..., Name, Name) Returns/into: (Name, Comma, Name, Comma, ..., Name, Comma, Name) iÿÿÿÿ(tappendR(tleafst new_leafstleaf((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt commatize?s   cCs¬x1|jdk r3|jjtjkr3|j}qW|jdkrGdS|jtjkr`|jS|jdk rŽ|jjtjkrŽ|jjS|jdkr¡dS|j SdS(sf Returns the indentation for this node Iff a node is in a suite, then it has indentation. uN( tparenttNonettypeR tsuiteR tINDENTtvaluet prev_siblingR(tnode((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt indentationKs' $ cCs@t|ƒ}td„|jƒDƒƒ}|s2dSt|ƒSdS(s Dirty little trick to get the difference between each indentation level Implemented by finding the shortest indentation string (technically, the "least" of all of the indentation strings, but tabs and spaces mixed won't get this far, so those are synonymous.) css*|] }|jtjkr|jVqdS(N(R+R R-R.(Rti((s5lib/python2.7/site-packages/libfuturize/fixer_util.pys isu N(Rtsett pre_ordertmin(R0trt all_indents((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytindentation_step`s  cCsåx'|jD]}|jtjkr dSq Wx?t|jƒD]"\}}|jtjkr:Pq:q:Wtdƒ‚ttjt ƒt tj t |ƒt |ƒƒgƒ}|j|d}|jƒd|_|j|ƒ|j|ƒdS(sj Turn the stuff after the first colon in parent's children into a suite, if it wasn't already NuNo class suite and no ':'!iu(tchildrenR+R R,t enumerateR tCOLONRRRRR-R1R8tremoveRt append_child(R)R0R2R,tone_node((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytsuitifyps 7   cCsw|dkrd}ntdd|ƒ|g}|dk rg|jtdddƒt|ddƒgƒnttj|ƒS(s— Accepts a package (Name node), name to import it as (string), and optional prefix and returns a node: import [as ] uuimportRuasu N(R*R textendRR t import_name(tpackagetas_nameRR9((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt NameImportˆs   ccs‡|jtkst‚|j}xC|jtjkrc|j}|jtjkrRPn|V|j}q!W|j}|jtj ks…t‚|j}x|dk r®|V|j}q‘W|j}|jt kr|}x$|jdk ró|jV|j}qÐW|j}n|j}|dkrdSxg|jt kr‚|jtj krE|Vn|j}|dkr|j}|j}|dkrPqqqWdS(s‘ Generator yields all nodes for which a node (an import_stmt) has scope The purpose of this is for a call to _find() on each of them N( R+t _import_stmtstAssertionErrort next_siblingR tSEMItNEWLINER)R t simple_stmtR*t_compound_stmtsR,(R0ttesttnxtR)tcontexttctp((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytimport_binding_scope™sB                cCsgt|ƒ}tdddƒ}t|ddƒ}ttj|||gƒ}|dk rc||_n|S(NuasRu (R RR timport_as_nameR*R(tnameRCRtnew_nametnew_ast new_as_nametnew_node((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt ImportAsNameÔs   cCs>|jtjko=t|jƒdko=|jdjtjkS(s< Returns True if the node appears to be a docstring i(R+R RJRR9R tSTRING(R0((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt is_docstringÞscCs2t|ƒ}td||ƒr"dSd}xvt|jƒD]e\}}t|ƒs\t|ƒre|}nt|ƒrwq8nt|ƒ}|sPn||kr8dSq8Wt dt t j |ddƒgƒ}|dkr|dkr|jdj |_ d|jd_ n|tƒg}|j|ttj|ƒƒdS(s This seems to work u __future__NRt iu(RRR*R:R9tis_shebang_commenttis_encoding_commentRZtcheck_future_importRRR tNAMERRt insert_childRR RJ(tfeatureR0troottshebang_encoding_idxtidxtnamestimport_R9((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt future_importæs(     $c Cs6t|ƒ}td||ƒr"dSd}x_t|jƒD]N\}}|jtjkr8|jr8|jdjtjkr8|d}Pq8q8WxP|j|D];}|jtj krÀ|d7}q˜n|j }d|_ Pq˜Wd}t dt tj |ddƒgƒ}|tƒg}|j|ttj|d|ƒƒdS(sD An alternative to future_import() which might not work ... u __future__NiiuRu (RRR:R9R+R RJR RYRIRRRR_RR`R( RaR0Rbt insert_posRdt thing_afterRRfR9((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytfuture_import2s(     $cCsÔg|D]}|jtjkr|^q}tg|D]}|df^q5ƒ}x}t|ƒD]o\}}|jtjkr¸|jdjtj kr¸|jdj }|jd|| -*- # vim: set fileencoding= : (R“R”R•tENCODING_REGEXR(R0((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyR]és cCsjt|ƒdkst‚t|ƒdkrK|\}}|tƒ|g}n|}tt|ƒ|d|ƒS(sö Example: >>> wrap_in_fn_call("oldstr", (arg,)) oldstr(arg) >>> wrap_in_fn_call("olddiv", (arg1, arg2)) olddiv(arg1, arg2) >>> wrap_in_fn_call("olddiv", [arg1, comma, arg2, comma, arg3]) olddiv(arg1, arg2, arg3) iiR(RRFRR R (tfn_nametargsRtexpr1texpr2tnewargs((s5lib/python2.7/site-packages/libfuturize/fixer_util.pytwrap_in_fn_callös  (5t__doc__tlib2to3.fixer_utilRRRRRRtlib2to3.pytreeRRtlib2to3.pygramRR R R R R RR”RR*RR!R#R(R1R8R?RDtif_stmtt while_stmttfor_stmtttry_stmtt with_stmtRKRARŒRERQRXRZRgRjRuRvR‹R^R–R—R\R]R(((s5lib/python2.7/site-packages/libfuturize/fixer_util.pyt s>..  "      $ ;  (   c *