3
�G�\�# � @ sp d dl mZ d dlmZ dZdZdZdZddd �Zdd
d �Z G d d
� d
e
�Z G dd� de�Z G dd� de�Z
dS )� )� NodeVisitor)� iteritems�paramZresolve�aliasZ undefinedNc C s. t |d�}t|�}x| D ]}|j|� qW |S )N)�parent)�Symbols�FrameSymbolVisitor�visit)�nodes�parent_symbols�sym�visitor�node� r � /usr/lib/python3.6/idtracking.py� find_symbols s
r c C s t |d�}|j| � |S )N)r )r � analyze_node)r r r r r r �symbols_for_node s
r c @ sx e Zd Zd dd�Zdd� Zd dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) r Nc C sD |d kr |d krd}n
|j d }|| _ || _i | _i | _t� | _d S )Nr � )�levelr �refs�loads�set�stores)�selfr r r r r �__init__ s
zSymbols.__init__c K s t | �}|j|f|� d S )N)� RootVisitorr )r r �kwargsr
r r r r ' s zSymbols.analyze_nodec C s. d| j |f }|| j|< |d k r*|| j|< |S )Nzl_%d_%s)r r r )r �name�loadZidentr r r � _define_ref+ s
zSymbols._define_refc C s. || j kr| j | S | jd k r*| jj|�S d S )N)r r � find_load)r �targetr r r r! 2 s
zSymbols.find_loadc C s. || j kr| j | S | jd k r*| jj|�S d S )N)r r �find_ref)r r r r r r# 8 s
zSymbols.find_refc C s" | j |�}|d kr td| ��|S )NzITried to resolve a name to a reference that was unknown to the frame (%r))r# �AssertionError)r r �rvr r r �ref> s
z Symbols.refc C sB t j| j�}|jj| j� | jj� |_| jj� |_| jj� |_|S )N) �object�__new__� __class__�__dict__�updater �copyr r )r r% r r r r, E s z Symbols.copyc C s` | j j|� || jkr\| jd k rJ| jj|�}|d k rJ| j|t|fd� d S | j|td fd� d S )N)r )r �addr r r# r �VAR_LOAD_ALIAS�VAR_LOAD_UNDEFINED)r r Z outer_refr r r �storeM s
z
Symbols.storec C s | j j|� | j|td fd�S )N)r )r r- r �VAR_LOAD_PARAMETER)r r r r r �declare_parameter_ s zSymbols.declare_parameterc C s( | j |�}|d kr$| j|t|fd� d S )N)r )r# r �VAR_LOAD_RESOLVE)r r r" r r r r c s
z Symbols.loadc C s� i }x<|D ]4}x.|j D ]$}|| j kr&q|j|d�d ||< qW q
W x6|D ].}| jj|j� | jj|j� | j j|j � qHW xxt|�D ]l\}}|t|�kr�q�| j|�}|d k s�td��| j d k r�| j j|�}|d k r�t
|f| j|< q�t |f| j|< q�W d S )Nr r zshould not happen) r �getr r+ r r �lenr# r$ r r. r3 ) r Zbranch_symbolsr �branchr" r r Z branch_countZ outer_targetr r r �
branch_updateh s*
zSymbols.branch_updatec C sF i }| }x8|d k r@x$|j D ]}||kr| j|�||< qW |j}q
W |S )N)r r# r )r r% r r r r r � dump_stores� s
zSymbols.dump_storesc C sP t � }| }x@|d k rJx,t| j�D ] \}\}}|tkr |j|� q W |j}q W |S )N)r r r r1 r- r )r r% r r" Zinstr�_r r r �dump_param_targets� s
zSymbols.dump_param_targets)NN)N)�__name__�
__module__� __qualname__r r r r! r# r& r, r0 r2 r r7 r8 r: r r r r r s
r c @ sj e Zd Zdd� Zdd� Ze Z Z Z Z Z Z
Z dd� Z dd� Z
d d
� Zdd d
�Zdd� Zdd� ZdS )r c C s t |�| _d S )N)r � sym_visitor)r �symbolsr r r r � s zRootVisitor.__init__c K s" x |j � D ]}| jj|� q
W d S )N)�iter_child_nodesr> r )r r r �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s x|j D ]}| jj|� qW d S )N)�bodyr> r )r r r rA r r r �visit_AssignBlock� s z RootVisitor.visit_AssignBlockc K s& x |j dd�D ]}| jj|� qW d S )N�call)�exclude)rE )r@ r> r )r r r rA r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s x|j D ]}| jj|� qW d S )N)rC r> r )r r r rA r r r �visit_OverlayScope� s z RootVisitor.visit_OverlayScoperC c K s� |dkr"| j j|jdd� |j}nN|dkr2|j}n>|dkrh| j j|jdd� |jd k rd| j j|j� d S td��x |pxf D ]}| j j|� qzW d S )NrC T)�store_as_param�else�testzUnknown for branch)r> r r" rC �else_rK � RuntimeError)r r Z
for_branchr r6 �itemr r r � visit_For� s
zRootVisitor.visit_Forc K s< x|j D ]}| jj|� qW x|jD ]}| jj|� q$W d S )N)Ztargetsr> r rC )r r r r" rA r r r �
visit_With� s zRootVisitor.visit_Withc O s t d|jj ��d S )NzCannot find symbols for %r)�NotImplementedErrorr) r; )r r �argsr r r r �
generic_visit� s zRootVisitor.generic_visitN)rC )r; r< r= r rB Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifierrD rG rH rO rP rS r r r r r � s
r c @ s� e Zd ZdZdd� Zd$dd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )%r z A visitor for `Frame.inspect`.c C s
|| _ d S )N)r? )r r? r r r r � s zFrameSymbolVisitor.__init__Fc K sT |s|j dkr | jj|j� n2|j dkr8| jj|j� n|j dkrP| jj|j� dS )z2All assignments to names go through this function.r r0 r N)Zctxr? r2 r r0 r )r r rI r r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namec K s | j j|j� d S )N)r? r r )r r r r r r � visit_NSRef� s z FrameSymbolVisitor.visit_NSRefc s^ �j |jf� � �j�� ��fdd�}||j�}||j�}||jpDf �}�jj|||g� d S )Nc s4 �j � �_}x| D ]}�j|f� � qW ��_|S )N)r, r? r )r
r% Zsubnode)r �original_symbolsr r r � inner_visit� s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visit)r rK r? rC Zelif_rL r7 )r r r r\ Z body_symbolsZ elif_symbolsZ else_symbolsr )r r[ r r rX � s
zFrameSymbolVisitor.visit_Ifc K s | j j|j� d S )N)r? r0 r )r r r r r r rU � s z FrameSymbolVisitor.visit_Macroc K s | j |f|� | jj|j� d S )N)rS r? r0 r" )r r r r r r � visit_Import� s z FrameSymbolVisitor.visit_Importc K sJ | j |f|� x6|jD ],}t|t�r6| jj|d � q| jj|� qW d S )Nr )rS �names�
isinstance�tupler? r0 )r r r r r r r �visit_FromImport� s
z#FrameSymbolVisitor.visit_FromImportc K s$ | j |jf|� | j |jf|� dS )z'Visit assignments in the correct order.N)r r r" )r r r r r r � visit_Assign� s z FrameSymbolVisitor.visit_Assignc K s | j |jf|� dS )zqVisiting stops at for blocks. However the block sequence
is visited as part of the outer scope.
N)r �iter)r r r r r r rO s z FrameSymbolVisitor.visit_Forc K s | j |jf|� d S )N)r rE )r r r r r r rG s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf|� d S )N)r �filter)r r r r r r rV s z$FrameSymbolVisitor.visit_FilterBlockc K s x|j D ]}| j|� qW d S )N)�valuesr )r r r r" r r r rP s z FrameSymbolVisitor.visit_Withc K s | j |jf|� dS )z Stop visiting at block assigns.N)r r" )r r r r r r rD s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr )r r r r r r rW s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr )r r r r r r rT s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr )r r r r r r rH s z%FrameSymbolVisitor.visit_OverlayScopeN)F)r; r< r= �__doc__r rY rZ rX rU r] ra rb rO rG rV rP rD rW rT rH r r r r r � s"
r )N)N)Zjinja2.visitorr Zjinja2._compatr r1 r3 r. r/ r r r' r r r r r r r �<module> s
~4