CasperSecurity
o
^�a�) � @ s� d dl ZddlmZ ddlmZ dZdZdZdZ dd ej
ej d
ej d d d fd
d�Z
ddej d
ej d d d fdd�ZG dd � d �ZG dd� de�ZG dd� de�ZdS )� N� )�nodes)� NodeVisitor�param�resolve�alias� undefinedr �parent_symbols�Symbols�returnc C s* t |d�}t|�}| D ]}|�|� q |S �N)�parent)r
�FrameSymbolVisitor�visit)r r �sym�visitor�node� r �3/usr/lib/python3/dist-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 . s zSymbols.analyze_node�name�loadc C s2 d| j � d|� �}|| j|< |d ur|| j|< |S )N�l_�_)r r r )r r" r# �identr r r � _define_ref2 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, D 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 �refM s
��z Symbols.refc C sJ t �tt�| j��}|j�| j� | j� � |_| j
� � |_
| j � � |_ |S r ) �t�castr
�object�__new__� __class__�__dict__�updater �copyr r )r r0 r r r r9 V 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" � outer_refr r r �store^ s
�z
Symbols.storec C s | j �|� | j|td fd�S r: )r r; r' �VAR_LOAD_PARAMETERr- r r r �declare_parameterp s zSymbols.declare_parameterc C s( | � |�d u r| j|t|fd� d S d S r: )r, r' �VAR_LOAD_RESOLVEr- r r r r# t s �z Symbols.load�branch_symbolsc C s� i }|D ]}|j D ]}|| j v rq |�|d�d ||< q q|D ]}| j�|j� | j�|j� | j �|j � q |�� D ]6\}}|t|�krFq;| �|�}|d usSJ d��| jd urj| j�|�}|d urjt |f| j|< q;t
|f| j|< q;d S )Nr r zshould not happen) r �getr r8 r �items�lenr, r
r< rB ) r rC r �branchr( r r" � branch_count� outer_targetr r r �
branch_updatex s.
�
�zSymbols.branch_updatec C sH i }| }|d ur"t |j�D ]
}||vr| �|�||< q
|j}|d us|S r )�sortedr r, r
)r r0 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 rE r@ r; r
)r r0 r r( �instrr% r r r �dump_param_targets� s
��zSymbols.dump_param_targets)NNr )r r
)�__name__�
__module__� __qualname__r2 �Optional�intr r �Node�Anyr �str�Tupler' r* r, r1 r9 r? rA r# �SequencerJ �DictrL �SetrN 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 r[ r r r r � s zRootVisitor.__init__r r c K s |� � D ]}| j�|� qd S r ��iter_child_nodesr\ r �r r r �childr r r �
_simple_visit� s �zRootVisitor._simple_visitc K � |j D ]}| j�|� qd S r ��bodyr\ r r` r r r �visit_AssignBlock� �
�z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� qd S )N)�call)�excluder^ r` r r r �visit_CallBlock� s �zRootVisitor.visit_CallBlockc K rc r rd r` r r r �visit_OverlayScope� rg z RootVisitor.visit_OverlayScopere �
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 )Nre T)�store_as_param�else�testzUnknown for branch)r\ r r( re �else_ro � RuntimeError)r r rl r rG �itemr r r � visit_For� s
�zRootVisitor.visit_Forc K s4 |j D ]}| j�|� q|jD ]}| j�|� qd S r )�targetsr\ r re )r r r r( ra r r r �
visit_With� s
�zRootVisitor.visit_With�argsc O s t dt|�j����)NzCannot find symbols for )�NotImplementedError�typerO )r r rv r r r r �
generic_visit� s zRootVisitor.generic_visit�r[ r
r N)re ) rO rP rQ r r rT r2 rU rb �visit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_If� visit_ScopedEvalContextModifier� AssignBlockrf � CallBlockrj � OverlayScoperk �ForrV rs �Withru ry 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`.r[ r
r Nc C s
|| _ d S r )r[ r] r r r r � s
zFrameSymbolVisitor.__init__Fr rm 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)�ctxr[ rA r" r? r# )r r rm r r r r �
visit_Name� s
�z FrameSymbolVisitor.visit_Namec K � | j �|j� d S r )r[ 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 )r9 r[ r )r r0 �subnode�r �original_symbolsr r r � inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r ro r[ r2 �Iterabler rT re �elif_rp rJ )r r r r� � body_symbols� elif_symbols� else_symbolsr r� r r� � s "
zFrameSymbolVisitor.visit_Ifc K r� r )r[ r? r" r� r r r r} r� z FrameSymbolVisitor.visit_Macroc K s$ | j |fi |�� | j�|j� d S r )ry r[ 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 )ry �names�
isinstance�tupler[ 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 rs # s z FrameSymbolVisitor.visit_Forc K � | j |jfi |�� d S r )r rh r� r r r rj ) � z"FrameSymbolVisitor.visit_CallBlockc K r� r )r �filterr� r r r r~ , r� z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr )r r r r( r r r ru / s
�z FrameSymbolVisitor.visit_Withc K r� )z Stop visiting at block assigns.N)r r( r� r r r rf 3 s z$FrameSymbolVisitor.visit_AssignBlockc K � dS )zStop visiting at scopes.Nr r� r r r r 7 � z FrameSymbolVisitor.visit_Scopec K r� )zStop visiting at blocks.Nr r� r r r r| : r� z FrameSymbolVisitor.visit_Blockc K r� )z!Do not visit into overlay scopes.Nr r� r r r rk = r� z%FrameSymbolVisitor.visit_OverlayScoperz )F)'rO rP rQ �__doc__r r �Name�boolr2 rU r� �NSRefr� �Ifr� �Macror} �Importr� �
FromImportr� �Assignr� r� rs r� rj � FilterBlockr~ r� ru r� rf �Scoper �Blockr| r� rk r r r r r � s6
����
� r r )�typingr2 � r r r r@ rB r< r= r� rT rR r r r
r! r r r r r �<module> s6 �
��
� ���
� :