Your IP : 216.73.216.85
B
��\:� � @ s� d Z ddlZddlZddlmZmZmZ ddlmZ dddd�Zdd d d d
d d �Z ej
d
kZ ej
dkZ ej
dkZ
dd� Zdd� Zdd� Zdd� Zdd� Zd!dd �ZG d d � d �ZG d d � d e�Zej
d
kr�eZdS )"zjthis module contains utilities for rebuilding a _ast tree in
order to get a single Astroid representation
� N)�_parse�_get_parser_module�parse_function_type_comment)�nodesTF)�None�True�False� Arguments�
Comprehension�
ExceptHandler�Keyword)Z argumentsZ
comprehensionZ ListCompForZ
GenExprForZ
excepthandler�keyword)� r )r � )r � c C sd | j d| jd| jd| jd| jd| jd| jd| jd| jd | j d
| j
d | j d i }t j
d
kr`d|| j< |S )N�+�&�|�^�/z//�%�*z**�-z<<z>>)r � �@)ZAddZBitAndZBitOrZBitXorZDivZFloorDivZModZMultZPowZSubZLShiftZRShift�sys� version_infoZMatMult)�moduleZbinary_operators� r �B/opt/alt/python37/lib/python3.7/site-packages/astroid/rebuilder.py� _binary_operators_from_module- s
r c C s | j d| jdiS )N�and�or)ZAndZOr)r r r r �_bool_operators_from_moduleA s r# c C s | j d| jd| jd| jdiS )Nr r �not�~)ZUAddZUSubZNotZInvert)r r r r � _unary_operators_from_moduleE s r& c C s@ | j d| jd| jd| jd| jd| jd| jd| jd| jd | j d
i
S ) Nz==�>z>=�in�iszis not�<z<=z!=znot in)
ZEqZGtZGtEZInZIsZIsNotZLtZLtEZNotEqZNotIn)r r r r � _compare_operators_from_moduleI s r+ c C s$ | j tj | jtj| jtj| jtjiS )N)�Load�astroid�Store�DelZParam)r r r r �_contexts_from_moduleX s r0 �visitc K s( t | |d�}|r$t ||�||f|�S dS )z_If the given node has an attribute, visits the attribute, and
otherwise returns None.
N)�getattr)�node�attrZvisitor�parentr1 �kws�valuer r r �_visit_or_nonea s r8 c @ sJ e Zd ZdZd�ed�dd�Zdd� Zdd � Zd
d � Zd d
� Z d�dd�Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd�dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd�d)d*�Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9e9Z:dodp� Z;dqdr� Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBdd�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJdS )��
TreeRebuilderz0Rebuilds the _ast tree to become an Astroid treeF)�parse_python_twoc C sj || _ g | _g | _g | _i | _t|d�| _t| j�| _t | j�| _
t | j�| _ t
| j�| _t| j�| _d S )N)r: )Z_manager�
_global_names�_import_from_nodes�_delayed_assattr� _visit_methsr �_parser_moduler& �_unary_op_classesr+ �_cmp_op_classesr# �_bool_op_classesr �_bin_op_classesr0 �_context_classes)�selfZmanagerr: r r r �__init__p s zTreeRebuilder.__init__c C s� yvt r t|d�r |j}||fS |jrtt|jd | jj�rtt|jd j| jj�rt|jd jj }|jdd � |_||fS W n t
k
r� Y nX |d fS )N� docstringr � ) �PY37�hasattrrG �body�
isinstancer? �Exprr7 ZStr�s�
IndexError)rE r3 �docr r r �_get_doc s zTreeRebuilder._get_docc C s | j �t|j�tj�S )N)rD �get�type�ctxr- r, )rE r3 r r r � _get_context� s zTreeRebuilder._get_contextc sF �� |�\}}tj||||g|dd�� � �� �fdd�|jD �� � S )z7visit a Module node by returning a fresh instance of itN)�namerP �file�path�packager5 c s g | ]}�� |� ��qS r )r1 )�.0�child)�newnoderE r r �
<listcomp>� s z.TreeRebuilder.visit_module.<locals>.<listcomp>)rQ r ZModule�postinitrK )rE r3 �modnameZmodpathrY rP r )r\ rE r � visit_module� s zTreeRebuilder.visit_modulec C sT |j }|| jkr | j| }n.|j}dt�||��� }t| |�}|| j|< |||�S )NZvisit_)� __class__r> �__name__�REDIRECTrR �lowerr2 )rE r3 r5 �clsZ visit_methodZcls_name�
visit_namer r r r1 � s
zTreeRebuilder.visitNc C s> | j r*|j| j d kr*|�� �|j|� n|j�|j|� dS )zAsave assignement situation since node.parent is not available yet���N)r; rV �root� set_localr5 )rE r3 rV r r r �_save_assignment� s z TreeRebuilder._save_assignmentc
sz |j |j }}tr6t�|r |jnd|r,|jnd|�� nt�|||�� � �fdd�|jD �}� �fdd�|jD �}d}d}|r�tr�|j jr��� |j j� �}|j}|r�tr�|jjr��� |jj� �}|j}t
�r$� �fdd�|j D �} � �fdd�|j D �}
� �fdd�|jD �} � �fdd�|j D �} ng } g }
g } g } � j
||| |
| | ||d � |�rb� j�|� � |�rv� j�|� � � S )
z;visit an Arguments node by returning a fresh instance of itNc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z1TreeRebuilder.visit_arguments.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s c s g | ]}|r�� |� �nd �qS )N)r1 )rZ r[ )r\ rE r r r] � s c s$ g | ] }|j r ��|j � �nd �qS )N)�
annotationr1 )rZ �arg)r\ rE r r r] � s c s$ g | ] }|j r ��|j � �nd �qS )N)rk r1 )rZ rl )r\ rE r r r] � s )�args�defaults�
kwonlyargs� kw_defaults� annotations�kwonlyargs_annotations�varargannotation�kwargannotation)�vararg�kwarg�PY34r r rl rm rn rk r1 �PY3ro rp r^ r5 ri )
rE r3 r5 ru rv rm rn rs rt ro rp rq rr r )r\ rE r �visit_arguments� sZ
z TreeRebuilder.visit_argumentsc C sF t �|j|j|�}|jr(| �|j|�}nd}|�| �|j|�|� |S )z7visit a Assert node by returning a fresh instance of itN)r ZAssert�lineno�
col_offset�msgr1 r^ �test)rE r3 r5 r\ r| r r r � visit_assert� s zTreeRebuilder.visit_assertc C s^ t |dd �}|sd S y t|�}W n tk
r4 d S X | �|jd |�}t|tj�sXd S |jS )N� type_commentr ) r2 r � SyntaxErrorr1 rK rL r rM r7 )rE r3 r �type_comment_astZ type_objectr r r �check_type_comment� s z TreeRebuilder.check_type_commentc sp t � dd �}|sd S y t|�}W n tk
r4 d S X d }� �fdd�|jpNg D �}|jrh��|j� �}||fS )Nr c s g | ]}�� |� ��qS r )r1 )rZ �elem)r3 rE r r r] s z=TreeRebuilder.check_function_type_comment.<locals>.<listcomp>)r2 r r� �argtypes�returnsr1 )rE r3 r r� r� r� r )r3 rE r �check_function_type_comment s z)TreeRebuilder.check_function_type_commentc sL �� |�}t�|j|j|�� � j� �fdd�|jD ���|j� �|d� � S )z7visit a Assign node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] % s z.TreeRebuilder.visit_assign.<locals>.<listcomp>)�targetsr7 �type_annotation) r� r ZAssignrz r{ r^ r� r1 r7 )rE r3 r5 r� r )r\ rE r � visit_assign s
zTreeRebuilder.visit_assignc C s. t �|t|dd�t|dd�|�}| �|� |S )z)visit a node and return a AssignName noderz Nr{ )r �
AssignNamer2 rj )rE r3 r5 Z node_namer\ r r r �visit_assignname+ s
z TreeRebuilder.visit_assignnamec C sH t �| jt|j� d |j|j|�}|�| �|j |�| �|j
|�� |S )z:visit a AugAssign node by returning a fresh instance of it�=) r Z AugAssignrC rS �oprz r{ r^ r1 �targetr7 )rE r3 r5 r\ r r r �visit_augassign6 s z TreeRebuilder.visit_augassignc C s* t �|j|j|�}|�| �|j|�� |S )z:visit a Backquote node by returning a fresh instance of it)r �Reprrz r{ r^ r1 r7 )rE r3 r5 r\ r r r �
visit_reprC s zTreeRebuilder.visit_reprc C sD t �| jt|j� |j|j|�}|�| �|j |�| �|j
|�� |S )z6visit a BinOp node by returning a fresh instance of it) r ZBinOprC rS r� rz r{ r^ r1 �left�right)rE r3 r5 r\ r r r � visit_binopI s
zTreeRebuilder.visit_binopc s@ t ��jt|j� |j|j|�� � �� �fdd�|jD �� � S )z7visit a BoolOp node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] X s z.TreeRebuilder.visit_boolop.<locals>.<listcomp>) r ZBoolOprB rS r� rz r{ r^ �values)rE r3 r5 r )r\ rE r � visit_boolopS s zTreeRebuilder.visit_boolopc C s t �t|dd�t|dd�|�S )z6visit a Break node by returning a fresh instance of itrz Nr{ )r ZBreakr2 )rE r3 r5 r r r � visit_break[ s zTreeRebuilder.visit_breakc s� t �|j|j|�� t|d�� �}t|d�� �}� �fdd�|jD �}|jrb� �fdd�|jD �}nd}|r�t j|j|j|jd�}|j |d� |�
|� |r�t j d|j|j|jd �}|j |d� |r�|�
|� n|g}� � �� |j
� �||� � S )
z9visit a CallFunc node by returning a fresh instance of it�starargs�kwargsc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] f s z,TreeRebuilder.visit_call.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] i s N)r{ rz r5 )r7 )rl r{ rz r5 )r ZCallrz r{ r8 rm �keywords�Starredr5 r^ �appendr r1 �func) rE r3 r5 r� r� rm r� Z new_starargsZ
new_kwargsr )r\ rE r �
visit_calla s4
zTreeRebuilder.visit_callc
s� �� |�\}}t�|j||j|j|�� d}trXx(|jD ] }|jdkr6�� |� �j
}P q6W |j rl�� |� �}nd}� �
� �fdd�|jD �� �fdd�|jD �|||tr�� �fdd�|jD �ng � � S )z'visit a ClassDef node to become astroidN� metaclassc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z0TreeRebuilder.visit_classdef.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s c s" g | ]}|j d kr��|� ��qS )r� )rl r1 )rZ Zkwd)r\ rE r r r] � s )rQ r ZClassDefrV rz r{ rx r� rl r1 r7 �decorator_list�visit_decoratorsr^ �basesrK )rE r3 r5 �newstylerP r� r
�
decoratorsr )r\ rE r �visit_classdef� s(
z TreeRebuilder.visit_classdefc C s" t �|jt|dd�t|dd�|�S )z6visit a Const node by returning a fresh instance of itrz Nr{ )r �Constr7 r2 )rE r3 r5 r r r � visit_const� s
zTreeRebuilder.visit_constc C s t �t|dd�t|dd�|�S )z9visit a Continue node by returning a fresh instance of itrz Nr{ )r ZContinuer2 )rE r3 r5 r r r �visit_continue� s z TreeRebuilder.visit_continuec sF t �|j|j|�� � ���|j� �� �fdd�t|j|j �D �� � S )z8visit a Compare node by returning a fresh instance of itc s( g | ] \}}�j |j ��|� �f�qS r )rA ra r1 )rZ r� �expr)r\ rE r r r] � s z/TreeRebuilder.visit_compare.<locals>.<listcomp>)
r ZComparerz r{ r^ r1 r� �zipZopsZ comparators)rE r3 r5 r )r\ rE r �
visit_compare� s zTreeRebuilder.visit_comparec sL t �|�� � ���|j� ���|j� �� �fdd�|jD �t|dd�� � S )z>visit a Comprehension node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z5TreeRebuilder.visit_comprehension.<locals>.<listcomp>Zis_asyncN)r r
r^ r1 r� �iterZifsr2 )rE r3 r5 r )r\ rE r �visit_comprehension� s
z!TreeRebuilder.visit_comprehensionc s2 t �|j|j|�� � �� �fdd�|jD �� � S )z;visit a Decorators node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z2TreeRebuilder.visit_decorators.<locals>.<listcomp>)r Z
Decoratorsrz r{ r^ r� )rE r3 r5 r )r\ rE r r� � s z TreeRebuilder.visit_decoratorsc s2 t �|j|j|�� � �� �fdd�|jD �� � S )z7visit a Delete node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z.TreeRebuilder.visit_delete.<locals>.<listcomp>)r ZDeleterz r{ r^ r� )rE r3 r5 r )r\ rE r � visit_delete� s zTreeRebuilder.visit_deletec c sZ xTt |j|j�D ]B\}}| �||�}|s<t�|j|j|�}n | �||�}||fV qW d S )N)r� �keysr� r1 r Z
DictUnpackrz r{ )rE r3 r5 r\ �keyr7 Z
rebuilt_valueZ rebuilt_keyr r r �_visit_dict_items� s z TreeRebuilder._visit_dict_itemsc C s2 t �|j|j|�}t| �|||��}|�|� |S )z5visit a Dict node by returning a fresh instance of it)r ZDictrz r{ �listr� r^ )rE r3 r5 r\ �itemsr r r �
visit_dict� s
zTreeRebuilder.visit_dictc sJ t �|j|j|�� � ���|j� ���|j� �� �fdd�|jD �� � S )z9visit a DictComp node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z0TreeRebuilder.visit_dictcomp.<locals>.<listcomp>) r ZDictComprz r{ r^ r1 r� r7 �
generators)rE r3 r5 r )r\ rE r �visit_dictcomp� s z TreeRebuilder.visit_dictcompc C s* t �|j|j|�}|�| �|j|�� |S )z5visit a Expr node by returning a fresh instance of it)r rM rz r{ r^ r1 r7 )rE r3 r5 r\ r r r �
visit_expr� s zTreeRebuilder.visit_exprc C s t �t|dd�t|dd�|�S )z:visit an Ellipsis node by returning a fresh instance of itrz Nr{ )r �Ellipsisr2 )rE r3 r5 r r r �visit_ellipsis� s z TreeRebuilder.visit_ellipsisc C s t �t|dd�t|dd�|�S )z;visit an EmptyNode node by returning a fresh instance of itrz Nr{ )r Z EmptyNoder2 )rE r3 r5 r r r �visit_emptynode s z TreeRebuilder.visit_emptynodec sJ t �|j|j|�� � �t|d�� �t|d�� �� �fdd�|jD �� � S )z?visit an ExceptHandler node by returning a fresh instance of itrS rV c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r]
s z5TreeRebuilder.visit_excepthandler.<locals>.<listcomp>)r r rz r{ r^ r8 rK )rE r3 r5 r )r\ rE r �visit_excepthandler s z!TreeRebuilder.visit_excepthandlerc C sB t �|j|j|�}|�| �|j|�t|d| |�t|d| |�� |S )z6visit an Exec node by returning a fresh instance of it�globals�locals)r ZExecrz r{ r^ r1 rK r8 )rE r3 r5 r\ r r r �
visit_exec s zTreeRebuilder.visit_execc s, t j|d�� � �� �fdd�|jD �� � S )z:visit an ExtSlice node by returning a fresh instance of it)r5 c s g | ]}�� |� ��qS r )r1 )rZ Zdim)r\ rE r r r] s z0TreeRebuilder.visit_extslice.<locals>.<listcomp>)r ZExtSlicer^ Zdims)rE r3 r5 r )r\ rE r �visit_extslice s z TreeRebuilder.visit_extslicec sj ||j |j|�� ��|�}� j��|j� ���|j� �� �fdd�|jD �� �fdd�|jD �|d� � S )z4visit a For node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] ) s z,TreeRebuilder._visit_for.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] * s )r� r� rK �orelser� ) rz r{ r� r^ r1 r� r� rK r� )rE re r3 r5 r� r )r\ rE r �
_visit_for" s
zTreeRebuilder._visit_forc C s | � tj||�S )N)r� r ZFor)rE r3 r5 r r r � visit_for/ s zTreeRebuilder.visit_forc
C sP dd� |j D �}t�|jp d||jp&dt|dd�t|dd�|�}| j�|� |S )z<visit an ImportFrom node by returning a fresh instance of itc S s g | ]}|j |jf�qS r )rV �asname)rZ �aliasr r r r] 4 s z2TreeRebuilder.visit_importfrom.<locals>.<listcomp>� Nrz r{ )�namesr Z
ImportFromr �levelr2 r< r� )rE r3 r5 r� r\ r r r �visit_importfrom2 s
z TreeRebuilder.visit_importfromc
s� �j �i � ��|�\}}||j||j|j|�� |jrD��|� �}nd}trb|j rb��
|j � �}nd}d }}�� |�} | r�| \}}� j ��
|j
� �� �fdd�|jD �||||d� �j �� � S )z+visit an FunctionDef node to become astroidNc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] U s z4TreeRebuilder._visit_functiondef.<locals>.<listcomp>)rm rK r� r� �type_comment_returns�type_comment_args)r; r� rQ rV rz r{ r� r� rx r� r1 r� r^ rm rK �pop)
rE re r3 r5 rP r� r� r� r� Ztype_comment_annotationr )r\ rE r �_visit_functiondefA s,
z TreeRebuilder._visit_functiondefc C s | � tj||�S )N)r� r Z FunctionDef)rE r3 r5 r r r �visit_functiondef^ s z TreeRebuilder.visit_functiondefc s> t �|j|j|�� � ���|j� �� �fdd�|jD �� � S )z=visit a GeneratorExp node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] f s z4TreeRebuilder.visit_generatorexp.<locals>.<listcomp>)r Z GeneratorExprz r{ r^ r1 �eltr� )rE r3 r5 r )r\ rE r �visit_generatorexpa s
z TreeRebuilder.visit_generatorexpc C s� | � |�}|tjkr,t�|j|j|j|�}nP|tjkrft� |j|j|j|�}t
|tj �s|| j �
|� nt�|j|j|j|�}|�| �|j|�� |S )z;visit an Attribute node by returning a fresh instance of it)rU r- r/ r ZDelAttrr4 rz r{ r. Z
AssignAttrrL r r= r� Z Attributer^ r1 r7 )rE r3 r5 �contextr\ r r r �visit_attributej s
z TreeRebuilder.visit_attributec C sT t �|jt|dd�t|dd�|�}| jrPx&|jD ] }| jd �|g ��|� q0W |S )z%visit a Global node to become astroidrz Nr{ rg )r ZGlobalr� r2 r; �
setdefaultr� )rE r3 r5 r\ rV r r r � visit_global{ s
zTreeRebuilder.visit_globalc sR t �|j|j|�� � ���|j� �� �fdd�|jD �� �fdd�|jD �� � S )z4visit an If node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z*TreeRebuilder.visit_if.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s ) r ZIfrz r{ r^ r1 r} rK r� )rE r3 r5 r )r\ rE r �visit_if� s zTreeRebuilder.visit_ifc C sB t �|j|j|�}|�| �|j|�| �|j|�| �|j|�� |S )z6visit a IfExp node by returning a fresh instance of it) r ZIfExprz r{ r^ r1 r} rK r� )rE r3 r5 r\ r r r � visit_ifexp� s zTreeRebuilder.visit_ifexpc C sf dd� |j D �}t�|t|dd�t|dd�|�}x0|j D ]&\}}|pF|}|�|�d�d |� q8W |S )z7visit a Import node by returning a fresh instance of itc S s g | ]}|j |jf�qS r )rV r� )rZ r� r r r r] � s z.TreeRebuilder.visit_import.<locals>.<listcomp>rz Nr{ �.r )r� r ZImportr2 ri �split)rE r3 r5 r� r\ rV r� r r r � visit_import� s
zTreeRebuilder.visit_importc C s$ t j|d�}|�| �|j|�� |S )z6visit a Index node by returning a fresh instance of it)r5 )r ZIndexr^ r1 r7 )rE r3 r5 r\ r r r � visit_index� s zTreeRebuilder.visit_indexc C s( t j|j|d�}|�| �|j|�� |S )z8visit a Keyword node by returning a fresh instance of it)r5 )r r rl r^ r1 r7 )rE r3 r5 r\ r r r �
visit_keyword� s zTreeRebuilder.visit_keywordc C s6 t �|j|j|�}|�| �|j|�| �|j|�� |S )z7visit a Lambda node by returning a fresh instance of it)r ZLambdarz r{ r^ r1 rm rK )rE r3 r5 r\ r r r � visit_lambda� s zTreeRebuilder.visit_lambdac s@ �� |�}tj||j|j|d�� � �� �fdd�|jD �� � S )z5visit a List node by returning a fresh instance of it)rT rz r{ r5 c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z,TreeRebuilder.visit_list.<locals>.<listcomp>)rU r ZListrz r{ r^ �elts)rE r3 r5 r� r )r\ rE r �
visit_list� s
zTreeRebuilder.visit_listc s> t �|j|j|�� � ���|j� �� �fdd�|jD �� � S )z9visit a ListComp node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z0TreeRebuilder.visit_listcomp.<locals>.<listcomp>)r ZListComprz r{ r^ r1 r� r� )rE r3 r5 r )r\ rE r �visit_listcomp� s
z TreeRebuilder.visit_listcompc C s� | � |�}|tjkr,t�|j|j|j|�}nl|tjkrNt� |j|j|j|�}nJ|jt
kr�t� t
|j t |dd�t |dd�|�}|S t�
|j|j|j|�}|tjtjfkr�| �|� |S )z5visit a Name node by returning a fresh instance of itrz Nr{ )rU r- r/ r ZDelName�idrz r{ r. r� �CONST_NAME_TRANSFORMSr� r2 �Namerj )rE r3 r5 r� r\ r r r rf � s
zTreeRebuilder.visit_namec C s" t �|jt|dd�t|dd�|�S )z<visit a Constant node by returning a fresh instance of Constrz Nr{ )r r� r7 r2 )rE r3 r5 r r r �visit_constant� s
z TreeRebuilder.visit_constantc C s" t �|jt|dd�t|dd�|�S )z@visit a String/Bytes node by returning a fresh instance of Constrz Nr{ )r r� rN r2 )rE r3 r5 r r r � visit_str� s
zTreeRebuilder.visit_strc C s" t �|jt|dd�t|dd�|�S )z7visit a Num node by returning a fresh instance of Constrz Nr{ )r r� �nr2 )rE r3 r5 r r r � visit_num� s
zTreeRebuilder.visit_numc C s t �|j|j|�S )z5visit a Pass node by returning a fresh instance of it)r ZPassrz r{ )rE r3 r5 r r r �
visit_pass s zTreeRebuilder.visit_passc sB t �|j|j|j|�� � �t|d�� �� �fdd�|jD �� � S )z6visit a Print node by returning a fresh instance of it�destc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] s z-TreeRebuilder.visit_print.<locals>.<listcomp>)r ZPrint�nlrz r{ r^ r8 r� )rE r3 r5 r )r\ rE r � visit_print s
zTreeRebuilder.visit_printc C sB t �|j|j|�}|�t|d| |�t|d| |�t|d| |�� |S )z6visit a Raise node by returning a fresh instance of itrS �instZtback)r �Raiserz r{ r^ r8 )rE r3 r5 r\ r r r � visit_raise s zTreeRebuilder.visit_raisec C s4 t �|j|j|�}|jdk r0|�| �|j|�� |S )z7visit a Return node by returning a fresh instance of itN)r ZReturnrz r{ r7 r^ r1 )rE r3 r5 r\ r r r � visit_return s
zTreeRebuilder.visit_returnc s2 t �|j|j|�� � �� �fdd�|jD �� � S )z4visit a Set node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] ) s z+TreeRebuilder.visit_set.<locals>.<listcomp>)r �Setrz r{ r^ r� )rE r3 r5 r )r\ rE r � visit_set& s zTreeRebuilder.visit_setc s> t �|j|j|�� � ���|j� �� �fdd�|jD �� � S )z8visit a SetComp node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] 1 s z/TreeRebuilder.visit_setcomp.<locals>.<listcomp>)r ZSetComprz r{ r^ r1 r� r� )rE r3 r5 r )r\ rE r �
visit_setcomp, s
zTreeRebuilder.visit_setcompc C s< t j|d�}|�t|d| |�t|d| |�t|d| |�� |S )z6visit a Slice node by returning a fresh instance of it)r5 rd �upper�step)r ZSlicer^ r8 )rE r3 r5 r\ r r r � visit_slice5 s zTreeRebuilder.visit_slicec C sD | � |�}tj||j|j|d�}|�| �|j|�| �|j|�� |S )z:visit a Subscript node by returning a fresh instance of it)rT rz r{ r5 ) rU r Z Subscriptrz r{ r^ r1 r7 �slice)rE r3 r5 r� r\ r r r �visit_subscript? s
z TreeRebuilder.visit_subscriptc sZ t �|j|j|�� � �� �fdd�|jD �� �fdd�|jD �� �fdd�|jD �� � S )z:visit a TryExcept node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] N s z1TreeRebuilder.visit_tryexcept.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] O s c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] P s )r Z TryExceptrz r{ r^ rK �handlersr� )rE r3 r5 r )r\ rE r �visit_tryexceptJ s z TreeRebuilder.visit_tryexceptc sF t �|j|j|�� � �� �fdd�|jD �� �fdd�|jD �� � S )z;visit a TryFinally node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] X s z2TreeRebuilder.visit_tryfinally.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r� )r\ rE r r r] Y s )r �
TryFinallyrz r{ r^ rK � finalbody)rE r3 r5 r )r\ rE r �visit_tryfinallyT s
z TreeRebuilder.visit_tryfinallyc s@ �� |�}tj||j|j|d�� � �� �fdd�|jD �� � S )z6visit a Tuple node by returning a fresh instance of it)rT rz r{ r5 c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] c s z-TreeRebuilder.visit_tuple.<locals>.<listcomp>)rU r ZTuplerz r{ r^ r� )rE r3 r5 r� r )r\ rE r � visit_tuple] s
zTreeRebuilder.visit_tuplec C s6 t �| j|jj |j|j|�}|�| �|j |�� |S )z8visit a UnaryOp node by returning a fresh instance of it)
r ZUnaryOpr@ r� ra rz r{ r^ r1 Zoperand)rE r3 r5 r\ r r r �
visit_unaryopf s zTreeRebuilder.visit_unaryopc sR t �|j|j|�� � ���|j� �� �fdd�|jD �� �fdd�|jD �� � S )z6visit a While node by returning a fresh instance of itc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] v s z-TreeRebuilder.visit_while.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] w s ) r ZWhilerz r{ r^ r1 r} rK r� )rE r3 r5 r )r\ rE r � visit_whileq s zTreeRebuilder.visit_whilec st t �|j|j|�� ��|j� �}|jd k r:��|j� �}nd }��|�}� j||fg� �fdd�|j D �|d� � S )Nc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z,TreeRebuilder.visit_with.<locals>.<listcomp>)r� rK r� )
r �Withrz r{ r1 � context_expr�
optional_varsr� r^ rK )rE r3 r5 r� r r� r )r\ rE r �
visit_with{ s
zTreeRebuilder.visit_withc C s4 t �|j|j|�}|jdk r0|�| �|j|�� |S )z6visit a Yield node by returning a fresh instance of itN)r ZYieldrz r{ r7 r^ r1 )rE r3 r5 r\ r r r � visit_yield� s
zTreeRebuilder.visit_yield)F)N)N)N)Krb �
__module__� __qualname__�__doc__�boolrF rQ rU r` r1 rj ry r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rf r� r� Z visit_bytesr� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r9 m s�
>
#
r9 c s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � fdd�Z dd� Z
dd� Zd'� fdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� ZS )(�TreeRebuilder3z/extend and overwrite TreeRebuilder for python3kc C s | � |||j�S )z7visit an arg node by returning a fresh AssName instance)r� rl )rE r3 r5 r r r � visit_arg� s zTreeRebuilder3.visit_argc C s" t �|jt|dd �t|dd �|�S )Nrz r{ )r r� r7 r2 )rE r3 r5 r r r �visit_nameconstant� s
z!TreeRebuilder3.visit_nameconstantc s\ t �|j|j|�� |jr*��|� |j�}nd}� �t|d�� �|� �fdd�|jD �� � S )z?visit an ExceptHandler node by returning a fresh instance of itNrS c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z6TreeRebuilder3.visit_excepthandler.<locals>.<listcomp>) r r rz r{ rV r� r^ r8 rK )rE r3 r5 rV r )r\ rE r r� � s z"TreeRebuilder3.visit_excepthandlerc C s" t �|jt|dd�t|dd�|�S )z5visit a Nonlocal node and return a new instance of itrz Nr{ )r ZNonlocalr� r2 )rE r3 r5 r r r �visit_nonlocal� s
z TreeRebuilder3.visit_nonlocalc C s6 t �|j|j|�}|�t|d| |�t|d| |�� |S )z6visit a Raise node by returning a fresh instance of it�exc�cause)r r� rz r{ r^ r8 )rE r3 r5 r\ r r r r� � s
zTreeRebuilder3.visit_raisec C s8 | � |�}tj||j|j|d�}|�| �|j|�� |S )z4visit a Starred node and return a new instance of it)rT rz r{ r5 )rU r r� rz r{ r^ r1 r7 )rE r3 r5 r� r\ r r r �
visit_starred� s
z TreeRebuilder3.visit_starredc s| |j rft�|j|j|�� |jr.��|� �g}n� �fdd�|jD �}� �|� �fdd�|j D �� � S |jrx��||�S d S )Nc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s z,TreeRebuilder3.visit_try.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r� )r\ rE r r r] � s ) r� r r� rz r{ r� r� rK r^ )rE r3 r5 rK r )r\ rE r � visit_try� s zTreeRebuilder3.visit_tryc C sL t �|j|j|�}t|d| |�}|j| �|j|�||jt|d| |�d� |S )z;visit an AnnAssign node by returning a fresh instance of itrk r7 )r� rk �simpler7 ) r Z AnnAssignrz r{ r8 r^ r1 r� r )rE r3 r5 r\ rk r r r �visit_annassign� s z TreeRebuilder3.visit_annassignc sz d|j kr tt���||�S ||j|j|�� � �fdd����|�}� j�fdd�|jD �� �fdd�|j D �|d� � S )Nr� c s$ �� | j� �}t| d�� �}||fS )Nr )r1 r� r8 )r[ r� �var)r\ rE r r � visit_child� s z/TreeRebuilder3._visit_with.<locals>.visit_childc s g | ] }� |��qS r r )rZ r[ )r r r r] � s z.TreeRebuilder3._visit_with.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] � s )r� rK r� )
�_fields�superr r rz r{ r� r^ r� rK )rE re r3 r5 r� )ra )r\ rE r r � _visit_with� s
zTreeRebuilder3._visit_withc C s | � tj||�S )N)r r r� )rE r3 r5 r r r r � s zTreeRebuilder3.visit_withc C s4 t �|j|j|�}|jd k r0|�| �|j|�� |S )N)r Z YieldFromrz r{ r7 r^ r1 )rE r3 r5 r\ r r r �visit_yieldfrom s
z TreeRebuilder3.visit_yieldfromTc s t t| �j|||d�S )N)r� )r r r� )rE r3 r5 r� )ra r r r� s
z TreeRebuilder3.visit_classdefc C s | � tj||�S )N)r� r ZAsyncFunctionDef)rE r3 r5 r r r �visit_asyncfunctiondef s z%TreeRebuilder3.visit_asyncfunctiondefc C s | � tj||�S )N)r� r ZAsyncFor)rE r3 r5 r r r �visit_asyncfor s z TreeRebuilder3.visit_asyncforc C s, t �|j|j|�}|j| �|j|�d� |S )N)r7 )r ZAwaitrz r{ r^ r1 r7 )rE r3 r5 r\ r r r � visit_await s zTreeRebuilder3.visit_awaitc C s | � tj||�S )N)r r Z AsyncWith)rE r3 r5 r r r �visit_asyncwith s z TreeRebuilder3.visit_asyncwithc s2 t �|j|j|�� � �� �fdd�|jD �� � S )Nc s g | ]}�� |� ��qS r )r1 )rZ r[ )r\ rE r r r] s z2TreeRebuilder3.visit_joinedstr.<locals>.<listcomp>)r Z JoinedStrrz r{ r^ r� )rE r3 r5 r )r\ rE r �visit_joinedstr s z TreeRebuilder3.visit_joinedstrc C s: t �|j|j|�}|�| �|j|�|jt|d| |�� |S )N� format_spec) r ZFormattedValuerz r{ r^ r1 r7 �
conversionr8 )rE r3 r5 r\ r r r �visit_formattedvalue s z#TreeRebuilder3.visit_formattedvalue)T)rb r r r r r r� r
r� r
r r r r r r� r r r r r r �
__classcell__r r )ra r r � s&
r )r1 )r r r- Z astroid._astr r r r r� rc r rx rw rI r r# r& r+ r0 r8 r9 r r r r r �<module> s>
,