a
XC?hf; � @ s� d dl mZ d dlmZmZ d dlmZ d dlm Z dd� Z
dd� Z d d
� Z d d � Z
dMdd�Zdd� Zdd� Zdd� Ze � e � fdd�ZdNdd�Zdd� Zd d � ZdOd d �Zd d!� ZdPd"d#�ZdQd$d%�Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z h d0�Z d1d2� Z d3a!d4a"d5a#d6a$d7d8� Z%d9d:� Z&d;d<� Z'd=d>� Z(d?d@� Z)dAdB� Z*dCdD� Z+dEdF� Z,ej-ej.hZ/dRdGdH�Z0ej.ej-ej1hZ2dIdJ� Z3dSdKdL�Z4d
S )T� )�token)�Leaf�Node)�python_symbols)�patcompc C s t tj| ttjd�|g�S )N�=)r �symsZargumentr r �EQUAL)�keyword�value� r �7/opt/alt/python39/lib64/python3.9/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 ur"||_|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�it�testZfor_leafZin_leafZ
inner_argsZif_leaf�innerr r r
�ListCompW s(
��rN 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
� rV 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
�
� � z!ImportAndCall..� r ZlparZrpar)
r- �typer r/ r r3 r' r! r# r )r1 �results�namesr% Z
newarglistrW �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� r[ r � r r )r r rT r r �lenr r �r1 r r r
�is_tuple� s
������re c C sX t | t�oVt| j�dkoVt | jd t�oVt | jd t�oV| jd jdkoV| jd jdkS )Nr r[ �����r> r? )r r rc rT r r rd r r r
�is_list� s
�����rg c C s t tjt� | t� g�S r" )r r r r r rd r r r
� parenthesize� s rh >
�sum�any�max�tuple� enumerater �sorted�min�all�setc c s$ t | |�}|r |V t ||�}q
d S r" )�getattr)r% r& �nextr r r
�
attr_chain� s
rt 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 | u r@ dS q@dS )NT�parentr1 F) �
pats_builtr Zcompile_pattern�p0�p1�p2�ziprt �match)r1 Zpatterns�patternru r] r r r
�in_special_context� s
r} c C s� | j }|d ur |jtjkr dS | j}|jtjtjfv r:dS |jtjkrX|j d | u rXdS |jtj
ks�|jtj kr�|d ur�|jtj ks�|j d | u r�dS dS )NFr[ T)
Z prev_siblingr\ r r, ru r �funcdef�classdef� expr_stmtrT �
parametersZ
typedargslistr) )r1 �prevru r r r
�is_probably_builtin� s&
��
� �r� c C sJ | d urF| j tjkr>t| j�dkr>| jd }|j tjkr>|jS | j} q dS )Nrb r r9 ) r\ r �suiterc rT r �INDENTr ru )r1 �indentr r r
�find_indentation s
r� c C s> | j tjkr| S | �� } | jd }| _ttj| g�}||_|S r" )r\ r r� r- ru r )r1 ru 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.)r\ r Z
file_inputru �
ValueErrorrd 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 tjtjfv S r" )r\ r � import_namerS rd 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 u r�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 )Nr[ )r\ r � simple_stmtrT r� rd r r r
�is_import_stmt> s �z$touch_import..is_import_stmtr[ r rP r r )r� r� rm rT r\ r r� r rC r r� r r rV r8 r. ) r� r r1 r� �rootZ
insert_pos�offset�idxZnode2�import_rT r r r
� touch_import; s8 �
�
r� c C s� |j D �]�}d }|jtjkrVt| |j d �r4| S t| t|j d �|�}|rR|}�n0|jtjtjfv r�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|jtv �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 rf rb ra �:� r[ )rT r\ r Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtrm r �COLONr � _def_syms�_is_import_bindingr� r� r� )r r1 r� �childZretr< �iZkidr r r
r� i sB
r� c C sT |g}|rP|� � }|jdkr4|jtvr4|�|j� q|jtjkr|j| kr|S qd S )N� )�popr\ � _block_syms�extendrT 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 rb rf ra �as)r\ r r� rT Zdotted_as_namesZdotted_as_namer r r rS �str�stripr� rR Zimport_as_name�STAR)r1 r r� rU 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 rN rV r` re rg rh Zconsuming_callsrt rw rx ry rv r} r� r� r� r� r� r� r� r r~ r� r� r# r� r� r� r r r r
� sV
-
*