← Back
Editing: fixer_util.cpython-33.pyc
� �� f�8 c @ s| d Z d d l m Z d d l m Z d d l m Z m Z d d l m Z d d l m Z d d � Z d d � Z d d � Z d d � Z dU d d � Z d d � Z d d � Z d d � Z e � e � d d � Z dU dU d d � Z d d � Z d d � Z dU d d! � Z d" d# � Z dU d$ d% � Z dU d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z! e" d0 d1 d2 d3 d4 d5 d6 d7 d8 d9 g � Z# d: d; � Z$ d< a% d= a& d> a' dV a) d? d@ � Z* dA dB � Z+ dC dD � Z, dE dF � Z- dG dH � Z. dI dJ � Z/ dK dL � Z0 dM dN � Z1 e" e j2 e j3 g � Z4 dU dO dP � Z5 e" e j3 e j2 e j6 g � Z7 dQ dR � Z8 dU dS dT � Z9 dU S(W u1 Utility functions, node construction macros, etc.i ( u islicei ( u token( u Leafu Node( u python_symbols( u patcompc C s% t t j | t t j d � | g � S( Nu =( u Nodeu symsu argumentu Leafu tokenu EQUAL( u keywordu value( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu KeywordArg s u KeywordArgc C s t t j d � S( Nu (( u Leafu tokenu LPAR( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu LParen s u LParenc C s t t j d � S( Nu )( u Leafu tokenu RPAR( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu RParen s u RParenc C sl t | t � s | g } n t | t � s? d | _ | g } n t t j | t t j d d d �g | � S( u Build an assignment statementu u =u prefix( u isinstanceu listu prefixu Nodeu symsu atomu Leafu tokenu EQUAL( u targetu source( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Assign s u Assignc C s t t j | d | �S( u Return a NAME leafu prefix( u Leafu tokenu NAME( u nameu prefix( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Name& s u Namec C s | t t j t � | g � g S( u A node tuple for obj.attr( u Nodeu symsu traileru Dot( u obju attr( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Attr* s u Attrc C s t t j d � S( u A comma leafu ,( u Leafu tokenu COMMA( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Comma. s u Commac C s t t j d � S( u A period (.) leafu .( u Leafu tokenu DOT( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Dot2 s u Dotc C sM t t j | j � | j � g � } | rI | j d t t j | � � n | S( u- A parenthesised argument list, used by Call()i ( u Nodeu symsu traileru cloneu insert_childu arglist( u argsu lparenu rparenu node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu ArgList6 s $ u ArgListc C s: t t j | t | � g � } | d k r6 | | _ n | S( u A function callN( u Nodeu symsu poweru ArgListu Noneu prefix( u func_nameu argsu prefixu node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu Call= s u Callc C s t t j d � S( u A newline literalu ( u Leafu tokenu NEWLINE( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu NewlineD s u Newlinec C s t t j d � S( u A blank lineu ( u Leafu tokenu NEWLINE( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu BlankLineH s u BlankLinec C s t t j | d | �S( Nu prefix( u Leafu tokenu NUMBER( u nu prefix( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu NumberL s u Numberc C s1 t t j t t j d � | t t j d � g � S( u A numeric or string subscriptu [u ]( u Nodeu symsu traileru Leafu tokenu LBRACEu RBRACE( u index_node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu SubscriptO s u Subscriptc C s t t j | d | �S( u A string leafu prefix( u Leafu tokenu STRING( u stringu prefix( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu StringU s u Stringc C s d | _ d | _ d | _ t t j d � } d | _ t t j d � } d | _ | | | | g } | r� d | _ t t j d � } d | _ | j t t j | | g � � n t t j | t t j | � g � } t t j t t j d � | t t j d � g � S( uu A list comprehension of the form [xp for fp in it if test]. If test is None, the "if test" part is omitted. u u u foru inu ifu [u ]( u prefixu Leafu tokenu NAMEu appendu Nodeu symsu comp_ifu listmakeru comp_foru atomu LBRACEu RBRACE( u xpu fpu itu testu for_leafu in_leafu inner_argsu if_leafu inner( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu ListCompY s$ "$ u ListCompc C s x | D] } | j � q Wt t j d � t t j | d d �t t j d d d �t t j | � g } t t j | � } | S( uO Return an import statement in the form: from package import name_leafsu fromu prefixu u import( u removeu Leafu tokenu NAMEu Nodeu symsu import_as_namesu import_from( u package_nameu name_leafsu leafu childrenu imp( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu FromImportq s u FromImportc C s� t | t � r. | j t � t � g k r. d St | t � o� t | j � d k o� t | j d t � o� t | j d t � o� t | j d t � o� | j d j d k o� | j d j d k S( u( Does the node represent a tuple literal?i i i i u (u )T( u isinstanceu Nodeu childrenu LParenu RParenu Trueu lenu Leafu value( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu is_tuple� s *u is_tuplec C sz t | t � oy t | j � d k oy t | j d t � oy t | j d t � oy | j d j d k oy | j d j d k S( u' Does the node represent a list literal?i i u [u ]i����i����( u isinstanceu Nodeu lenu childrenu Leafu value( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu is_list� s u is_listc C s t t j t � | t � g � S( N( u Nodeu symsu atomu LParenu RParen( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu parenthesize� s u parenthesizeu sortedu listu setu anyu allu tupleu sumu minu maxu enumeratec c s4 t | | � } x | r/ | Vt | | � } q Wd S( ul Follow 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( u getattr( u obju attru next( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu attr_chain� s u attr_chainue for_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' | 'enumerate' | (any* trailer< '.' 'join' >) ) trailer< '(' node=any ')' > any* > u` power< ( 'sorted' | 'enumerate' ) trailer< '(' arglist<node=any any*> ')' > any* > c C s� t s< t j t � a t j t � a t j t � a d a n t t t g } xR t | t | d � � D]8 \ } } i } | j | | � rd | d | k rd d Sqd Wd S( u 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. u parentu nodeTF( u pats_builtu patcompu compile_patternu p0u p1u p2u Trueu zipu attr_chainu matchu False( u nodeu patternsu patternu parentu results( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu in_special_context� s %"u in_special_contextc C s� | j } | d k r+ | j t j k r+ d S| j } | j t j t j f k rS d S| j t j k r| | j d | k r| d S| j t j k s� | j t j k r� | d k r� | j t j k s� | j d | k r� d Sd S( uG Check that something isn't an attribute or function name etc. i NFT( u prev_siblingu Noneu typeu tokenu DOTu Falseu parentu symsu funcdefu classdefu expr_stmtu childrenu parametersu typedargslistu COMMAu True( u nodeu prevu parent( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu is_probably_builtin� s % u is_probably_builtinc C sp xi | d k rk | j t j k r_ t | j � d k r_ | j d } | j t j k r_ | j Sn | j } q Wd S( u Find the indentation of *node*.i i u N( u Noneu typeu symsu suiteu lenu childrenu tokenu INDENTu valueu parent( u nodeu indent( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu find_indentation� s ' u find_indentationc C sW | j t j k r | S| j � } | j d } | _ t t j | g � } | | _ | S( N( u typeu symsu suiteu cloneu parentu Noneu Node( u nodeu parentu suite( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu make_suite s u make_suitec C s; x4 | j t j k r6 | j } | s t d � � q q W| S( u Find the top level namespace.u, root found before file_input node was found.( u typeu symsu file_inputu parentu ValueError( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu find_root s u find_rootc C s" t | t | � | � } 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 None for the package and 'foo' for the name. ( u find_bindingu find_rootu bool( u packageu nameu nodeu binding( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu does_tree_import s u does_tree_importc C s | j t j t j f k S( u0 Returns true if the node is an import statement.( u typeu symsu import_nameu import_from( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu is_import" s u is_importc C s� d d � } t | � } t | | | � r. d Sd } } xt t | j � D]c \ } } | | � sf qH n x7 t | j | d � � D] \ } } | | � s� Pq� q� W| | } PqH W| d k r xb t | j � D]N \ } } | j t j k r� | j r� | j d j t j k r� | d } Pq� q� Wn | d k ret t j t t j d � t t j | d d �g � } n$ t | t t j | d d �g � } | t � g } | j | t t j | � � d S( u\ Works like `does_tree_import` but adds an import statement if it was not imported. c S s, | j t j k o+ | j o+ t | j d � S( Ni ( u typeu symsu simple_stmtu childrenu is_import( u node( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu is_import_stmt) s u$ touch_import.<locals>.is_import_stmtNi i u importu prefixu ( u find_rootu does_tree_importu enumerateu childrenu typeu symsu simple_stmtu tokenu STRINGu Noneu Nodeu import_nameu Leafu NAMEu FromImportu Newlineu insert_child( u packageu nameu nodeu is_import_stmtu rootu insert_posu offsetu idxu node2u import_u children( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu touch_import& s4 & !$u touch_importc C sQ xJ| j D]?} d } | j t j k rs t | | j d � rB | St | t | j d � | � } | r&| } q&n�| j t j t j f k r� t | t | j d � | � } | r&| } q&ng| j t j k r�t | t | j d � | � } | r� | } q&x$t | j d d � � D]b \ } } | j t j k r| j d k rt | t | j | d � | � } | r{| } q{qqWn� | j t k r�| j d j | k r�| } nv t | | | � r�| } n[ | j t j k r�t | | | � } n4 | j t j k r&t | | j d � r&| } q&n | r | s6| St | � rI| Sq q Wd S( 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.i i i Nu :i i i����i����( u childrenu Noneu typeu symsu for_stmtu _findu find_bindingu make_suiteu if_stmtu while_stmtu try_stmtu enumerateu tokenu COLONu valueu _def_symsu _is_import_bindingu simple_stmtu expr_stmtu is_import( u nameu nodeu packageu childu retu nu iu kid( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu find_bindingT sH &!# % u find_bindingc C s| | g } xl | rw | j � } | j d k rO | j t k rO | j | j � q | j t j k r | j | k r | Sq Wd S( Ni ( u popu typeu _block_symsu extendu childrenu tokenu NAMEu valueu None( u nameu nodeu nodes( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu _find s !u _findc C s' | j t j k r| r| j d } | j t j k r� x� | j D]Z } | j t j k rw | j d j | k r� | SqB | j t j k rB | j | k rB | SqB Wq#| j t j k r� | j d } | j t j k r| j | k r| Sq#| j t j k r#| j | k r#| Sn| j t j k r#| rMt | j d � j � | k rMd S| j d } | rst d | � rsd S| j t j k r�t | | � r�| S| j t j k r�| j d } | j t j k r | j | k r | Sq#| j t j k r| j | k r| S| r#| j t j k r#| Sn d S( u� Will reuturn node if node will import name, or node will import * from package. None is returned otherwise. See test cases for examples. i i i u asi����N( u typeu symsu import_nameu childrenu dotted_as_namesu dotted_as_nameu valueu tokenu NAMEu import_fromu stru stripu Noneu _findu import_as_namesu import_as_nameu STAR( u nodeu nameu packageu impu childu lastu n( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu _is_import_binding� s@ ! !!% ! !!u _is_import_bindingNF(: u __doc__u itertoolsu isliceu pgen2u tokenu pytreeu Leafu Nodeu pygramu python_symbolsu symsu u patcompu KeywordArgu LParenu RParenu Assignu Noneu Nameu Attru Commau Dotu ArgListu Callu Newlineu BlankLineu Numberu Subscriptu Stringu ListCompu FromImportu is_tupleu is_listu parenthesizeu setu consuming_callsu attr_chainu p0u p1u p2u Falseu pats_builtu in_special_contextu is_probably_builtinu find_indentationu make_suiteu find_rootu does_tree_importu is_importu touch_importu classdefu funcdefu _def_symsu find_bindingu traileru _block_symsu _findu _is_import_binding( ( ( u7 /opt/alt/python33/lib64/python3.3/lib2to3/fixer_util.pyu <module> sZ -*
Save File
Cancel