← Back
Editing: idtracking.cpython-37.opt-1.pyc
B ^�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|�}x| D ]}|�|� qW |S )N)�parent)r �FrameSymbolVisitor�visit)r r �sym�visitor�node� r �B/opt/alt/python37/lib/python3.7/site-packages/jinja2/idtracking.py� find_symbols s r )r r r c C s t |d�}|�| � |S )N)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 )N)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 )N)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 )N) �t�castr �object�__new__� __class__�__dict__�updater �copyr r )r r( r r r r1 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 )N)r )r r2 r" �VAR_LOAD_PARAMETER)r r r r r �declare_parameterp s zSymbols.declare_parameterc C s$ | � |�d kr | j|t|fd� d S )N)r )r% r" �VAR_LOAD_RESOLVE)r r r r r r t s z Symbols.load)�branch_symbolsr c C s� i }x<|D ]4}x.|j D ]$}|| j kr&q|�|d�d ||< qW q W x6|D ].}| j�|j� | j�|j� | j �|j � qHW xx|�� D ]l\}}|t|�kr�q�| �|�}|d k s�td��| j d k r�| j �|�}|d k r�t |f| j|< q�t |f| j|< q�W d S )Nr r zshould not happen) r �getr r0 r �items�lenr% r' r r3 r8 ) r r9 r �branchr# r r Z branch_countZ outer_targetr r r � branch_updatex s* zSymbols.branch_updatec C sJ i }| }x<|d k rDx(t |j�D ]}||kr | �|�||< q W |j}q W |S )N)�sortedr r% r )r r( r r r r r � dump_stores� s zSymbols.dump_storesc C sP t � }| }x@|d k rJx,| j�� D ] \}\}}|tkr |�|� q W |j}q W |S )N)r r r; r6 r2 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) r1 r5 r7 r �Sequencer> ZDictr@ �SetrA 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 )N)r � sym_visitor)r rL r r r r � s zRootVisitor.__init__)r r r c K s" x |� � D ]}| j�|� q W d S )N)�iter_child_nodesrM r )r r r �childr r r � _simple_visit� s zRootVisitor._simple_visitc K s x|j D ]}| j�|� qW d S )N)�bodyrM r )r r r rO r r r �visit_AssignBlock� s z RootVisitor.visit_AssignBlockc K s& x |j dd�D ]}| j�|� qW d S )N)�call)Zexclude)rN rM r )r r r rO r r r �visit_CallBlock� s zRootVisitor.visit_CallBlockc K s x|j D ]}| j�|� qW d S )N)rQ rM r )r r r rO r r r �visit_OverlayScope� s z RootVisitor.visit_OverlayScoperQ )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�x|D ]}| j �|� qzW d S )NrQ T)�store_as_param�else�testzUnknown for branch)rM r r# rQ �else_rY � RuntimeError)r r rV r r= �itemr r r � visit_For� s zRootVisitor.visit_Forc K s< x|j D ]}| j�|� qW x|jD ]}| j�|� q$W d S )N)�targetsrM r rQ )r r r r# rO 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�typerB )r r r` r r r r � generic_visit� s zRootVisitor.generic_visit)rQ ) rB rC rD r r rG r* rH rP Zvisit_Template� visit_Block� visit_Macro�visit_FilterBlock� visit_Scope�visit_IfZ visit_ScopedEvalContextModifier� AssignBlockrR � CallBlockrT � OverlayScoperU �ForrI r] �Withr_ rc 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 N)rL r c C s || _ d S )N)rL )r rL r r r r � s zFrameSymbolVisitor.__init__F)r rW 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 r5 r N)ZctxrL r7 r r5 r )r r rW r r r r � visit_Name� s z FrameSymbolVisitor.visit_Name)r r r c K s | j �|j� d S )N)rL 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 s4 �� � �_}x| D ]}�j|f� � qW ��_|S )N)r1 rL r )r r( Zsubnode)r �original_symbolsr r r � inner_visit s z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r rY rL r* �Iterabler rG rQ Zelif_rZ r> )r r r rq Z body_symbolsZ elif_symbolsZ else_symbolsr )r rp r r rh � s zFrameSymbolVisitor.visit_Ifc K s | j �|j� d S )N)rL r5 r )r r r r r r re s z FrameSymbolVisitor.visit_Macroc K s | j |f|� | j�|j� d S )N)rc rL r5 r# )r r r r r r � visit_Import s z FrameSymbolVisitor.visit_Importc K sJ | j |f|� x6|jD ],}t|t�r6| j�|d � q| j�|� qW d S )Nr )rc �names� isinstance�tuplerL r5 )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 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 �iter)r r r r r r r] # s z FrameSymbolVisitor.visit_Forc K s | j |jf|� d S )N)r rS )r r r r r r rT ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf|� d S )N)r �filter)r r r r r r rf , s z$FrameSymbolVisitor.visit_FilterBlockc K s x|j D ]}| �|� qW d S )N)�valuesr )r r r r# r r r r_ / s z FrameSymbolVisitor.visit_Withc K s | j |jf|� dS )z Stop visiting at block assigns.N)r r# )r r r r r r rR 3 s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr )r r r r r r rg 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr )r r r r r r rd : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr )r r r r r r rU = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rB rC rD �__doc__r r �Name�boolr* rH rn ZNSRefro ZIfrh ZMacrore ZImportrs Z FromImportrw ZAssignrx rl r] rj rT Z FilterBlockrf rm r_ ri rR ZScoperg ZBlockrd rk rU r r r r r � s$ r )N)N)�typingr* � r r r r6 r8 r3 r4 rr rG rE r r r r r r r r r �<module> s :
Save File
Cancel