File: //lib64/python3.9/lib2to3/__pycache__/fixer_util.cpython-39.opt-2.pyc
a
�DOgf; � @ 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 �syms�argumentr r �EQUAL)�keyword�value� r
�*/usr/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 r3 c C s& t tj| t|�g�}|d ur"||_|S r# )r r �powerr3 r )� func_namer1 r r2 r
r
r �Call; s r6 c C s t tjd�S )N�
�r r �NEWLINEr
r
r
r �NewlineB s r: c C s t tjd�S )N� r8 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
�rD c C s t tj| |d�S r )r r �STRING)�stringr r
r
r �StringS s rG 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 )Nr; r �for�in�ifr@ rA )
r r r r �appendr r Zcomp_ifZ listmakerZcomp_forr rB rC ) Zxp�fp�it�testZfor_leafZin_leafZ
inner_argsZif_leaf�innerr
r
r �ListCompW s(
��rP 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
� rX 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 �
<listcomp>� � z!ImportAndCall.<locals>.<listcomp>� r ZlparZrpar)
r. �typer r0 r r4 r( r" r$ r )r2 �results�namesr& Z
newarglistrY �newr
r
r �
ImportAndCall� s*
�����rb 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 rV r r �lenr r �r2 r
r
r �is_tuple� s
������rg c C sX t | t�oVt| j�dkoVt | jd t�oVt | jd t�oV| jd jdkoV| jd jdkS )Nr r] ���r@ rA )r r re rV r r rf r
r
r �is_list� s
�����ri c C s t tjt� | t� g�S r# )r r r r r rf r
r
r � parenthesize� s rj >
�sorted�any�max�all�sum�set�tuple�min� enumerater c c s$ t | |�}|r |V t ||�}q
d S r# )�getattr)r& r' �nextr
r
r �
attr_chain� s
rv 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 sp t s&t�t�at�t�at�t�ada tttg}t|t| d��D ]*\}}i }|�||�r@|d | u r@ dS q@dS )NT�parentr2 F) �
pats_builtr �compile_pattern�p0�p1�p2�ziprv �match)r2 �patterns�patternrw 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- rw r �funcdef�classdef� expr_stmtrV �
parametersZ
typedargslistr* )r2 �prevrw 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 )Nrd r r; ) r^ r �suitere rV r �INDENTr rw )r2 �indentr
r
r �find_indentation s
r� c C s> | j tjkr| S | �� } | jd }| _ttj| g�}||_|S r# )r^ r r� r. rw r )r2 rw 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_inputrw �
ValueErrorrf r
r
r � find_root& s
r� c C s t |t|�| �}t|�S r# )� find_bindingr� �bool)�packager! r2 Zbindingr
r
r �does_tree_import/ s r� c C s | j tjtjfv S r# )r^ r � import_namerU rf 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_stmtrV r� rf r
r
r �is_import_stmt> s �z$touch_import.<locals>.is_import_stmtr] r rR r r )r� r� rs rV r^ r r� r rE r r� r r rX r: r/ ) r� r! r2 r� �rootZ
insert_pos�offset�idxZnode2�import_rV 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 rh rd rc �:� r] )rV r^ r Zfor_stmt�_findr� r� Zif_stmtZ
while_stmtZtry_stmtrs r �COLONr � _def_syms�_is_import_bindingr� r� r� )r! r2 r� �child�retr> �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�extendrV r r r )r! r2 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 rd rh rc �as)r^ r r� rV Zdotted_as_namesZdotted_as_namer r r rU �str�stripr� rT Zimport_as_name�STAR)r2 r! r� rW r� �lastr> r
r
r r� � s@
r� )N)NN)N)N)N)N)N)5Zpgen2r Zpytreer r Zpygramr r r; r r r r r r" r( r+ r% r3 r6 r: r< r? rD rG rP rX rb rg ri rj Zconsuming_callsrv rz r{ r| rx r� r� r� r� r� r� r� r� r� r� r� r� r$ r� r� r� r
r
r
r �<module> sV
-
*