File: //opt/saltstack/salt/lib/python3.10/site-packages/jinja2/__pycache__/idtracking.cpython-310.pyc
o
�N�g;) � @ s� d dl ZddlmZ ddlmZ ejrd dlZdZ dZ
dZ dZ dd ej
ej d
ejd d d fd
d�Z ddejd
ejd d d fdd�ZG dd � d �ZG dd� de�ZG dd� de�ZdS )� N� )�nodes)� NodeVisitor�param�resolve�aliasZ undefinedr �parent_symbols�Symbols�returnc C s* t |d�}t|�}| D ]}|�|� q |S �N)�parent)r �FrameSymbolVisitor�visit)r r �sym�visitor�node� r �E/opt/saltstack/salt/lib/python3.10/site-packages/jinja2/idtracking.py� find_symbols s
r r c C s t |d�}|�| � |S r )r � analyze_node)r r r r r r �symbols_for_node s
r c @ sD e Zd Z d&dejd deje ddfdd�Zdejdej ddfd d
�Z
d'd e d ejej e eje f de fd
d�Z
de dejej fdd�Zd e deje fdd�Zd e de fdd�Zd(dd�Zd e ddfdd�Zd e de fdd �Zd e ddfd d �Zd ejd ddfd d!�Zdeje e f fd"d#�Zdeje fd$d%�ZdS ))r Nr �levelr
c C sD |d u r|d u r d}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 �kwargsc K s t | �}|j|fi |�� d S �N)� RootVisitorr )r r r r r r r r 1 s zSymbols.analyze_node�name�loadc C s2 d| j � d|� �}|| j|< |d ur|| j|< |S )NZl_�_)r r r )r r! r" �identr r r � _define_ref5 s
zSymbols._define_ref�targetc C �. || j v r
| j | S | jd ur| j�|�S d S r )r r � find_load)r r& r r r r( > �
zSymbols.find_loadc C r' r )r r �find_ref�r r! r r r r* G r) zSymbols.find_refc C s& | � |�}|d u rtd|�d���|S )NzFTried to resolve a name to a reference that was unknown to the frame (�))r* �AssertionError)r r! �rvr r r �refP s
��z Symbols.ref�te.Selfc C sB t �| j�}|j�| j� | j�� |_| j�� |_| j�� |_|S r ) �object�__new__� __class__�__dict__�updater �copyr r )r r. r r r r6 Y s z Symbols.copyc C sd | j �|� || jvr0| jd ur%| j�|�}|d ur%| j|t|fd� d S | j|td fd� d S d S �N)r" )r �addr r r* r% �VAR_LOAD_ALIAS�VAR_LOAD_UNDEFINED)r r! Z outer_refr r r �storea s
�z
Symbols.storec C s | j �|� | j|td fd�S r7 )r r8 r% �VAR_LOAD_PARAMETERr+ r r r �declare_parameters s zSymbols.declare_parameterc C s( | � |�d u r| j|t|fd� d S d S r7 )r* r% �VAR_LOAD_RESOLVEr+ r r r r" w s �z Symbols.load�branch_symbolsc C s� t � }|D ]}|�|j� q|�| j� |D ]}| j�|j� | j�|j� | j�|j� q|D ]-}| �|�}|d us?J d��| jd urV| j�|�}|d urVt|f| j|< q0t |f| j|< q0d S )Nzshould not happen)
r r5 r �difference_updater r r* r r9 r> )r r? r �branchr r! r& Z outer_targetr r r �
branch_update{ s$
�zSymbols.branch_updatec C sH i }| }|d ur"t |j�D ]
}||vr| �|�||< q
|j}|d us|S r )�sortedr r* r )r r. r r! r r r � dump_stores� s ��zSymbols.dump_storesc C sN t � }| }|d ur%| j�� D ]\}\}}|tkr |�|� q|j}|d us |S r )r r �itemsr< r8 r )r r. r r& Zinstrr# r r r �dump_param_targets� s
��zSymbols.dump_param_targets)NNr )r
r0 )�__name__�
__module__� __qualname__�t�Optional�intr r �Node�Anyr �strZTupler% r( r* r/ r6 r; r= r" �SequencerB ZDictrD �SetrF r r r r r ! s8 ���
����
�
c @ s� e Zd Zddd�Zdejdejddfd d
�ZeZ eZ
eZ eZ eZ
eZeZdejdejddfd d �Zdejdejddfd
d�Zdejdejddfdd�Z ddejdedejddfdd�Zdejdejddfdd�Zdejdejdejddfdd�ZdS ) r �symbolsr r
Nc C s t |�| _d S r )r
� sym_visitor�r rR r r r r � s zRootVisitor.__init__r r c K s |� � D ]}| j�|� qd S r �Ziter_child_nodesrS r �r r r �childr r r �
_simple_visit� s �zRootVisitor._simple_visitc K � |j D ]}| j�|� qd S r ��bodyrS r rV r r r �visit_AssignBlock� �
�z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� qd S )N)�call)ZexcluderU rV r r r �visit_CallBlock� s �zRootVisitor.visit_CallBlockc K rY r rZ rV r r r �visit_OverlayScope� r] z RootVisitor.visit_OverlayScoper[ �
for_branchc K s� |dkr| j j|jdd� |j}n'|dkr|j}n |dkr4| j j|jdd� |jd ur2| j �|j� d S td��|rE|D ]
}| j �|� q<d S d S )Nr[ T)�store_as_param�else�testzUnknown for branch)rS r r& r[ �else_rd � RuntimeError)r r ra r rA �itemr r r � visit_For� s
�zRootVisitor.visit_Forc K s4 |j D ]}| j�|� q|jD ]}| j�|� qd S r )�targetsrS r r[ )r r r r& rW r r r �
visit_With� s
�zRootVisitor.visit_With�argsc O s t dt|�j����)NzCannot find symbols for )�NotImplementedError�typerG )r r rk r r r r �
generic_visit� s zRootVisitor.generic_visit�rR r r
N)r[ ) rG rH rI r r rM rJ rN rX Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifier� AssignBlockr\ � CallBlockr_ � OverlayScoper` �ForrO rh �Withrj rn r r r r r � s2
����
�$r c @ s� e Zd ZdZd*dd�Z d+d ejd
ed ej ddfd d
�Z
d ej d ej ddfdd�Z d ej
d ej ddfdd�Zd ejd ej ddfdd�Zd ejd ej ddfdd�Zd ejd ej ddfdd�Zd ejd ej ddfdd�Zd ejd ej ddfdd�Zd ejd ej ddfd d �Zd ejd ej ddfd d �Z d ej d ej ddfd d!�Z d ej d ej ddfd"d#�Z d ej!d ej ddfd$d%�Z"d ej#d ej ddfd&d'�Z$d ej%d ej ddfd(d)�Z&dS ),r
z A visitor for `Frame.inspect`.rR r r
Nc C s
|| _ d S r )rR rT r r r r � s
zFrameSymbolVisitor.__init__Fr rb r c K s\ |s|j dkr| j�|j� dS |j dkr | j�|j� dS |j dkr,| j�|j� dS dS )z2All assignments to names go through this function.r r; r" N)�ctxrR r= r! r; r" )r r rb r r r r �
visit_Name� s
�z FrameSymbolVisitor.visit_Namec K � | j �|j� d S r )rR r" r! �r r r r r r � visit_NSRef� � z FrameSymbolVisitor.visit_NSRefc st �j |jfi � �� �j�dtjtj ddf� ��fdd� }||j�}||j�}||j p-d�}�j�
|||g� d S )Nr r
r c s4 �� � �_}| D ] }�j|fi � �� q ��_|S r )r6 rR r )r r. Zsubnode�r Zoriginal_symbolsr r r � inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r rd rR rJ �Iterabler rM r[ Zelif_re rB )r r r r� Z body_symbolsZ elif_symbolsZ else_symbolsr r� r rt � s "
zFrameSymbolVisitor.visit_Ifc K r| r )rR r; r! r} r r r rq r z FrameSymbolVisitor.visit_Macroc K s$ | j |fi |�� | j�|j� d S r )rn rR r; r& r} r r r � visit_Import s z FrameSymbolVisitor.visit_Importc K sJ | j |fi |�� |jD ]}t|t�r | j�|d � q | j�|� q d S )Nr )rn �names�
isinstance�tuplerR r; )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& r} r r r � visit_Assign s z FrameSymbolVisitor.visit_Assignc K � | j |jfi |�� dS )zqVisiting stops at for blocks. However the block sequence
is visited as part of the outer scope.
N)r �iterr} r r r rh # s z FrameSymbolVisitor.visit_Forc K � | j |jfi |�� d S r )r r^ r} r r r r_ ) � z"FrameSymbolVisitor.visit_CallBlockc K r� r )r �filterr} r r r rr , r� z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr )r r r r& r r r rj / s
�z FrameSymbolVisitor.visit_Withc K r� )z Stop visiting at block assigns.N)r r& r} r r r r\ 3 s z$FrameSymbolVisitor.visit_AssignBlockc K � dS )zStop visiting at scopes.Nr r} r r r rs 7 � z FrameSymbolVisitor.visit_Scopec K r� )zStop visiting at blocks.Nr r} r r r rp : r� z FrameSymbolVisitor.visit_Blockc K r� )z!Do not visit into overlay scopes.Nr r} r r r r` = r� z%FrameSymbolVisitor.visit_OverlayScopero )F)'rG rH rI �__doc__r r �Name�boolrJ rN r{ ZNSRefr~ ZIfrt ZMacrorq ZImportr� Z
FromImportr� ZAssignr� rx rh rv r_ Z FilterBlockrr ry rj ru r\ ZScopers ZBlockrp rw r` r r r r r
� s6
����
� r
r )�typingrJ � r r r Z
TYPE_CHECKINGZtyping_extensions�ter< r> r9 r: r� rM rK r r r r r
r r r r �<module> s: �
��
� ���
� :