U
e5dg; �
@ 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.8/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 �Nr )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|�� |�� g�}| r0|�dt tj| �� |S )Nr )r r r# �clone� insert_child�arglist)�argsZlparenZrparen�noder r r
�ArgList4 s r2 c C s& t tj| t|�g�}|d k r"||_|S r" )r r �powerr2 r )Z func_namer0 r r1 r r r
�Call; s r4 c C s t tjd�S )N�
�r r �NEWLINEr r r r
�NewlineB s r8 c C s t tjd�S )N� r6 r r r r
� BlankLineF s r: c C s t tj| |d�S r )r r �NUMBER)�nr r r r
�NumberJ s r= c C s" t tjttjd�| ttjd�g�S )N�[�])r r r# r r �LBRACE�RBRACE)Z
index_noder r r
� SubscriptM s
�rB c C s t tj| |d�S r )r r �STRING)�stringr r r r
�StringS s rE c C s� d| _ d|_ d|_ ttjd�}d|_ ttjd�}d|_ ||||g}|rtd|_ ttjd�}d|_ |�ttj||g�� ttj| ttj |�g�}ttj
ttj d�|ttj 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
�ListCompW s(
��rM c C sV |D ] }|� � qttjd�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
� rU c C s� |d � � }|jtjkr"|� � }nttj|� � g�}|d }|rNdd� |D �}ttjtt|d �t|d ��ttj|d � � ||d � � g�g | �}| j |_ |S ) Nr% �afterc S s g | ] }|� � �qS r )r- )�.0r< r r r
�
� s z!ImportAndCall..� r ZlparZrpar)
r- �typer r/ r r3 r' r! r# r )r1 �results�namesr% Z
newarglistrV �newr r r
�
ImportAndCall� s*
�����r^ 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� rY r � r r )r r rS r r �lenr r �r1 r r r
�is_tuple� s
������rc c C sX t | t�oVt| j�dkoVt | jd t�oVt | jd t�oV| jd jdkoV| jd jdkS )Nr rY �����r> r? )r r ra rS r r rb r r r
�is_list� s
�����re c C s t tjt� | t� g�S r" )r r r r r rb r r r
� parenthesize� s rf �sortedr �set�any�all�tuple�sum�min�max� enumeratec c s$ t | |�}|r |V t ||�}q
d S r" )�getattr)r% r& �nextr r r
�
attr_chain� s
rr 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 sp t s&t�t�at�t�at�t�ada tttg}t|t| d��D ]*\}}i }|�||�r@|d | kr@ dS q@dS )NT�parentr1 F) �
pats_builtr Zcompile_pattern�p0�p1�p2�ziprr �match)r1 Zpatterns�patternrs r[ r r r
�in_special_context� s
r{ c C s� | j }|d k r |jtjkr dS | j}|jtjtjfkr:dS |jtjkrX|j d | krXdS |jtj
ks�|jtj kr�|d k r�|jtj ks�|j d | kr�dS dS )NFrY T)
Z prev_siblingrZ r r, rs r �funcdef�classdef� expr_stmtrS Z
parametersZ
typedargslistr) )r1 �prevrs r r r
�is_probably_builtin� s&
��
� �r� c C sJ | d k rF| j tjkr>t| j�dkr>| jd }|j tjkr>|jS | j} q dS )Nr` r r9 ) rZ r �suitera rS r �INDENTr rs )r1 �indentr r r
�find_indentation s
r� c C s> | j tjkr| S | �� } | jd }| _ttj| g�}||_|S r" )rZ r r� r- rs r )r1 rs r� r r r
�
make_suite s r� c C s$ | j tjkr | j} | s td��q | S )Nz,root found before file_input node was found.)rZ r Z
file_inputrs �
ValueErrorrb r r r
� find_root& s
r� c C s t |t|�| �}t|�S r" )� find_bindingr� �bool)�packager r1 Zbindingr r r
�does_tree_import/ s r� c C s | j tjtjfkS r" )rZ r � import_namerR rb r r r
� is_import7 s r� c C s. dd� }t |�}t| ||�r d S d }}t|j�D ]F\}}||�sDq2t|j|d � �D ]\}}||�sV qlqV|| } qzq2|dkr�t|j�D ]8\}}|jtjkr�|jr�|jd jtjkr�|d } q�q�| d kr�t tj
t tj d�t tj |dd�g�} nt
| t tj |dd�g�} | t� g}
|�|t tj|
�� d S )Nc S s | j tjko | jo t| jd �S )NrY )rZ r � simple_stmtrS r� rb r r r
�is_import_stmt> s �z$touch_import..is_import_stmtrY r rO r r )r� r� ro rS rZ 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; s8 �
�
r� c C s� |j D �]�}d }|jtjkrVt| |j d �r4| S t| t|j d �|�}|rR|}�n0|jtjtjfkr�t| t|j d �|�}|r�|}� n�|jtj k�rt| t|j d �|�}|r�|}nTt
|j dd � �D ]@\}}|jt j kr�|j
dkr�t| t|j |d �|�}|r�|}q�nx|jtk�r2|j d j
| k�r2|}nTt|| |��rF|}n@|jtjk�rbt| ||�}n$|jtjk�r�t| |j d ��r�|}|r|�s�| S t|�r| S qd S )Nr rd r` r_ �:� rY )rS rZ r Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtro r �COLONr � _def_syms�_is_import_bindingr� r~ r� )r r1 r� �childZretr< �iZkidr r r
r� i sH
r� c C sT |g}|rP|� � }|jdkr4|jtkr4|�|j� q|jtjkr|j| kr|S qd S )N� )�poprZ � _block_syms�extendrS r r r )r r1 Znodesr r r
r� � s r� c C s� | j tjkr�|s�| jd }|j tjkrx|jD ]H}|j tjkrV|jd j|krt| S q,|j tjkr,|j|kr,| S q,nL|j tjkr�|jd }|j tjkr�|j|kr�| S n|j tjkr�|j|kr�| S n�| j tj k�r�|r�t
| jd �� � |kr�d S | jd }|�rt d|��rd S |j tj
k�r0t ||��r0| S |j tjk�rh|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` rd r_ �as)rZ 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� )N)NN)N)N)N)N)N)5Zpgen2r Zpytreer r Zpygramr r r9 r r r r r r! r' r* r$ r2 r4 r8 r: r= rB rE rM rU r^ rc re rf Zconsuming_callsrr ru rv rw rt r{ r� r� r� r� r� r� r� r} r| r� r� r# r� r� r� r r r r
� s^
� -
*