Your IP : 3.139.87.151
o
6��ff; � @ 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 � ZdMd d �Zd d!� ZdMd"d#�ZdMd$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/dMdGdH�Z0ej.ej-ej1hZ2dIdJ� Z3dMdKdL�Z4d
S )O� )�token)�Leaf�Node)�python_symbols)�patcompc C s t tj| ttjd�|g�S )N�=)r �symsZargumentr r �EQUAL)�keyword�value� r �9/opt/alt/python310/lib64/python3.10/lib2to3/fixer_util.py�
KeywordArg s �r c C � t tjd�S )N�()r r �LPARr r r r
�LParen � r c C r )N�))r r �RPARr r r r
�RParen r r c C sJ t | t�s | g} t |t�sd|_|g}ttj| ttjddd�g | �S )N� r ��prefix) �
isinstance�listr r r �atomr r r )�target�sourcer r r
�Assign s
�r Nc C � t tj| |d�S �Nr )r r �NAME)�namer r r r
�Name$ � r$ c C s | t tjt� |g�gS �N)r r �trailer�Dot)�obj�attrr r r
�Attr( s r+ c C � t tjd�S )N�,)r r �COMMAr r r r
�Comma, � r/ c C r, )N�.)r r �DOTr r r r
r( 0 r0 r( c C s6 t tj|�� |�� g�}| r|�dt tj| �� |S )Nr )r r r' �clone� insert_child�arglist)�argsZlparenZrparen�noder r r
�ArgList4 s
r8 c C s( t tj| t|�g�}|d ur||_|S r& )r r �powerr8 r )Z func_namer6 r r7 r r r
�Call; s
r: c C r, )N�
�r r �NEWLINEr r r r
�NewlineB r0 r> c C r, )N� r< r r r r
� BlankLineF r0 r@ c C s t tj| |d�S r! )r r �NUMBER)�nr r r r
�NumberJ s rC c C s$ t tjttjd�| ttjd�g�S )N�[�])r r r' r r �LBRACE�RBRACE)Z
index_noder r r
� SubscriptM s
�rH c C r r! )r r �STRING)�stringr r r r
�StringS r% rK c C s� d| _ d|_ d|_ ttjd�}d|_ ttjd�}d|_ ||||g}|r;d|_ ttjd�}d|_ |�ttj||g�� ttj| ttj |�g�}ttj
ttj d�|ttj d�g�S )Nr? r �for�in�ifrD rE )
r r r r" �appendr r Zcomp_ifZ listmakerZcomp_forr rF rG ) Zxp�fp�it�testZfor_leafZin_leafZ
inner_argsZif_leaf�innerr r r
�ListCompW s*
��rT c C sX |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
� r\ c C s� |d � � }|jtjkr|� � }n ttj|� � g�}|d }|r(dd� |D �}ttjtt|d �t|d ��ttj|d � � ||d � � g�g | �}| j |_ |S ) Nr) �afterc S s g | ]}|� � �qS r )r3 )�.0rB r r r
�
<listcomp>� s z!ImportAndCall.<locals>.<listcomp>� r ZlparZrpar)
r3 �typer r5 r r9 r+ r$ r' r )r7 �results�namesr) Z
newarglistr] �newr r r
�
ImportAndCall� s,
�����re c C s� t | t�r| jt� t� gkrdS t | t�oDt| j�dkoDt | jd t�oDt | jd t�oDt | jd t�oD| jd jdkoD| jd jdkS )NT� r` r � r r )r r rZ r r �lenr r �r7 r r r
�is_tuple� s
������rj c C sZ t | t�o,t| j�dko,t | jd t�o,t | jd t�o,| jd jdko,| jd jdkS )Nr r` ���rD rE )r r rh rZ r r ri r r r
�is_list� s
�����rl c C s t tjt� | t� g�S r& )r r r r r ri r r r
� parenthesize� s rm >
�allr �max�set�any�min�tuple�sorted�sum� enumeratec c s. � t | |�}|r|V t ||�}|s d S d S r& )�getattr)r) r* �nextr r r
�
attr_chain� s �
�ry 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 sr t st�t�at�t�at�t�ada tttg}t|t| d��D ]\}}i }|�||�r6|d | u r6 dS q!dS )NT�parentr7 F) �
pats_builtr Zcompile_pattern�p0�p1�p2�zipry �match)r7 �patterns�patternrz rb r r r
�in_special_context� s
�r� c C s� | j }|d ur|jtjkrdS | j}|jtjtjfv r dS |jtjkr-|j d | u r-dS |jtj
ksJ|jtj krL|d urC|jtj ksJ|j d | u rLdS dS )NFr` T)
Z prev_siblingra r r2 rz r �funcdef�classdef� expr_stmtrZ Z
parametersZ
typedargslistr. )r7 �prevrz r r r
�is_probably_builtin� s r� c C sR | d ur'| j tjkr t| j�dkr | jd }|j tjkr |jS | j} | d usdS )Nrg r r? ) ra r �suiterh rZ r �INDENTr rz )r7 �indentr r r
�find_indentation s
�r� c C s> | j tjkr| S | �� } | jd }| _ttj| g�}||_|S r& )ra r r� r3 rz r )r7 rz r� r r r
�
make_suite s r� c C s0 | j tjkr| j} | std��| j tjks| S )Nz,root found before file_input node was found.)ra r Z
file_inputrz �
ValueErrorri r r r
� find_root& s �r� c C s t |t|�| �}t|�S r& )� find_bindingr� �bool)�packager# r7 Zbindingr r r
�does_tree_import/ s r� c C s | j tjtjfv S r& )ra r � import_namerY ri r r r
� is_import7 s r� c C s, dd� }t |�}t| ||�rd S d }}t|j�D ]!\}}||�s#qt|j|d � �D ]
\}}||�s6 nq,|| } |dkrbt|j�D ] \}}|jtjkra|jra|jd jtjkra|d } nqE| d u ryt tj
t tj d�t tj |dd�g�} n t
| t tj |dd�g�} | t� g}
|�|t tj|
�� d S )Nc S s | j tjko| jot| jd �S )Nr` )ra r � simple_stmtrZ r� ri r r r
�is_import_stmt> s �z$touch_import.<locals>.is_import_stmtr` r rV r r )r� r� rv rZ ra r r� r rI r r� r r" r\ r> r4 ) r� r# r7 r� �rootZ
insert_pos�offset�idxZnode2�import_rZ r r r
� touch_import; s< ��
�
r� c C s� |j D ]�}d }|jtjkr*t| |j d �r| S t| t|j d �|�}|r)|}n�|jtjtjfv rCt| t|j d �|�}|rB|}nw|jtj kr�t| t|j d �|�}|rY|}nat
|j dd � �D ] \}}|jt j kr�|j
dkr�t| t|j |d �|�}|r�|}qbn6|jtv r�|j d j
| kr�|}n&t|| |�r�|}n |jtjkr�t| ||�}n|jtjkr�t| |j d �r�|}|r�|s�| S t|�r�| S qd S )Nr rk rg rf �:� r` )rZ ra r Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtrv r �COLONr � _def_syms�_is_import_bindingr� r� r� )r# r7 r� �childZretrB �iZkidr r r
r� i sN
�� �� �r� c C sV |g}|r)|� � }|jdkr|jtvr|�|j� n
|jtjkr'|j| kr'|S |sd S )N� )�popra � _block_syms�extendrZ r r" r )r# r7 Znodesr r r
r� � s �r� c C s� | j tjkrg|sg| jd }|j tjkr>|jD ]$}|j tjkr,|jd j|kr+| S q|j tjkr;|j|kr;| S qd S |j tjkrX|jd }|j tjkrV|j|krV| S d S |j tjkre|j|kre| S d S | j tj kr�|r|t
| jd �� � |kr|d S | jd }|r�t d|�r�d S |j tj
kr�t ||�r�| S |j tjkr�|jd }|j tjkr�|j|kr�| S d S |j tjkr�|j|kr�| S |r�|j tjkr�| S d S )Nr rg rk rf �as)ra r r� rZ Zdotted_as_namesZdotted_as_namer r r" rY �str�stripr� rX Zimport_as_name�STAR)r7 r# r� r[ r� ZlastrB r r r
r� � sN
�� �
� �
�r� r& )NN)5Zpgen2r Zpytreer r Zpygramr r r? r r r r r r$ r+ r/ r( r8 r: r> r@ rC rH rK rT r\ re rj rl rm Zconsuming_callsry r| r} r~ r{ r� r� r� r� r� r� r� r� r� r� r� r� r' r� r� r� r r r r
�<module> sZ
-
*