← Back
Editing: fixer_util.cpython-39.opt-2.pyc
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 � <listcomp>� � z!ImportAndCall.<locals>.<listcomp>� 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<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�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.<locals>.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 �<module> sV - *
Save File
Cancel