File: //snap/core20/2599/lib/python3/dist-packages/jinja2/__pycache__/idtracking.cpython-38.pyc
U
�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|�}| D ]}|�|� q|S �N)�parent)�Symbols�FrameSymbolVisitor�visit)�nodes�parent_symbols�sym�visitor�node� r �3/usr/lib/python3/dist-packages/jinja2/idtracking.py� find_symbols s
r c C s t |d�}|�| � |S 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*| j�|�S d S r )r r � find_load)r �targetr r r r# 2 s
zSymbols.find_loadc C s. || j kr| j | S | jd k r*| j�|�S d S r )r r �find_ref�r r r r r r% 8 s
zSymbols.find_refc C s" | � |�}|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�| j� | j�� |_| j�� |_| j�� |_|S r ) �object�__new__� __class__�__dict__�updater �copyr r )r r( r r r r/ E s z Symbols.copyc C s` | j �|� || jkr\| jd k rJ| j�|�}|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|td fd�S r0 )r r1 r" �VAR_LOAD_PARAMETERr& r r r �declare_parameter_ s zSymbols.declare_parameterc C s( | � |�}|d kr$| j|t|fd� d S r0 )r% r" �VAR_LOAD_RESOLVE)r r r$ r r r r! c s
z Symbols.loadc C s� i }|D ]0}|j D ]$}|| j kr"q|�|d�d ||< qq|D ].}| j�|j� | j�|j� | j �|j � q>t|�D ]l\}}|t|�kr�qv| �|�}|d k s�td��| j d k r�| j �|�}|d k r�t
|f| j|< qvt |f| j|< qvd S )Nr r zshould not happen) r �getr r. r r �lenr% r' r r2 r7 ) r Zbranch_symbolsr �branchr$ r
r Z branch_countZ outer_targetr r r �
branch_updateh s*
zSymbols.branch_updatec C s> i }| }|d k r:|j D ]}||kr| �|�||< q|j}q|S r )r r% r )r r( r r r r r � dump_stores� s
zSymbols.dump_storesc C sH t � }| }|d k rDt| j�D ] \}\}}|tkr |�|� q |j}q
|S r )r r r r5 r1 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/ r4 r6 r! r; r< 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 r )r � sym_visitor�r �symbolsr r r r � s zRootVisitor.__init__c K s |� � D ]}| j�|� qd S r �Ziter_child_nodesrB r
�r r r �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s |j D ]}| j�|� qd S r ��bodyrB r
rF r r r �visit_AssignBlock� s
z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� q d S )N)�call)ZexcluderE rF r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s |j D ]}| j�|� qd S r rI rF r r r �visit_OverlayScope� s
z RootVisitor.visit_OverlayScoperJ c K s� |dkr"| j j|jdd� |j}nN|dkr2|j}n>|dkrh| j j|jdd� |jd k rd| j �|j� d S td��|pvdD ]}| j �|� qxd S )NrJ T)�store_as_param�else�testzUnknown for branchr )rB r
r$ rJ �else_rQ � RuntimeError)r r Z
for_branchr r: �itemr r r � visit_For� s
zRootVisitor.visit_Forc K s4 |j D ]}| j�|� q|jD ]}| j�|� q d S r )ZtargetsrB r
rJ )r r r r$ rG 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)rJ )r? r@ rA r rH Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifierrK rM rN rU rV rY 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 r )rD rC r r r r � s zFrameSymbolVisitor.__init__Fc K sT |s|j dkr | j�|j� n2|j dkr8| j�|j� n|j dkrP| j�|j� dS )z2All assignments to names go through this function.r r4 r! N)ZctxrD r6 r r4 r! )r r rO r r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namec K s | j �|j� d S r )rD r! r �r r r r r r � visit_NSRef� s z FrameSymbolVisitor.visit_NSRefc s^ �j |jf� � �j�� ��fdd�}||j�}||j�}||jpDd�}�j�|||g� d S )Nc s0 �� � �_}| D ]}�j|f� � q��_|S r )r/ rD r
)r r( Zsubnode�r Zoriginal_symbolsr r r � inner_visit� s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr )r
rQ rD rJ Zelif_rR r; )r r r rc Z body_symbolsZ elif_symbolsZ else_symbolsr rb r r^ � s
zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S r )rD r4 r r` r r r r[ � s z FrameSymbolVisitor.visit_Macroc K s | j |f|� | j�|j� d S r )rY rD r4 r$ r` r r r � visit_Import� s z FrameSymbolVisitor.visit_Importc K sF | j |f|� |jD ],}t|t�r4| j�|d � q| j�|� qd S )Nr )rY �names�
isinstance�tuplerD r4 )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 � 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
�iterr` r r r rU s z FrameSymbolVisitor.visit_Forc K s | j |jf|� d S r )r
rL r` r r r rM s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf|� d S r )r
�filterr` r r r r\ s z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr
)r r r r$ r r r rV s
z FrameSymbolVisitor.visit_Withc K s | j |jf|� dS )z Stop visiting at block assigns.N)r
r$ r` r r r rK s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr r` r r r r] s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr r` r r r rZ s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr r` r r r rN s z%FrameSymbolVisitor.visit_OverlayScopeN)F)r? r@ rA �__doc__r r_ ra r^ r[ rd rh ri rU rM r\ rV rK r] rZ rN r r r r r � s"
r )N)N)Zjinja2.visitorr Zjinja2._compatr r5 r7 r2 r3 r r r* r r r r r r r �<module> s
~4