U
|
Þ=®dÐ) ã @ 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|}| D ]}| |¡ q|S ©N)Úparent)r ÚFrameSymbolVisitorÚvisit)r r ÚsymÚvisitorÚnode© r úHd:\z\workplace\vscode\pyvenv\venv\Lib\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 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*| 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 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 |¡ || 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 fdS r5 )r r6 r& ÚVAR_LOAD_PARAMETERr* r r r Údeclare_parameterp s zSymbols.declare_parameterc C s$ | |¡d kr | 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 kr"q| |d¡d ||< qq|D ].}| j |j¡ | j |j¡ | j |j ¡ q>| ¡ D ]l\}}|t|krqv| |¡}|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 r3 r ÚitemsÚlenr) r, 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 k r>t |jD ]}||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 r? r: r6 r )r r- r r' Úinstrr$ 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 ÚSetrF r r r r r s0 ÿ þ ÿ þ 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 r )r Ú sym_visitor©r rU r r r r ¯ s zRootVisitor.__init__r c K s | ¡ D ]}| j |¡ qd S r ©Ziter_child_nodesrV r ©r r r Úchildr r r Ú _simple_visit² s zRootVisitor._simple_visitc K s |j D ]}| j |¡ qd S r ©ÚbodyrV r rY r r r Úvisit_AssignBlock¾ s
|
z RootVisitor.visit_AssignBlockc K s" |j ddD ]}| j |¡ q d S )N)Úcall)ÚexcluderX rY r r r Úvisit_CallBlock s zRootVisitor.visit_CallBlockc K s |j D ]}| j |¡ qd S r r\ rY 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)rV r r' r] Úelse_rf Ú RuntimeError)r r rc r rA Úitemr r r Ú visit_ForÊ s
|
zRootVisitor.visit_Forc K s4 |j D ]}| j |¡ q|jD ]}| j |¡ q d S r )ÚtargetsrV r r] )r r r r' rZ r r r Ú
|
visit_WithÞ s
|
|
zRootVisitor.visit_With)r Úargsr r
|
c O s t dt|jd S )NzCannot find symbols for )ÚNotImplementedErrorÚtyperG )r r rm r r r r Ú generic_visitä s zRootVisitor.generic_visit)r] ) rG rH rI r r rM rJ rN r[ Zvisit_TemplateÚ visit_BlockÚ visit_MacroÚvisit_FilterBlockÚ visit_ScopeÚvisit_IfZ visit_ScopedEvalContextModifierÚ AssignBlockr^ Ú CallBlockra Ú OverlayScoperb ÚForrO rj ÚWithrl rp 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 NrT c C s
|
|| _ d S r )rU rW r r r r ë s zFrameSymbolVisitor.__init__F)r rd 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)ÚctxrU r; r" r9 r# )r r rd r r r r Ú
|
visit_Nameî s
|
|
z FrameSymbolVisitor.visit_Namer c K s | j |j¡ d S r )rU r# r" ©r r r r r r Ú visit_NSRefù s z FrameSymbolVisitor.visit_NSRefc sn j |jf jtjtj dd fdd }||j}||j}||j pTd}j
|
|||g¡ d S )Nr )r r
|
c s0 ¡ _}| D ]}j|f q_|S r )r4 rU r )r r- Zsubnode©r Zoriginal_symbolsr r r Ú inner_visit s
|
z0FrameSymbolVisitor.visit_If.<locals>.inner_visitr ) r rf rU rJ ÚIterabler rM r] Zelif_rg rB )r r r r Z body_symbolsZ elif_symbolsZ else_symbolsr r r ru ü s
|
|
zFrameSymbolVisitor.visit_Ifc K s | j |j¡ d S r )rU r9 r" r} r r r rr s z FrameSymbolVisitor.visit_Macroc K s | j |f| | j |j¡ d S r )rp rU r9 r' r} r r r Ú visit_Import s z FrameSymbolVisitor.visit_Importc K sF | j |f| |jD ],}t|tr4| j |d ¡ q| j |¡ qd S )Nr )rp ÚnamesÚ
|
isinstanceÚtuplerU r9 )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 rj # s z FrameSymbolVisitor.visit_Forc K s | j |jf| d S r )r r_ r} r r r ra ) s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf| d S r )r Úfilterr} r r r rs , s z$FrameSymbolVisitor.visit_FilterBlockc K s |j D ]}| |¡ qd S r )Úvaluesr )r r r r' r r r rl / 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 rt 7 s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr r} r r r rq : s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr r} r r r rb = s z%FrameSymbolVisitor.visit_OverlayScope)F)'rG rH rI Ú__doc__r r ÚNameÚboolrJ rN r| ZNSRefr~ ZIfru ZMacrorr ZImportr Z
|
FromImportr ZAssignr ry rj rw ra Z FilterBlockrs rz rl rv r^ ZScopert ZBlockrq rx rb r r r r r è s. ÿ þ r )N)N)ÚtypingrJ Ú r r r r: r< r7 r8 r rM rK r r r r! r r r r r Ú<module> s, ÿ
|
þ ÿ þ :
|
