File: //opt/alt/python35/lib64/python3.5/lib2to3/__pycache__/fixer_util.cpython-35.opt-2.pyc
���]�; �
@ sp 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 d d d � Z d d � Z d d � Z d d � Z e
� e � d d � Z d d d d � Z d d � Z d d � Z d d d! � Z d" d# � Z d d$ d% � Z d d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 d4 d5 d6 d7 d8 d9 d: d; h
Z! d< d= � Z" d> a# d? a$ d@ a% dA a& dB dC � Z' dD dE � Z( dF dG � Z) dH dI � Z* dJ dK � Z+ dL dM � Z, dN dO � Z- dP dQ � Z. e j/ e j0 h Z1 d dR dS � Z2 e j0 e j/ e j3 h Z4 dT dU � Z5 d dV dW � Z6 d S)X� )�islice� )�token)�Leaf�Node)�python_symbols)�patcompc C s% t t j | t t j d � | g � S)N�=)r �syms�argumentr r �EQUAL)�keyword�value� r �//opt/alt/python35/lib64/python3.5/fixer_util.py�
KeywordArg s r c C s t t j d � S)N�()r r �LPARr r r r �LParen s r c C s t t j d � S)N�))r r �RPARr r r r �RParen s r c C sf t | t � s | g } t | t � s9 d | _ | g } t t j | t t j d d d �g | � S)N� r �prefix) �
isinstance�listr r r
�atomr r r )�target�sourcer r r �Assign s r Nc C s t t j | d | �S)Nr )r r �NAME)�namer r r r �Name& s r" c C s | t t j t � | g � g S)N)r r
�trailer�Dot)�obj�attrr r r �Attr* s r' c C s t t j d � S)N�,)r r �COMMAr r r r �Comma. s r* c C s t t j d � S)N�.)r r �DOTr r r r r$ 2 s r$ c C sJ t t j | j � | j � g � } | rF | j d t t j | � � | S)Nr )r r
r# �clone� insert_child�arglist)�argsZlparenZrparen�noder r r �ArgList6 s $ r2 c C s7 t t j | t | � g � } | d k r3 | | _ | S)N)r r
�powerr2 r )� func_namer0 r r1 r r r �Call= s r5 c C s t t j d � S)N�
)r r �NEWLINEr r r r �NewlineD s r8 c C s t t j d � S)N� )r r r7 r r r r � BlankLineH s r: c C s t t j | d | �S)Nr )r r �NUMBER)�nr r r r �NumberL s r= c C s1 t t j t t j d � | t t j d � g � S)N�[�])r r
r# r r �LBRACE�RBRACE)Z
index_noder r r � SubscriptO s rB c C s t t j | d | �S)Nr )r r �STRING)�stringr r r r �StringU s rE c 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 � � t t j | t t j | � g � } t t j
t t j d � | t t j d � g � S)Nr9 r �for�in�ifr> r? )
r r r r �appendr r
Zcomp_ifZ listmakerZcomp_forr r@ rA ) Zxp�fp�itZtestZfor_leafZin_leafZ
inner_argsZif_leaf�innerr r r �ListCompY s$ $ rM c 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)N�fromr r �import)�remover r r r r
�import_as_names� import_from)Z package_nameZ
name_leafsZleaf�children�impr r r �
FromImportq s
rU c C s� | d j � } | j t j k r1 | j � } n t t j | j � g � } | d } | ro d d � | D� } t t j t t | d � t | d � � t t j | d j � | | d j � g � g | � } | j | _ | S) Nr% �afterc S s g | ] } | j � � q Sr )r- )�.0r< r r r �
<listcomp>� s z!ImportAndCall.<locals>.<listcomp>r r ZlparZrpar)
r- �typer
r/ r r3 r' r" r# r )r1 �results�namesr% Z
newarglistrV �newr r r �
ImportAndCall� s
Z r] c 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)NT� r r � r r )r r rS r r �lenr r )r1 r r r �is_tuple� s *ra c 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)Nr r r> r? ���rb )r r r` rS r r )r1 r r r �is_list� s rc c C s t t j t � | t � g � S)N)r r
r r r )r1 r r r � parenthesize� s rd �sortedr �set�any�all�tuple�sum�min�max� enumeratec c s4 t | | � } x | r/ | Vt | | � } q Wd S)N)�getattr)r% r& �nextr r r �
attr_chain� s rp zefor_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<node=any any*> ')' >
any*
>
Fc C s� t s9 t j t � a t j t � a t j t � a d a t t t g } xR t | t | d � � D]8 \ } } i } | j | | � ra | d | k ra d Sqa Wd S)NT�parentr1 F) �
pats_builtr Zcompile_pattern�p0�p1�p2�ziprp �match)r1 �patterns�patternrq rZ r r r �in_special_context� s %"rz c 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)NFr T)
Z prev_siblingrY r r, rq r
�funcdef�classdef� expr_stmtrS Z
parametersZ
typedargslistr) )r1 �prevrq r r r �is_probably_builtin� s % r c C sm xf | d k rh | j t j k r\ t | j � d k r\ | j d } | j t j k r\ | j S| j } q Wd S)Nr_ r r9 ) rY r
�suiter` rS r �INDENTr rq )r1 �indentr r r �find_indentation s '
r� c C sW | j t j k r | S| j � } | j d } | _ t t j | g � } | | _ | S)N)rY r
r� r- rq r )r1 rq r� r r r �
make_suite s r� c C s8 x1 | j t j k r3 | j } | s t d � � q W| S)Nz,root found before file_input node was found.)rY r
Z
file_inputrq �
ValueError)r1 r r r � find_root( s
r� c C s" t | t | � | � } t | � S)N)� find_bindingr� �bool)�packager! r1 Zbindingr r r �does_tree_import1 s r� c C s | j t j t j f k S)N)rY r
� import_namerR )r1 r r r � is_import9 s r� c C s� d d � } t | � } t | | | � r. d Sd } } xn t | j � D]] \ } } | | � sc qH x4 t | j | d � � D] \ } } | | � s} Pq} W| | } PqH W| d k rx\ t | j � D]K \ } } | j t j k r� | j r� | j d j t j k r� | d } Pq� W| d k rYt 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)Nc S s, | j t j k o+ | j o+ t | j d � S)Nr )rY r
� simple_stmtrS r� )r1 r r r �is_import_stmt@ s z$touch_import.<locals>.is_import_stmtr r rO r r )r� r� rm rS rY r
r� r rC r r� r r rU r8 r. ) r� r! r1 r� �rootZ
insert_pos�offset�idxZnode2�import_rS r r r � touch_import= s4
&
!$r� c C s< x5| j D]*} d } | j t j k rp t | | j d � rB | St | t | j d � | � } | r| } n�| j t j t j f k r� t | t | j d � | � } | r| } n[| j t j k rvt | t | j d � | � } | r� | } qxt
| j d d � � D]\ \ } } | j t j k r| j
d k rt | t | j | d � | � } | r| } qWn� | j t k r�| j d j
| k r�| } np t | | | � r�| } nU | j t j k r�t | | | � } n. | j t j k rt | | j d � r| } | r
| s$| St | � r
| Sq
Wd S) Nr r_ r^ �:� r rb rb )rS rY r
Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtrm r �COLONr � _def_syms�_is_import_bindingr� r} r� )r! r1 r� �child�retr< �iZkidr r r r� k sH &!#
% r� c 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)N� )�poprY � _block_syms�extendrS r r r )r! r1 Znodesr r r r� � s !r� c 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 | SnA | j t j k r| j | k r| S| r | j t j k r | Sd S)Nr r_ r^ �asrb )rY r
r� rS Zdotted_as_namesZdotted_as_namer r r rR �str�stripr� rQ Zimport_as_name�STAR)r1 r! r� rT r� Zlastr< r r r r� � s@
!
!!%
!
!!r� )7� itertoolsr Zpgen2r Zpytreer r Zpygramr r
r9 r r r r r r" r' r* r$ r2 r5 r8 r: r= rB rE rM rU r] ra rc rd Zconsuming_callsrp rs rt ru rr rz r r� r� r� r� r� r� r| r{ r� r� r# r� r� r� r r r r �<module> sZ -*