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�param�resolve�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# �identr 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 r6 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 r7 )r r8 r& �VAR_LOAD_PARAMETERr* r r r �declare_parameterp s zSymbols.declare_parameterc C s$ | � |�d kr | j|t|fd� d S r7 )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 r5 r �items�lenr) r, r r9 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 rA r< r8 r )r r- r r' �instrr$ r r r �dump_param_targets� s zSymbols.dump_param_targets)NN)N)�__name__�
__module__� __qualname__r/ �Optional�intr r �Node�Anyr �str�Tupler& r( r) r. r6 r; r= r# �SequencerD �DictrF �SetrH 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 rV r r r r � s zRootVisitor.__init__r c K s |� � D ]}| j�|� qd S r ��iter_child_nodesrW r �r r r �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s |j D ]}| j�|� qd S r ��bodyrW r r[ r r r �visit_AssignBlock� s
z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� q d S )N)�call)�excluderY r[ r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s |j D ]}| j�|� qd S r r^ r[ r r r �visit_OverlayScope� s
z RootVisitor.visit_OverlayScoper_ )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 )Nr_ T)�store_as_param�else�testzUnknown for branch)rW r r' r_ �else_rh � RuntimeError)r r re r rC �itemr r r � visit_For� s
zRootVisitor.visit_Forc K s4 |j D ]}| j�|� q|jD ]}| j�|� q d S r )�targetsrW r r_ )r r r r' r\ 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�typerI )r r ro r r r r �
generic_visit� s zRootVisitor.generic_visit)r_ ) rI rJ rK r r rN r/ rO r] Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifier� AssignBlockr` � CallBlockrc � OverlayScoperd �ForrP rl �Withrn rr 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 NrU c C s
|| _ d S r )rV rX r r r r � s zFrameSymbolVisitor.__init__F)r rf 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 r; r# N)�ctxrV r= r" r; r# )r r rf r r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namer c K s | j �|j� d S r )rV 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 )r6 rV r )r r- Zsubnode�r Zoriginal_symbolsr r r � inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r rh rV r/ �Iterabler rN r_ Zelif_ri rD )r r r r� Z body_symbolsZ elif_symbolsZ else_symbolsr r� r rw � s
zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S r )rV r; r" r r r r rt s z FrameSymbolVisitor.visit_Macroc K s | j |f|� | j�|j� d S r )rr rV r; 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 )rr �names�
isinstance�tuplerV r; )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 rl # s z FrameSymbolVisitor.visit_Forc K s | j |jf|� d S r )r ra r r r r rc ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf|� d S r )r �filterr r r r ru , s z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| �|� qd S r )�valuesr )r r r r' r r r rn / s
z FrameSymbolVisitor.visit_Withc K s | j |jf|� dS )z Stop visiting at block assigns.N)r r' r r r r r` 3 s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr r r r r rv 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr r r r r rs : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr r r r r rd = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rI rJ rK �__doc__r r �Name�boolr/ rO r~ ZNSRefr� �Ifrw ZMacrort �Importr� Z
FromImportr� �Assignr� r{ rl ry rc Z FilterBlockru r| rn rx r` ZScoperv ZBlockrs rz rd r r r r r
� s. � � r
)N)N)�typingr/ � r r r r< r> r9 r: r� rN rL r r r r! r
r r r r �<module> s, �
� � � :