3
\g; �
@ s� d dl mZ d dlmZmZ d dlmZ d dlm Z dd� Z
dd� Z d d
� Z d d � Z
dVdd�Zdd� Zdd� Zdd� Ze � e � fdd�ZdWdd�Zdd� Zd d � ZdXd d �Zd d!� ZdYd"d#�ZdZd$d%�Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z d0d1d2d3d4d5d6d7d8d9h
Z d:d;� Z da#d?a$d@dA� Z%dBdC� Z&dDdE� Z'dFdG� Z(dHdI� Z)dJdK� Z*dLdM� Z+dNdO� Z,ej-ej.hZ/d[dPdQ�Z0ej.ej-ej1hZ2dRdS� Z3d\dTdU�Z4d
S )]� )�token)�Leaf�Node)�python_symbols)�patcompc C s t tj| ttjd�|g�S )N�=)r �symsZargumentr r �EQUAL)�keyword�value� r �*/usr/lib64/python3.6/lib2to3/fixer_util.py�
KeywordArg s r c C s t tjd�S )N�()r r �LPARr r r r
�LParen s r c C s t tjd�S )N�))r r �RPARr r r r
�RParen s r c C sH t | t�s| g} t |t�s&d|_|g}ttj| ttjddd�g | �S )N� r )�prefix) �
isinstance�listr r r �atomr r r )�target�sourcer r r
�Assign s
r Nc C s t tj| |d�S )N)r )r r �NAME)�namer r r r
�Name$ s r c C s | t tjt� |g�gS )N)r r �trailer�Dot)�obj�attrr r r
�Attr( s r$ c C s t tjd�S )N�,)r r �COMMAr r r r
�Comma, s r' c C s t tjd�S )N�.)r r �DOTr r r r
r! 0 s r! c C s4 t tj|j� |j� g�}| r0|jdt tj| �� |S )Nr )r r r �clone� insert_child�arglist)�argsZlparenZrparen�noder r r
�ArgList4 s r/ c C s& t tj| t|�g�}|d k r"||_|S )N)r r �powerr/ r )Z func_namer- r r. r r r
�Call; s r1 c C s t tjd�S )N�
)r r �NEWLINEr r r r
�NewlineB s r4 c C s t tjd�S )N� )r r r3 r r r r
� BlankLineF s r6 c C s t tj| |d�S )N)r )r r �NUMBER)�nr r r r
�NumberJ s r9 c C s" t tjttjd�| ttjd�g�S )N�[�])r r r r r �LBRACE�RBRACE)Z
index_noder r r
� SubscriptM s r> c C s t tj| |d�S )N)r )r r �STRING)�stringr r r r
�StringS s rA c C s� d| _ d|_ d|_ ttjd�}d|_ ttjd�}d|_ ||||g}|rtd|_ ttjd�}d|_ |jttj||g�� ttj| ttj |�g�}ttj
ttj d�|ttj d�g�S )Nr5 r �for�in�ifr: r; )
r r r r �appendr r Zcomp_ifZ listmakerZcomp_forr r< r= ) Zxp�fp�itZtestZfor_leafZin_leafZ
inner_argsZif_leaf�innerr r r
�ListCompW s$
rI c C sZ x|D ] }|j � qW ttjd�ttj| dd�ttjddd�ttj|�g}ttj|�}|S )N�fromr )r �import)�remover r r r r �import_as_names� import_from)Z package_nameZ
name_leafsZleaf�children�impr r r
�
FromImporto s
rQ c C s� |d j � }|jtjkr"|j � }nttj|j � g�}|d }|rNdd� |D �}ttjtt|d �t|d ��ttj|d j � ||d j � g�g | �}| j |_ |S ) Nr" �afterc S s g | ] }|j � �qS r )r* )�.0r8 r r r
�
� s z!ImportAndCall..� r ZlparZrpar)
r* �typer r, r r0 r$ r r r )r. �results�namesr" Z
newarglistrR �newr r r
�
ImportAndCall� s
DrZ c C s� t | t�r | jt� t� gkr dS t | t�o�t| j�dko�t | jd t�o�t | jd t�o�t | jd t�o�| jd jdko�| jd jdkS )NT� rU r � r r )r r rO r r �lenr r )r. r r r
�is_tuple� s
r^ c C sX t | t�oVt| j�dkoVt | jd t�oVt | jd t�oV| jd jdkoV| jd jdkS )Nr rU r: r; �����r_ )r r r] rO r r )r. r r r
�is_list� s
r` c C s t tjt� | t� g�S )N)r r r r r )r. r r r
� parenthesize� s ra �sortedr �set�any�all�tuple�sum�min�max� enumeratec c s( t | |�}x|r"|V t ||�}q W d S )N)�getattr)r" r# �nextr r r
�
attr_chain� s
rm 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 ')' >
any*
>
Fc C sr t s&tjt�atjt�atjt�ada tttg}x | j tjkr| S | j� } | jd }| _ttj| g�}||_|S )N)rV r r| r* rn r )r. rn r| r r r
�
make_suite s r� c C s( x"| j tjkr"| j} | std��qW | S )Nz,root found before file_input node was found.)rV r Z
file_inputrn �
ValueError)r. r r r
� find_root& s
r� c C s t |t|�| �}t|�S )N)� find_bindingr� �bool)�packager r. Zbindingr r r
�does_tree_import/ s r� c C s | j tjtjfkS )N)rV r � import_namerN )r. r r r
� is_import7 s r� c C s4 dd� }t |�}t| ||�r d S d }}xTt|j�D ]F\}}||�sFq4x(t|j|d � �D ]\}}||�sZP qZW || }P q4W |dkr�xDt|j�D ]6\}}|jtjkr�|jr�|jd jtjkr�|d }P q�W | d kr�t tj
t tj d�t tj |dd�g�} nt
| t tj |dd�g�} | t� g}
|j|t tj|
�� d S )Nc S s | j tjko | jo t| jd �S )NrU )rV r � simple_stmtrO r� )r. r r r
�is_import_stmt> s z$touch_import..is_import_stmtrU r rK r )r )r� r� rj rO rV r r� r r? r r� r r rQ r4 r+ ) r� r r. r� �rootZ
insert_pos�offset�idxZnode2�import_rO r r r
� touch_import; s4
r� c C s� �x�|j D �]�}d }|jtjkrVt| |j d �r4|S t| t|j d �|�}|rR|}�n4|jtjtjfkr�t| t|j d �|�}|r�|}�n |jtj k�rt| t|j d �|�}|r�|}nXx�t
|j dd � �D ]@\}}|jt j ko�|j
dkr�t| t|j |d �|�}|r�|}q�W nx|jtk�r6|j d j
| k�r6|}nTt|| |��rJ|}n@|jtjk�rft| ||�}n$|jtjk�r�t| |j d ��r�|}|r
|�s�|S t|�r
|S q
W d S ) Nr r\ r[ �:� rU r_ r_ )rO rV r Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtrj r �COLONr � _def_syms�_is_import_bindingr� ry r� )r r. r� �childZretr8 �iZkidr r r
r� i sH
r� c C sX |g}xL|rR|j � }|jdkr6|jtkr6|j|j� q|jtjkr|j| kr|S qW d S )N� )�poprV � _block_syms�extendrO r r r )r r. Znodesr r r
r� � s r� c C s� | j tjkr�| r�| jd }|j tjkrvx�|jD ]@}|j tjkrV|jd j|krp| S q0|j tjkr0|j|kr0| S q0W nL|j tjkr�|jd }|j tjkr�|j|kr�| S n|j tjkr�|j|kr�| S n�| j tj k�r�|r�t
| jd �j � |kr�d S | jd }|�rt d|��rd S |j tj
k�r.t ||��r.| S |j tjk�rf|jd }|j tjk�r�|j|k�r�| S n6|j tjk�r�|j|k�r�| S |�r�|j tjk�r�| S d S )Nr r\ r[ �asr_ )rV r r� rO Zdotted_as_namesZdotted_as_namer r r rN �str�stripr� rM Zimport_as_name�STAR)r. r r� rP r� Zlastr8 r r r
r� � s@
r� )N)NN)N)N)N)N)N)5Zpgen2r Zpytreer r Zpygramr r r5 r r r r r r r$ r' r! r/ r1 r4 r6 r9 r> rA rI rQ rZ r^ r` ra Zconsuming_callsrm rp rq rr ro rv r{ r r� r� r� r� r� rx rw r� r� r r� r� r� r r r r
� sX
-
*