3
U'ícá) ã @ s d dl ZddlmZ ddlmZ dZdZdZdZ dej
ej ej d d d
d d Z
dej ej d d d
ddZG dd d ZG dd deZG dd deZdS )é Né )Únodes)Ú NodeVisitorÚparamÚresolveÚaliasZ undefinedÚSymbols)r Úparent_symbolsÚreturnc C s. t |d}t|}x| D ]}|j| qW |S )N)Úparent)r ÚFrameSymbolVisitorÚvisit)r r ÚsymÚvisitorÚnode© r ú3/tmp/pip-build-gk9425m9/Jinja2/jinja2/idtracking.pyÚ find_symbols s
r )r r r
c C s t |d}|j| |S )N)r )r Ú analyze_node)r r r r r r Úsymbols_for_node s
r c @ s* e Zd Zd#ejd eje ddddZejej ddddZ
d$e ejej e eje f e dd d
Z
e ejej d d d
Ze eje dddZe e dddZd dddZe ddddZe e dddZe ddddZejd dd d d Zeje e f dd d Zeje dd!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*| jj|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*| jj|S d S )N)r r Úfind_ref)r r r r r r& D s
zSymbols.find_refc C s& | j |}|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 jttj| j}|jj| j | jj |_| j
j |_
| j j |_ |S )N) ÚtÚcastr ÚobjectÚ__new__Ú __class__Ú__dict__Úupdater Úcopyr r )r r) r r r r2 V s z Symbols.copyc C s` | j j| || jkr\| jd k rJ| jj|}|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| | j|td fdS )N)r )r r3 r# ÚVAR_LOAD_PARAMETER)r r r r r Údeclare_parameterp s zSymbols.declare_parameterc C s$ | j |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|j|dd ||< qW q
W x6|D ].}| jj|j | jj|j | j j|j qHW xx|j D ]l\}}|t|krq| j|}|d k s´td| j d k râ| j j|}|d k rât
|f| j|< qt |f| j|< qW d S )Nr r zshould not happen) r Úgetr r1 r ÚitemsÚlenr& r( r r4 r9 ) r r: 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 |jD ]}||kr | j|||< 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,| jj D ] \}\}}|tkr |j| q W |j}q W |S )N)r r r< r7 r3 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* r2 r6 r8 r ÚSequencer? ÚDictrA ÚSetrC r r r r r s
c @ sÚ e Zd ZdddddZejejddddZeZ eZ
eZ eZ eZ
eZeZejejddd d
Zejejddd d Zejejddd
dZdejeejddddZejejddddZejejejddddZdS )r r N)Úsymbolsr
c C s t || _d S )N)r Ú sym_visitor)r rP r r r r ¯ s zRootVisitor.__init__)r r r
c K s" x |j D ]}| jj| q
W d S )N)Úiter_child_nodesrQ r
)r r r Úchildr r r Ú
_simple_visit² s zRootVisitor._simple_visitc K s x|j D ]}| jj| qW d S )N)ÚbodyrQ r
)r r r rS r r r Úvisit_AssignBlock¾ s z RootVisitor.visit_AssignBlockc K s& x |j ddD ]}| jj| qW d S )NÚcall)Úexclude)rW )rR rQ r
)r r r rS r r r Úvisit_CallBlock s zRootVisitor.visit_CallBlockc K s x|j D ]}| jj| qW d S )N)rU rQ r
)r r r rS r r r Úvisit_OverlayScopeÆ s z RootVisitor.visit_OverlayScoperU )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|j d S td|rx|D ]}| j j| qzW d S )NrU T)Ústore_as_paramÚelseÚtestzUnknown for branch)rQ r
r$ rU Úelse_r^ Ú RuntimeError)r r r[ r r> Úitemr r r Ú visit_ForÊ s
zRootVisitor.visit_Forc K s< x|j D ]}| jj| qW x|jD ]}| jj| q$W d S )N)ÚtargetsrQ r
rU )r r r r$ rS r r r Ú
visit_WithÞ s zRootVisitor.visit_With)r Úargsr r
c O s t dt|jd S )NzCannot find symbols for )ÚNotImplementedErrorÚtyperD )r r re r r r r Ú
generic_visitä s zRootVisitor.generic_visit)rU ) rD rE rF r r rI r+ rJ rT Zvisit_TemplateÚ visit_BlockÚ visit_MacroÚvisit_FilterBlockÚ visit_ScopeÚvisit_IfZ visit_ScopedEvalContextModifierÚ AssignBlockrV Ú CallBlockrY Ú OverlayScoperZ ÚForrK rb ÚWithrd rh r r r r r ® s r c @ sn e Zd ZdZdddddZd(ejeej ddd d
Z
ej ej dd d d
Z ej
ej dd ddZejej dd ddZejej dd ddZejej dd ddZejej dd ddZejej dd ddZejej dd ddZejej 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)rP r
c C s
|| _ d S )N)rP )r rP r r r r ë s zFrameSymbolVisitor.__init__F)r r\ r r
c K sT |s|j dkr | jj|j n2|j dkr8| jj|j n|j dkrP| jj|j dS )z2All assignments to names go through this function.r r6 r N)ÚctxrP r8 r r6 r )r r r\ r r r r Ú
visit_Nameî s
z FrameSymbolVisitor.visit_Name)r r r
c K s | j j|j d S )N)rP r r )r r r r r r Ú visit_NSRefù s z FrameSymbolVisitor.visit_NSRefc sn j |jf jtjtj dd fdd }||j}||j}||j pTf }jj
|||g d S )Nr )r r
c s4 j _}x| D ]}j|f qW _|S )N)r2 rP r
)r r) Zsubnode)r Úoriginal_symbolsr r r Ú inner_visit s
z0FrameSymbolVisitor.visit_If.<locals>.inner_visit) r
r^ rP r+ ÚIterabler rI rU Zelif_r_ r? )r r r rw Z body_symbolsZ elif_symbolsZ else_symbolsr )r rv r r rm ü s
zFrameSymbolVisitor.visit_Ifc K s | j j|j d S )N)rP r6 r )r r r r r r rj s z FrameSymbolVisitor.visit_Macroc K s | j |f| | jj|j d S )N)rh rP r6 r$ )r r r r r r Ú visit_Import s z FrameSymbolVisitor.visit_Importc K sJ | j |f| x6|jD ],}t|tr6| jj|d q| jj| qW d S )Nr )rh ÚnamesÚ
isinstanceÚtuplerP r6 )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 rb # s z FrameSymbolVisitor.visit_Forc K s | j |jf| d S )N)r
rW )r r r r r r rY ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf| d S )N)r
Úfilter)r r r r r r rk , s z$FrameSymbolVisitor.visit_FilterBlockc K s x|j D ]}| j| qW d S )N)Úvaluesr
)r r r r$ r r r rd / 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 rV 3 s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr )r r r r r r rl 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr )r r r r r r ri : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr )r r r r r r rZ = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rD rE rF Ú__doc__r r ÚNameÚboolr+ rJ rt ZNSRefru ÚIfrm ZMacrorj ÚImportry Z
FromImportr} ÚAssignr~ rq rb ro rY Z FilterBlockrk rr rd rn rV ZScoperl ZBlockri rp rZ r r r r r è s$ r )N)N)Útypingr+ Ú r r r r7 r9 r4 r5 rx rI rG r r r r r r r r r Ú<module> s :