Current File : //lib/python3.9/site-packages/ansible/_vendor/jinja2/__pycache__/idtracking.cpython-39.pyc
a
��+b�) � @ 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 �E/usr/lib/python3.9/site-packages/ansible/_vendor/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 u r |d u rd}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|fi |�� 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 ur.|| j|< |S )NZl_�_)r r r )r r" r# �identr r r � _define_ref2 s
zSymbols._define_ref)�targetr
c C s. || j v r| j | S | jd ur*| 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 v r| j | S | jd ur*| j�|�S d S r )r r �find_ref�r r" r r r r) D s
zSymbols.find_refc C s& | � |�}|d u r"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 sB t �| j�}|j�| j� | j�� |_| j�� |_| j�� |_|S r ) �object�__new__� __class__�__dict__�updater �copyr r )r r- r r r r4 V 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 �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 u r | 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 v r"q|�|d�d ||< qq|D ].}| j�|j� | j�|j� | j �|j � q>|�� D ]l\}}|t|�kr�qv| �|�}|d us�J d��| jd ur�| j�|�}|d ur�t |f| j|< qvt
|f| j|< qvd S )Nr r zshould not happen) r �getr r3 r �items�lenr) 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 ur>t |j�D ]}||vr| �|�||< 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 urD| 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__�t�Optional�intr r �Node�Anyr �str�Tupler& r( r) r. r4 r9 r; r# �SequencerB �DictrD �SetrE r r r r r s* �� ��
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 rT r r r r � s zRootVisitor.__init__r c K s |� � D ]}| j�|� qd S r �Ziter_child_nodesrU r �r r r �childr r r �
_simple_visit� s zRootVisitor._simple_visitc K s |j D ]}| j�|� qd S r ��bodyrU r rX r r r �visit_AssignBlock� s
z RootVisitor.visit_AssignBlockc K s" |j dd�D ]}| j�|� q d S )N)�call)ZexcluderW rX r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s |j D ]}| j�|� qd S r r[ rX 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 urd| j �|j� d S td��|r�|D ]}| j �|� qxd S )Nr\ T)�store_as_param�else�testzUnknown for branch)rU 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�|� q d S r )�targetsrU r r\ )r r r r' rY 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 rk r r r r �
generic_visit� s zRootVisitor.generic_visit)r\ ) rF rG rH r r rL rI rM rZ Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifier� AssignBlockr] � CallBlockr_ � OverlayScoper` �ForrN rh �Withrj rn 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 NrS c C s
|| _ d S r )rT rV r r r r � s zFrameSymbolVisitor.__init__F)r rb 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)�ctxrT r; r" r9 r# )r r rb r r r r �
visit_Name� s
z FrameSymbolVisitor.visit_Namer c K s | j �|j� d S r )rT r# r" �r r r r r r � visit_NSRef� s z FrameSymbolVisitor.visit_NSRefc sr �j |jfi � �� �j�tjtj dd�� ��fdd� }||j�}||j�}||j pXd�}�j�
|||g� d S )Nr )r r
c s4 �� � �_}| D ]}�j|fi � �� q��_|S r )r4 rT r )r r- Zsubnode�r Zoriginal_symbolsr r r � inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r rd rT rI �Iterabler rL r\ Zelif_re rB )r r r r~ Z body_symbolsZ elif_symbolsZ else_symbolsr r} r rs � s
zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S r )rT r9 r" r{ r r r rp s z FrameSymbolVisitor.visit_Macroc K s$ | j |fi |�� | j�|j� d S r )rn rT r9 r' r{ 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 )rn �names�
isinstance�tuplerT r9 )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 s | 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 s | j |jfi |�� d S r )r r^ r{ r r r r_ ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jfi |�� d S r )r �filterr{ r r r rq , s 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 s | j |jfi |�� 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 rr 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr r{ r r r ro : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr r{ r r r r` = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rF rG rH �__doc__r r �Name�boolrI rM rz ZNSRefr| ZIfrs ZMacrorp ZImportr� Z
FromImportr� ZAssignr� rw rh ru r_ Z FilterBlockrq rx rj rt r] ZScoperr ZBlockro rv r` r r r r r
� s* �
� r
)N)N)�typingrI � r r r r: r< r7 r8 r rL rJ r r r r! r
r r r r �<module> s( �� � � :
Mini Shell