3
G\# @ sp d dl mZ d dlmZ dZdZdZdZddd Zdd
d Z G d d
d
e
Z G dd deZ G dd deZ
dS ) ) NodeVisitor) iteritemsparamZresolvealiasZ undefinedNc C s. t |d}t|}x| D ]}|j| qW |S )N)parent)SymbolsFrameSymbolVisitorvisit)nodesparent_symbolssymvisitornode r /usr/lib/python3.6/idtracking.py find_symbols s
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 @ sx e Zd Zd ddZdd Zd ddZdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS ) r Nc C sD |d kr |d krd}n
|j d }|| _ || _i | _i | _t | _d S )Nr )levelr refsloadssetstores)selfr r r r r __init__ s
zSymbols.__init__c K s t | }|j|f| d S )N) RootVisitorr )r r kwargsr
r r r r ' s zSymbols.analyze_nodec C s. d| j |f }|| j|< |d k r*|| j|< |S )Nzl_%d_%s)r r r )r nameloadZidentr r r _define_ref+ s
zSymbols._define_refc C s. || j kr| j | S | jd k r*| jj|S d S )N)r r find_load)r targetr r r r! 2 s
zSymbols.find_loadc C s. || j kr| j | S | jd k r*| jj|S d S )N)r r find_ref)r r r r r r# 8 s
zSymbols.find_refc C s" | j |}|d kr td| |S )NzITried to resolve a name to a reference that was unknown to the frame (%r))r# AssertionError)r r rvr r r ref> s
z Symbols.refc C sB t j| j}|jj| j | jj |_| jj |_| jj |_|S )N) object__new__ __class____dict__updater copyr r )r r% r r r r, E 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_ALIASVAR_LOAD_UNDEFINED)r r Z outer_refr r r storeM s
z
Symbols.storec C s | j j| | j|td fdS )N)r )r r- r VAR_LOAD_PARAMETER)r r r r r declare_parameter_ 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 r c s
z Symbols.loadc 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 xxt|D ]l\}}|t|krq| j|}|d k std| j d k r| j j|}|d k rt
|f| j|< qt |f| j|< qW d S )Nr r zshould not happen) r getr r+ r r lenr# r$ r r. r3 ) r Zbranch_symbolsr branchr" r r Z branch_countZ outer_targetr r r
branch_updateh s*
zSymbols.branch_updatec C sF i }| }x8|d k r@x$|j D ]}||kr| j|||< qW |j}q
W |S )N)r r# r )r r% r r r r r dump_stores s
zSymbols.dump_storesc C sP t }| }x@|d k rJx,t| jD ] \}\}}|tkr |j| q W |j}q W |S )N)r r r r1 r- r )r r% r r" Zinstr_r r r dump_param_targets s
zSymbols.dump_param_targets)NN)N)__name__
__module__ __qualname__r r r r! r# r& r, r0 r2 r r7 r8 r: r r r r r s
r c @ sj e Zd Zdd Zdd Ze Z Z Z Z Z Z
Z dd Z dd Z
d d
Zdd d
Zdd Zdd ZdS )r c C s t || _d S )N)r sym_visitor)r symbolsr r r r s zRootVisitor.__init__c K s" x |j D ]}| jj| q
W d S )N)iter_child_nodesr> r )r r r childr r r
_simple_visit s zRootVisitor._simple_visitc K s x|j D ]}| jj| qW d S )N)bodyr> r )r r r rA r r r visit_AssignBlock s z RootVisitor.visit_AssignBlockc K s& x |j ddD ]}| jj| qW d S )Ncall)exclude)rE )r@ r> r )r r r rA r r r visit_CallBlock s zRootVisitor.visit_CallBlockc K s x|j D ]}| jj| qW d S )N)rC r> r )r r r rA r r r visit_OverlayScope s z RootVisitor.visit_OverlayScoperC c K s |dkr"| j j|jdd |j}nN|dkr2|j}n>|dkrh| j j|jdd |jd k rd| j j|j d S tdx |pxf D ]}| j j| qzW d S )NrC T)store_as_paramelsetestzUnknown for branch)r> r r" rC else_rK RuntimeError)r r Z
for_branchr r6 itemr r r visit_For s
zRootVisitor.visit_Forc K s< x|j D ]}| jj| qW x|jD ]}| jj| q$W d S )N)Ztargetsr> r rC )r r r r" rA r r r
visit_With s zRootVisitor.visit_Withc O s t d|jj d S )NzCannot find symbols for %r)NotImplementedErrorr) r; )r r argsr r r r
generic_visit s zRootVisitor.generic_visitN)rC )r; r< r= r rB Zvisit_Template visit_Block visit_Macrovisit_FilterBlock visit_Scopevisit_IfZ visit_ScopedEvalContextModifierrD rG rH rO rP rS r r r r r s
r c @ s e Zd ZdZdd Zd$ddZdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#S )%r z A visitor for `Frame.inspect`.c C s
|| _ d S )N)r? )r r? r r r r s zFrameSymbolVisitor.__init__Fc 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 r0 r N)Zctxr? r2 r r0 r )r r rI r r r r
visit_Name s
z FrameSymbolVisitor.visit_Namec K s | j j|j d S )N)r? r r )r r r r r r visit_NSRef s z FrameSymbolVisitor.visit_NSRefc s^ j |jf j fdd}||j}||j}||jpDf }jj|||g d S )Nc s4 j _}x| D ]}j|f qW _|S )N)r, r? r )r
r% Zsubnode)r original_symbolsr r r inner_visit s
z0FrameSymbolVisitor.visit_If..inner_visit)r rK r? rC Zelif_rL r7 )r r r r\ Z body_symbolsZ elif_symbolsZ else_symbolsr )r r[ r r rX s
zFrameSymbolVisitor.visit_Ifc K s | j j|j d S )N)r? r0 r )r r r r r r rU s z FrameSymbolVisitor.visit_Macroc K s | j |f| | jj|j d S )N)rS r? r0 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 )rS names
isinstancetupler? r0 )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 rO s z FrameSymbolVisitor.visit_Forc K s | j |jf| d S )N)r rE )r r r r r r rG s z"FrameSymbolVisitor.visit_CallBlockc K s | j |jf| d S )N)r filter)r r r r r r rV s z$FrameSymbolVisitor.visit_FilterBlockc K s x|j D ]}| j| qW d S )N)valuesr )r r r r" r r r rP 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 rD s z$FrameSymbolVisitor.visit_AssignBlockc K s dS )zStop visiting at scopes.Nr )r r r r r r rW s z FrameSymbolVisitor.visit_Scopec K s dS )zStop visiting at blocks.Nr )r r r r r r rT s z FrameSymbolVisitor.visit_Blockc K s dS )z!Do not visit into overlay scopes.Nr )r r r r r r rH s z%FrameSymbolVisitor.visit_OverlayScopeN)F)r; r< r= __doc__r rY rZ rX rU r] ra rb rO rG rV rP rD rW rT rH r r r r r s"
r )N)N)Zjinja2.visitorr Zjinja2._compatr r1 r3 r. r/ r r r' r r r r r r r s
~4