File: //opt/alt/python38/lib/python3.8/site-packages/jinja2/__pycache__/idtracking.cpython-38.pyc
U
^�a�) � @ s� d dl ZddlmZ ddlmZ dZdZdZdZ dej
ej ej d d d
�d d �Z
dej ej d d d
�dd�ZG dd � d �ZG dd� de�ZG dd� de�ZdS )� N� )�nodes)� NodeVisitor�paramZresolve�aliasZ undefined�Symbols)r �parent_symbols�returnc C s* t |d�}t|�}| D ]}|�|� q|S �N)�parent)r �FrameSymbolVisitor�visit)r r �sym�visitor�node� r �B/opt/alt/python38/lib/python3.8/site-packages/jinja2/idtracking.py� find_symbols s
r )r r r c C s t |d�}|�| � |S r
)r � analyze_node)r r r r r r �symbols_for_node s
r c @ s* e Zd Zd#ejd eje dd�dd�Zejej dd�dd�Z
d$e ejej e eje f e d�d d
�Z
e ejej d �d d
�Ze eje d�dd�Ze e d�dd�Zd d�dd�Ze dd�dd�Ze e d�dd�Ze dd�dd�Zejd dd �d d �Zeje e f d�d d �Zeje d�d!d"�ZdS )%r N)r �levelr c C sD |d kr |d krd}n
|j d }|| _ || _i | _i | _t� | _d S )Nr r )r r �refs�loads�set�stores)�selfr r r r r �__init__ s
zSymbols.__init__�r �kwargsr c K s t | �}|j|f|� d S �N)� RootVisitorr
)r r r r r r r r . s zSymbols.analyze_node)�name�loadr c C s2 d| j � d|� �}|| j|< |d k r.|| j|< |S )NZl_�_)r r r )r r! r" Zidentr r r � _define_ref2 s
zSymbols._define_ref)�targetr c C s. || j kr| j | S | jd k r*| j�|�S d S r )r r � find_load)r r% r r r r& ; s
zSymbols.find_load)r! r c C s. || j kr| j | S | jd k r*| j�|�S d S r )r r �find_ref�r r! r r r r' D s
zSymbols.find_refc C s& | � |�}|d kr"td|�d���|S )NzFTried to resolve a name to a reference that was unknown to the frame (�))r' �AssertionError)r r! �rvr r r �refM s
�z Symbols.ref)r c C sJ t �tt�| j��}|j�| j� | j� � |_| j
� � |_
| j � � |_ |S r ) �t�castr �object�__new__� __class__�__dict__�updater �copyr r )r r+ r r r r4 V 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 �store^ s
z
Symbols.storec C s | j �|� | j|td fd�S r5 )r r6 r$ �VAR_LOAD_PARAMETERr( r r r �declare_parameterp s zSymbols.declare_parameterc C s$ | � |�d kr | j|t|fd� d S r5 )r' r$ �VAR_LOAD_RESOLVEr( r r r r" t s z Symbols.load)�branch_symbolsr c C s� i }|D ]0}|j D ]$}|| j kr"q|�|d�d ||< qq|D ].}| j�|j� | j�|j� | j �|j � q>|�� 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 r3 r �items�lenr' r* r r7 r< ) r r= r �branchr% r r! Z branch_countZ outer_targetr r r �
branch_updatex s*
zSymbols.branch_updatec C sB i }| }|d k r>t |j�D ]}||kr| �|�||< q|j}q|S r )�sortedr r' r )r r+ r r! r r r � dump_stores� s zSymbols.dump_storesc C sH t � }| }|d k rD| j�� D ] \}\}}|tkr |�|� q |j}q
|S r )r r r? r: r6 r )r r+ r r% Zinstrr# r r r �dump_param_targets� s zSymbols.dump_param_targets)NN)N)�__name__�
__module__� __qualname__r- �Optional�intr r �Node�Anyr �strZTupler$ r& r' r, r4 r9 r; r" �SequencerB ZDictrD �SetrE r r r r r s0 � � � �
c @ s� e Zd Zddd�dd�Zejejdd�dd�ZeZ eZ
eZ eZ eZ
eZeZejejdd�d d
�Zejejdd�d d �Zejejdd�d
d�Zdejeejdd�dd�Zejejdd�dd�Zejejejdd�dd�ZdS )r r N��symbolsr c C s t |�| _d S r )r � sym_visitor�r rQ r r r r � s zRootVisitor.__init__r c K s |� � D ]}| j�|� qd S r �Ziter_child_nodesrR r
�r r r �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s |j D ]}| j�|� qd S r ��bodyrR r
rU r r r �visit_AssignBlock� s
z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� q d S )N)�call)ZexcluderT rU r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s |j D ]}| j�|� qd S r rX rU r r r �visit_OverlayScope� s
z RootVisitor.visit_OverlayScoperY )r �
for_branchr r c K s� |dkr"| j j|jdd� |j}nN|dkr2|j}n>|dkrh| j j|jdd� |jd k rd| j �|j� d S td��|r�|D ]}| j �|� qxd S )NrY T)�store_as_param�else�testzUnknown for branch)rR r
r% rY �else_ra � RuntimeError)r r r^ r rA �itemr r r � visit_For� s
zRootVisitor.visit_Forc K s4 |j D ]}| j�|� q|jD ]}| j�|� q d S r )ZtargetsrR r
rY )r r r r% rV r r r �
visit_With� s
zRootVisitor.visit_With)r �argsr r c O s t dt|�j����d S )NzCannot find symbols for )�NotImplementedError�typerF )r r rg r r r r �
generic_visit� s zRootVisitor.generic_visit)rY ) rF rG rH r r rK r- rL rW Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifier� AssignBlockrZ � CallBlockr\ � OverlayScoper] �ForrM re �Withrf rj r r r r r � s* � � r c @ sn e Zd ZdZddd�dd�Zd(ejeej dd�d d
�Z
ej ej dd �d d
�Z ej
ej dd �dd�Zejej dd �dd�Zejej dd �dd�Zejej dd �dd�Zejej dd �dd�Zejej dd �dd�Zejej dd �dd�Zejej dd �d d �Z ej ej dd �d d �Z ej ej dd �d d!�Z ej!ej dd �d"d#�Z"ej#ej dd �d$d%�Z$ej%ej dd �d&d'�Z&dS ))r z A visitor for `Frame.inspect`.r NrP c C s
|| _ d S r )rQ rS r r r r � s zFrameSymbolVisitor.__init__F)r r_ r r c 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 r9 r" N)ZctxrQ r; r! r9 r" )r r r_ r r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namer c K s | j �|j� d S r )rQ r" r! �r r r r r r � visit_NSRef� s z FrameSymbolVisitor.visit_NSRefc sn �j |jf� � �j�tjtj dd�� ��fdd� }||j�}||j�}||j pTd�}�j�
|||g� d S )Nr )r r c s0 �� � �_}| D ]}�j|f� � q��_|S r )r4 rQ r
)r r+ Zsubnode�r Zoriginal_symbolsr r r � inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r
ra rQ r- �Iterabler rK rY Zelif_rb rB )r r r ry Z body_symbolsZ elif_symbolsZ else_symbolsr rx r ro � s
zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S r )rQ r9 r! rv r r r rl s z FrameSymbolVisitor.visit_Macroc K s | j |f|� | j�|j� d S r )rj rQ r9 r% rv 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 )rj �names�
isinstance�tuplerQ r9 )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% rv 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
�iterrv r r r re # s z FrameSymbolVisitor.visit_Forc K s | j |jf|� d S r )r
r[ rv r r r r\ ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf|� d S r )r
�filterrv r r r rm , s z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr
)r r r r% r r r rf / s
z FrameSymbolVisitor.visit_Withc K s | j |jf|� dS )z Stop visiting at block assigns.N)r
r% rv r r r rZ 3 s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr rv r r r rn 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr rv r r r rk : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr rv r r r r] = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rF rG rH �__doc__r r �Name�boolr- rL ru ZNSRefrw ZIfro ZMacrorl ZImportr{ Z
FromImportr ZAssignr� rs re rq r\ Z FilterBlockrm rt rf rp rZ ZScopern ZBlockrk rr r] r r r r r � s. � � r )N)N)�typingr- � r r r r: r< r7 r8 rz rK rI r r r r r r r r r �<module> s, �
� � � :