Your IP : 216.73.216.213
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)� 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)�And�Or)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 r3 �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�parentr4 �kws�valuer r r �_visit_or_nonea s r; 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= )�_manager�
_global_names�_import_from_nodes�_delayed_assattr� _visit_methsr �_parser_moduler) �_unary_op_classesr. �_cmp_op_classesr& �_bool_op_classesr! �_bin_op_classesr3 �_context_classes)�self�managerr= 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�hasattrrL �body�
isinstancerC �Exprr: �Str�s�
IndexError)rI r6 �docr r r �_get_doc s zTreeRebuilder._get_docc C s | j �t|j�tj�S )N)rH �get�type�ctxr0 r/ )rI r6 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)�namerV �file�path�packager8 c s g | ]}�� |� ��qS r )r4 )�.0�child)�newnoderI r r �
<listcomp>� s z.TreeRebuilder.visit_module.<locals>.<listcomp>)rW r �Module�postinitrP )rI r6 �modnameZmodpathr_ rV r )rb rI r � visit_module� s zTreeRebuilder.visit_modulec C sT |j }|| jkr | j| }n.|j}dt�||��� }t| |�}|| j|< |||�S )NZvisit_)� __class__rB �__name__�REDIRECTrX �lowerr5 )rI r6 r8 �clsZ visit_methodZcls_name�
visit_namer r r r4 � 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? r\ �root� set_localr8 )rI r6 r\ 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 )r4 )r` ra )rb rI r r rc � s z1TreeRebuilder.visit_arguments.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s c s g | ]}|r�� |� �nd �qS )N)r4 )r` ra )rb rI r r rc � s c s$ g | ] }|j r ��|j � �nd �qS )N)�
annotationr4 )r` �arg)rb rI r r rc � s c s$ g | ] }|j r ��|j � �nd �qS )N)rr r4 )r` rs )rb rI r r rc � s )�args�defaults�
kwonlyargs� kw_defaults� annotations�kwonlyargs_annotations�varargannotation�kwargannotation)�vararg�kwarg�PY34r r rs rt ru rr r4 �PY3rv rw re r8 rp )
rI r6 r8 r| r} rt ru rz r{ rv rw rx ry r )rb rI 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�msgr4 re �test)rI r6 r8 rb 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 ) r5 r � SyntaxErrorr4 rP rQ r rR r: )rI r6 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 )r4 )r` �elem)r6 rI r r rc s z=TreeRebuilder.check_function_type_comment.<locals>.<listcomp>)r5 r r� �argtypes�returnsr4 )rI r6 r� r� r� r� r )r6 rI 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 )r4 )r` ra )rb rI r r rc % s z.TreeRebuilder.visit_assign.<locals>.<listcomp>)�targetsr: �type_annotation) r� r ZAssignr� r� re r� r4 r: )rI r6 r8 r� r )rb rI r � visit_assign s
zTreeRebuilder.visit_assignc C s. t �|t|dd�t|dd�|�}| �|� |S )z)visit a node and return a AssignName noder� Nr� )r �
AssignNamer5 rq )rI r6 r8 Z node_namerb 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 AugAssignrG rY �opr� r� re r4 �targetr: )rI r6 r8 rb 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 �Reprr� r� re r4 r: )rI r6 r8 rb 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 ZBinOprG rY r� r� r� re r4 �left�right)rI r6 r8 rb 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 )r4 )r` ra )rb rI r r rc X s z.TreeRebuilder.visit_boolop.<locals>.<listcomp>) r ZBoolOprF rY r� r� r� re �values)rI r6 r8 r )rb rI 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 itr� Nr� )r ZBreakr5 )rI r6 r8 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 )r4 )r` ra )rb rI r r rc f s z,TreeRebuilder.visit_call.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc i s N)r� r� r8 )r: )rs r� r� r8 )r ZCallr� r� r; rt �keywords�Starredr8 re �appendr r4 �func) rI r6 r8 r� r� rt r� Z new_starargsZ
new_kwargsr )rb rI 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 )r4 )r` ra )rb rI r r rc � s z0TreeRebuilder.visit_classdef.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s c s" g | ]}|j d kr��|� ��qS )r� )rs r4 )r` Zkwd)rb rI r r rc � s )rW r ZClassDefr\ r� r� r r� rs r4 r: �decorator_list�visit_decoratorsre �basesrP )rI r6 r8 �newstylerV r� r �
decoratorsr )rb rI 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 itr� Nr� )r �Constr: r5 )rI r6 r8 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 itr� Nr� )r �Continuer5 )rI r6 r8 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 )rE rh r4 )r` r� �expr)rb rI r r rc � s z/TreeRebuilder.visit_compare.<locals>.<listcomp>)
r ZComparer� r� re r4 r� �zipZopsZ comparators)rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc � s z5TreeRebuilder.visit_comprehension.<locals>.<listcomp>Zis_asyncN)r r
re r4 r� �iterZifsr5 )rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc � s z2TreeRebuilder.visit_decorators.<locals>.<listcomp>)r Z
Decoratorsr� r� re r� )rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc � s z.TreeRebuilder.visit_delete.<locals>.<listcomp>)r ZDeleter� r� re r� )rI r6 r8 r )rb rI 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� r4 r Z
DictUnpackr� r� )rI r6 r8 rb �keyr: 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 �Dictr� r� �listr� re )rI r6 r8 rb �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 )r4 )r` ra )rb rI r r rc � s z0TreeRebuilder.visit_dictcomp.<locals>.<listcomp>) r ZDictCompr� r� re r4 r� r: �
generators)rI r6 r8 r )rb rI 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 rR r� r� re r4 r: )rI r6 r8 rb 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 itr� Nr� )r �Ellipsisr5 )rI r6 r8 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 itr� Nr� )r Z EmptyNoder5 )rI r6 r8 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 itrY r\ c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc
s z5TreeRebuilder.visit_excepthandler.<locals>.<listcomp>)r r r� r� re r; rP )rI r6 r8 r )rb rI 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 ZExecr� r� re r4 rP r; )rI r6 r8 rb 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)r8 c s g | ]}�� |� ��qS r )r4 )r` Zdim)rb rI r r rc s z0TreeRebuilder.visit_extslice.<locals>.<listcomp>)r ZExtSlicere Zdims)rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc ) s z,TreeRebuilder._visit_for.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc * s )r� r� rP �orelser� ) r� r� r� re r4 r� r� rP r� )rI rl r6 r8 r� r )rb rI r �
_visit_for" s
zTreeRebuilder._visit_forc C s | � tj||�S )N)r� r ZFor)rI r6 r8 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 )r\ �asname)r` �aliasr r r rc 4 s z2TreeRebuilder.visit_importfrom.<locals>.<listcomp>� Nr� r� )�namesr Z
ImportFromr �levelr5 r@ r� )rI r6 r8 r� rb 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 )r4 )r` ra )rb rI r r rc U s z4TreeRebuilder._visit_functiondef.<locals>.<listcomp>)rt rP r� r� �type_comment_returns�type_comment_args)r? r� rW r\ r� r� r� r� r r� r4 r� re rt rP �pop)
rI rl r6 r8 rV r� r� r� r� Ztype_comment_annotationr )rb rI r �_visit_functiondefA s,
z TreeRebuilder._visit_functiondefc C s | � tj||�S )N)r� r Z FunctionDef)rI r6 r8 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 )r4 )r` ra )rb rI r r rc f s z4TreeRebuilder.visit_generatorexp.<locals>.<listcomp>)r Z GeneratorExpr� r� re r4 �eltr� )rI r6 r8 r )rb rI 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)r[ r0 r2 r ZDelAttrr7 r� r� r1 Z
AssignAttrrQ r rA r� � Attributere r4 r: )rI r6 r8 �contextrb 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 astroidr� Nr� rn )r ZGlobalr� r5 r? �
setdefaultr� )rI r6 r8 rb r\ 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 )r4 )r` ra )rb rI r r rc � s z*TreeRebuilder.visit_if.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s ) r ZIfr� r� re r4 r� rP r� )rI r6 r8 r )rb rI 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 ZIfExpr� r� re r4 r� rP r� )rI r6 r8 rb 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 )r\ r� )r` r� r r r rc � s z.TreeRebuilder.visit_import.<locals>.<listcomp>r� Nr� �.r )r� r ZImportr5 rp �split)rI r6 r8 r� rb r\ 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)r8 )r ZIndexre r4 r: )rI r6 r8 rb 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)r8 )r r rs re r4 r: )rI r6 r8 rb 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 ZLambdar� r� re r4 rt rP )rI r6 r8 rb 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)rZ r� r� r8 c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s z,TreeRebuilder.visit_list.<locals>.<listcomp>)r[ r ZListr� r� re �elts)rI r6 r8 r� r )rb rI 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 )r4 )r` ra )rb rI r r rc � s z0TreeRebuilder.visit_listcomp.<locals>.<listcomp>)r ZListCompr� r� re r4 r� r� )rI r6 r8 r )rb rI 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 itr� Nr� )r[ r0 r2 r ZDelName�idr� r� r1 r� �CONST_NAME_TRANSFORMSr� r5 �Namerq )rI r6 r8 r� rb r r r rm � s
zTreeRebuilder.visit_namec C s" t �|jt|dd�t|dd�|�S )z<visit a Constant node by returning a fresh instance of Constr� Nr� )r r� r: r5 )rI r6 r8 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 Constr� Nr� )r r� rT r5 )rI r6 r8 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 Constr� Nr� )r r� �nr5 )rI r6 r8 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 ZPassr� r� )rI r6 r8 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 )r4 )r` ra )rb rI r r rc s z-TreeRebuilder.visit_print.<locals>.<listcomp>)r ZPrint�nlr� r� re r; r� )rI r6 r8 r )rb rI 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 itrY �instZtback)r �Raiser� r� re r; )rI r6 r8 rb 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 ZReturnr� r� r: re r4 )rI r6 r8 rb 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 )r4 )r` ra )rb rI r r rc ) s z+TreeRebuilder.visit_set.<locals>.<listcomp>)r �Setr� r� re r� )rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc 1 s z/TreeRebuilder.visit_setcomp.<locals>.<listcomp>)r ZSetCompr� r� re r4 r� r� )rI r6 r8 r )rb rI 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)r8 rk �upper�step)r ZSlicere r; )rI r6 r8 rb 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)rZ r� r� r8 ) r[ r Z Subscriptr� r� re r4 r: �slice)rI r6 r8 r� rb 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 )r4 )r` ra )rb rI r r rc N s z1TreeRebuilder.visit_tryexcept.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc O s c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc P s )r Z TryExceptr� r� re rP �handlersr� )rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc X s z2TreeRebuilder.visit_tryfinally.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` r� )rb rI r r rc Y s )r �
TryFinallyr� r� re rP � finalbody)rI r6 r8 r )rb rI 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)rZ r� r� r8 c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc c s z-TreeRebuilder.visit_tuple.<locals>.<listcomp>)r[ r ZTupler� r� re r� )rI r6 r8 r� r )rb rI 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 ZUnaryOprD r� rh r� r� re r4 Zoperand)rI r6 r8 rb 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 )r4 )r` ra )rb rI r r rc v s z-TreeRebuilder.visit_while.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc w s ) r ZWhiler� r� re r4 r� rP r� )rI r6 r8 r )rb rI 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 )r4 )r` ra )rb rI r r rc � s z,TreeRebuilder.visit_with.<locals>.<listcomp>)r� rP r� )
r �Withr� r� r4 � context_expr�
optional_varsr� re rP )rI r6 r8 r� r
r� r )rb rI 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 ZYieldr� r� r: re r4 )rI r6 r8 rb r r r � visit_yield� s
zTreeRebuilder.visit_yield)F)N)N)N)Kri �
__module__� __qualname__�__doc__�boolrK rW r[ rg r4 rq 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� r� rm r� r� Z visit_bytesr� r� r� r� r� r� r� r� r� r r r r r r r r r r r r< m s�
>
#
r< 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� rs )rI r6 r8 r r r � visit_arg� s zTreeRebuilder3.visit_argc C s" t �|jt|dd �t|dd �|�S )Nr� r� )r r� r: r5 )rI r6 r8 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 itNrY c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s z6TreeRebuilder3.visit_excepthandler.<locals>.<listcomp>) r r r� r� r\ r� re r; rP )rI r6 r8 r\ r )rb rI 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 itr� Nr� )r ZNonlocalr� r5 )rI r6 r8 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� r� r� re r; )rI r6 r8 rb 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)rZ r� r� r8 )r[ r r� r� r� re r4 r: )rI r6 r8 r� rb 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 )r4 )r` ra )rb rI r r rc � s z,TreeRebuilder3.visit_try.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` r� )rb rI r r rc � s ) r r r r� r� r r rP re )rI r6 r8 rP r )rb rI 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 itrr r: )r� rr �simpler: ) r Z AnnAssignr� r� r; re r4 r� r )rI r6 r8 rb rr 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
)r4 r r; )ra r� �var)rb rI r r � visit_child� s z/TreeRebuilder3._visit_with.<locals>.visit_childc s g | ] }� |��qS r r )r` ra )r r r rc � s z.TreeRebuilder3._visit_with.<locals>.<listcomp>c s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc � s )r� rP r� )
�_fields�superr r r� r� r� re r� rP )rI rl r6 r8 r� )rh )rb rI r r � _visit_with� s
zTreeRebuilder3._visit_withc C s | � tj||�S )N)r r r )rI r6 r8 r r r r � s zTreeRebuilder3.visit_withc C s4 t �|j|j|�}|jd k r0|�| �|j|�� |S )N)r Z YieldFromr� r� r: re r4 )rI r6 r8 rb r r r �visit_yieldfrom s
z TreeRebuilder3.visit_yieldfromTc s t t| �j|||d�S )N)r� )r r r� )rI r6 r8 r� )rh r r r� s
z TreeRebuilder3.visit_classdefc C s | � tj||�S )N)r� r ZAsyncFunctionDef)rI r6 r8 r r r �visit_asyncfunctiondef s z%TreeRebuilder3.visit_asyncfunctiondefc C s | � tj||�S )N)r� r ZAsyncFor)rI r6 r8 r r r �visit_asyncfor s z TreeRebuilder3.visit_asyncforc C s, t �|j|j|�}|j| �|j|�d� |S )N)r: )r ZAwaitr� r� re r4 r: )rI r6 r8 rb r r r � visit_await s zTreeRebuilder3.visit_awaitc C s | � tj||�S )N)r r Z AsyncWith)rI r6 r8 r r r �visit_asyncwith s z TreeRebuilder3.visit_asyncwithc s2 t �|j|j|�� � �� �fdd�|jD �� � S )Nc s g | ]}�� |� ��qS r )r4 )r` ra )rb rI r r rc s z2TreeRebuilder3.visit_joinedstr.<locals>.<listcomp>)r Z JoinedStrr� r� re r� )rI r6 r8 r )rb rI r �visit_joinedstr s z TreeRebuilder3.visit_joinedstrc C s: t �|j|j|�}|�| �|j|�|jt|d| |�� |S )N� format_spec) r ZFormattedValuer� r� re r4 r: �
conversionr; )rI r6 r8 rb r r r �visit_formattedvalue s z#TreeRebuilder3.visit_formattedvalue)T)ri r
r r r r r� r r� r r r r r r r� r! r" r# r$ r% r( �
__classcell__r r )rh r r � s&
r )r4 )r r r0 Z astroid._astr r r r r� rj r r r~ rN r! r& r) r. r3 r; r< r r r r r �<module> s>
,