a
�`�# � @ 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 )� )� iteritems)� NodeVisitor�param�resolve�aliasZ undefinedNc C s* t |d�}t|�}| D ]}|�|� q|S �N)�parent)�Symbols�FrameSymbolVisitor�visit)�nodes�parent_symbols�sym�visitor�node� r �5/usr/lib/python3.9/site-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 u r |d u rd}n
|j d }|| _ || _i | _i | _t� | _d S �N� r )�levelr �refs�loads�set�stores)�selfr r r r r �__init__ s
zSymbols.__init__c K s t | �}|j|fi |�� d S �N)� RootVisitorr )r r �kwargsr r r r r % s zSymbols.analyze_nodec C s. d| j |f }|| j|< |d ur*|| j|< |S )Nzl_%d_%s)r r r )r �name�load�identr r r � _define_ref) s
zSymbols._define_refc C s. || j v r| j | S | jd ur*| j�|�S d S r )r r � find_load)r �targetr r r r& 0 s
zSymbols.find_loadc C s. || j v r| j | S | jd ur*| j�|�S d S r )r r �find_ref�r r" r r r r( 6 s
zSymbols.find_refc C s" | � |�}|d u r 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 r2 E s z Symbols.copyc C s` | j �|� || jvr\| jd urJ| j�|�}|d urJ| 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 r3 )r r4 r% �VAR_LOAD_PARAMETERr) r r r �declare_parameter_ s zSymbols.declare_parameterc C s( | � |�}|d u r$| j|t|fd� d S r3 )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 v r"q|�|d�d ||< qq|D ].}| j�|j� | j�|j� | j �|j � q>t|�D ]\\}}|t|�kr�qv| �|�}| jd ur�| j�|�}|d ur�t |f| j|< qvt
|f| j|< qvd S r ) r �getr r1 r r �lenr( r r5 r: ) r Zbranch_symbolsr �branchr' r r" Z branch_countZ outer_targetr r r �
branch_updateh s(
zSymbols.branch_updatec C s> i }| }|d ur:|j D ]}||vr| �|�||< 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 urDt| j�D ] \}\}}|tkr |�|� q |j}q
|S r )r r r r8 r4 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, r2 r7 r9 r# r> r? rA 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_nodesrE r �r r r! �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s |j D ]}| j�|� qd S r ��bodyrE r rI r r r �visit_AssignBlock� s
z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� q d S )N)�call)ZexcluderH rI r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s |j D ]}| j�|� qd S r rL rI r r r �visit_OverlayScope� s
z RootVisitor.visit_OverlayScoperM c K s� |dkr"| j j|jdd� |j}nN|dkr2|j}n>|dkrh| j j|jdd� |jd urd| j �|j� d S td��|pvdD ]}| j �|� qxd S )NrM T)�store_as_param�else�testzUnknown for branchr )rE r r' rM �else_rT � 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 )�targetsrE r rM )r r r! r' rJ r r r �
visit_With� s
zRootVisitor.visit_Withc O s t d|jj ��d S )NzCannot find symbols for %r)�NotImplementedErrorr/ rB )r r �argsr! r r r �
generic_visit� s
�zRootVisitor.generic_visitN)rM )rB rC rD r rK Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifierrN rP rQ rX rZ r] 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 )rG rF 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 r7 r# N)�ctxrG r9 r" r7 r# )r r rR r! r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namec K s | j �|j� d S r )rG r# r" �r r r! r r r � visit_NSRef� s z FrameSymbolVisitor.visit_NSRefc sb �j |jfi � �� �j�� ��fdd�}||j�}||j�}||jpHd�}�j�|||g� d S )Nc s4 �� � �_}| D ]}�j|fi � �� q��_|S r )r2 rG r )r r+ Zsubnode�r! Zoriginal_symbolsr r r � inner_visit� s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr )r rT rG rM Zelif_rU r> )r r r! rh Z body_symbolsZ elif_symbolsZ else_symbolsr rg r rb � s
zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S r )rG r7 r" re r r r r_ � s z FrameSymbolVisitor.visit_Macroc K s$ | j |fi |�� | j�|j� d S r )r] rG r7 r' re r r r � visit_Import� s z FrameSymbolVisitor.visit_Importc K sJ | j |fi |�� |jD ],}t|t�r8| j�|d � q| j�|� qd S )Nr )r] �names�
isinstance�tuplerG r7 )r r r! r" r r r �visit_FromImport� s
z#FrameSymbolVisitor.visit_FromImportc K s, | j |jfi |�� | j |jfi |�� dS )z'Visit assignments in the correct order.N)r r r' re r r r � visit_Assign s z FrameSymbolVisitor.visit_Assignc K s | j |jfi |�� dS )zqVisiting stops at for blocks. However the block sequence
is visited as part of the outer scope.
N)r �iterre r r r rX s z FrameSymbolVisitor.visit_Forc K s | j |jfi |�� d S r )r rO re r r r rP
s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jfi |�� d S r )r �filterre r r r r` s z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr )r r r! r' r r r rZ s
z FrameSymbolVisitor.visit_Withc K s | j |jfi |�� dS )z Stop visiting at block assigns.N)r r' re r r r rN s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr re r r r ra s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr re r r r r^ s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr re r r r rQ ! s z%FrameSymbolVisitor.visit_OverlayScopeN)F)rB rC rD �__doc__r rd rf rb r_ ri rm rn rX rP r` rZ rN ra r^ rQ r r r r r
� s"
r
)N)N)Z_compatr r r r8 r: r5 r6 r r r- r r r
r r r r �<module> s
8