o
    hx                  
   @   s   d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ d	Zd
dhZddheZddddddddde f
ddZG dd dZG dd dZG dd dZedZdd ZG dd dZdS )zVprovides functionality for rendering a parsetree constructing into module
source code.    N)ast)
exceptions)filters)	parsetree)util)PythonPrinter
   	UNDEFINEDSTOP_RENDERINGcontextloopTFc                 C   s<   t  }t|}t|t|||||||||	|
||  | S )zYGenerate module source code given a parsetree node,
    uri, and optional source filename)r   FastEncodingBufferr   _GenerateRenderMethod_CompileContextgetvalue)nodeurifilenamedefault_filtersbuffer_filtersimportsfuture_importssource_encodinggenerate_magic_commentstrict_undefinedenable_loopreserved_namesbufprinter r   `/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/mako/codegen.pycompile   s(   r!   c                   @   s   e Zd Zdd ZdS )r   c                 C   sF   || _ || _|| _|| _|| _|| _|| _|| _|	| _|
| _	|| _
d S N)r   r   r   r   r   r   r   r   r   r   r   )selfr   r   r   r   r   r   r   r   r   r   r   r   r   r    __init__H   s   
z_CompileContext.__init__N)__name__
__module____qualname__r$   r   r   r   r    r   G   s    r   c                   @   s   e Zd ZdZdd Zdd Zedd Zdd	 Zd
d Z	dd Z
dd Zdd Zd7ddZdd Zdd Z	d8ddZ		d9ddZdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 ZdS ):r   zZA template visitor object which generates the
    full module source for a template.

    c                 C   s  || _ || _|| _d g| _t|tjtjf| _| jrMd|j	 }|
 }t|jjdk}t|jdd}t|jdd}d }	d }
|jrL|jsL|dg7 }n>|  }	| jj}
d}|
d ur|
j
 }|
jjsi|dg7 }t|
jdd}| jjp~t|
jdd| j_ndg}d	}d	 }}|d u rd
g}n
dd d
g| D }| |
p|||||| |	d ur|	D ]}t||| q| js|   d S d S )Nz	render_%sr   bufferedFalsecached
**pageargsrender_bodyr   Fr   c                 S   s   g | ]}|qS r   r   .0ar   r   r    
<listcomp>   s    z2_GenerateRenderMethod.__init__.<locals>.<listcomp>)r   compilerr   identifier_stack
isinstancer   DefTagBlockTagin_deffuncnameget_argument_expressionslenfilter_argsargseval
attributesgetis_blockis_anonymouswrite_toplevelpagetag	body_declkwargsr   write_render_callabler   write_metadata_struct)r#   r   r1   r   namer;   filteredr(   r*   defsrB   r   r   r    r$   j   sT   





z_GenerateRenderMethod.__init__c                 C   sR   t | jj| jj| jj< | jj| jj| jj| jjd}| jddt	
|d d S )N)r   r   r   line_mapz"""__M_BEGIN_METADATAz__M_END_METADATA
""")maxr   
source_maplinenor1   r   r   r   
writelinesjsondumps)r#   structr   r   r    rF      s   z+_GenerateRenderMethod.write_metadata_structc                 C   s
   | j d S )N)r2   r#   r   r   r    identifiers   s   
z!_GenerateRenderMethod.identifiersc                    s  g  i g dj _G  fddd}| }jjD ]}|| qj _t }D ]	}|| }q.t	j }||_
j jrRj jrRjdj j  j jrdjddj jf  jd jd jd	 jd
 jd jdt  jdt   jdj j  jdj j  jdj j  jdj j  j jrd}j jD ]}||d 7 }j| qtj|ddddd}nd}|j}	|j}
|
|	j|_|j
t |r|j
|j |j _jddd |	j D   jd t r&! t  r8" # d  n
t rB" t$|	j S )zxTraverse a template structure for module-level directives and
        generate the start of module-level code.

        Nc                       s<   e Zd Z fddZfddZfddZfddZd	S )
z:_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevelc                    s     | d S r"   )appendsr   )inheritr   r    visitInheritTag      zJ_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitInheritTagc                    s   | |j < d S r"   rG   rW   )
namespacesr   r    visitNamespaceTag   r[   zL_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitNamespaceTagc                    s   | j _d S r"   )r1   rB   rW   rT   r   r    visitPageTag   s   zG_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitPageTagc                    s   |j r
 | d S d S r"   )ismodulerV   rW   )module_coder   r    	visitCode   s   zD_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitCodeN)r%   r&   r'   rZ   r^   r_   rb   r   rY   ra   r]   r#   r   r    FindTopLevel   s
    rd   z# -*- coding:%s -*-zfrom __future__ import %s, z(from mako import runtime, filters, cachezUNDEFINED = runtime.UNDEFINEDz'STOP_RENDERING = runtime.STOP_RENDERINGz__M_dict_builtin = dictz__M_locals_builtin = localsz_magic_number = %rz_modified_time = %rz_enable_loop = %rz_template_filename = %rz_template_uri = %rz_source_encoding = %r 
r   ztemplate defined imports)sourcerN   posr   z_exports = %rc                 S      g | ]}|j qS r   r\   )r.   nr   r   r    r0         z8_GenerateRenderMethod.write_toplevel.<locals>.<listcomp>   rS   )%r1   rB   r   nodesaccept_visitorr]   setuniondeclared_identifiers_Identifiersdeclaredr   r   r   	writeliner   joinMAGIC_NUMBERtimer   r   r   r   r   
PythonCodebranchtopleveldefsupdateTOPLEVEL_DECLAREDrU   valueswrite_blanksr9   write_module_codewrite_namespaceswrite_inheritlist)r#   rd   frk   module_identmodule_identifiersr   impimpcodemain_identifiersmitr   rc   r    rA      s   









z$_GenerateRenderMethod.write_toplevelc           	      C   sd  | j r|j}|r| jd|  | j|j | jd|d|f dd |s,|s,|r2| jd | j	| j
j| j | j rE| jjrRd|v rR| jd jd	 | j swt| jjd
kset| jjd
krw| jdddd | jjD   | j| jdd | jjD ]}||  q| | j||| | jd | jd |r| j||||| jdd dS dS )znwrite a top-level render callable.

        this could be the main render() method or that of a top-level def.z@runtime._decorate_toplevel(%s)def %s(%s):,/__M_caller = context.caller_stack._push_frame()try:context._push_buffer()r+   rS   pageargsr   z!__M_locals = __M_dict_builtin(%s)c                 S   s   g | ]}d ||f qS )%s=%sr   r.   xr   r   r    r0   B  s    
z?_GenerateRenderMethod.write_render_callable.<locals>.<listcomp>T)toplevelNrm   )r6   	decoratorr   ru   start_sourcerN   rO   rv   r2   rV   r1   rU   rz   r   r?   argument_declaredaddr9   locally_assignedwrite_variable_declaresrn   ro   write_def_finishr   write_cache_decorator)	r#   r   rG   r;   r(   rH   r*   r   rk   r   r   r    rE     sT   

z+_GenerateRenderMethod.write_render_callablec                 C   s"   |D ]}| j j|j|jd qdS )zewrite module-level template code, i.e. that which
        is enclosed in <%! %> tags in the template.starting_linenoN)r   write_indented_blocktextrN   )r#   ra   rk   r   r   r    r   V  s   z'_GenerateRenderMethod.write_module_codec                 C   s    | j ddd|jd  d dS )z:write the module-level inheritance-determination callable.z%def _mako_inherit(template, context):"_mako_generate_namespaces(context)z8return runtime._inherit_from(context, %s, _template_uri)fileN)r   rO   parsed_attributesr#   r   r   r   r    r   \  s   z#_GenerateRenderMethod.write_inheritc              
      s  j dddddddd j d | D ]}d|jv r"d	j_j |j t	|j
rpj d
 g  jj|d	_G  fddd}| }|j
D ]}|| qQj dd   j d d_d}nd}d|jv rj d|j|jdd|f  n"d|jv rj d|j||jddf  nj d|j|f  t|jddrj d|j  j dt|j  j d qt	|sوj d j d dS )z5write the module-level namespace-generating callable.z'def _mako_get_namespace(context, name):r   z+return context.namespaces[(__name__, name)]except KeyError:r   Nz'def _mako_generate_namespaces(context):importTzdef make_namespace():c                       s,   e Zd Zdd Zdd Z fddZdS )z<_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitorc                 S      |  | d S r"   visitDefOrBaserW   r   r   r    visitDefTag  r[   zH_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefTagc                 S   r   r"   r   rW   r   r   r    visitBlockTag  r[   zJ_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitBlockTagc                    s:   |j rtj	di |jj|dd  |j d S )N.Can't put anonymous blocks inside <%namespace>Fnested)r   )r@   r   CompileExceptionexception_kwargswrite_inline_defrV   r7   rW   exportrU   r#   r   r    r     s   zK_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefOrBaseNr%   r&   r'   r   r   r   r   r   r   r    NSDefVisitor  s    r   return [%s]r   Fzmake_namespace()Noner   zns = runtime.TemplateNamespace(%r, context._clean_inheritance_tokens(), templateuri=%s, callables=%s,  calling_uri=_template_uri)modulezyns = runtime.ModuleNamespace(%r, context._clean_inheritance_tokens(), callables=%s, calling_uri=_template_uri, module=%s)zhns = runtime.Namespace(%r, context._clean_inheritance_tokens(), callables=%s, calling_uri=_template_uri)inheritabler)   zcontext['self'].%s = nsz'context.namespaces[(__name__, %s)] = ns   pass)r   rO   ru   r~   r=   r1   has_ns_importsr   rN   r9   rn   rU   rz   r6   ro   rv   r   rG   r>   r<   reprr   )r#   r]   r   r   visrk   callable_namer   r   r    r   g  sz   





z&_GenerateRenderMethod.write_namespacesFNc           
      C   s@  dd |j D }t }||j}|dd |j D }||j}||j}| j	j
r8d|v }|d nd}|durC||}|rwt| j	ddrw| jd	 d
| j	_| j	j D ]\}}d|jv rv| jd|td|jd f  q\|r| jd |D ]}||v r|| }	|	jr|	js| |	| q| j|	|d
d q|	 r| |	| q| j|	|d
d q|| j	jv r| jd||f  qt| j	ddr| j	jr| jd||f d| dd||f dd| dd q| jd|||f  q| j	jr| jdd||f dd| d q| jd||f  q| jd dS )a  write variable declarations at the top of a function.

        the variable declarations are in the form of callable
        definitions for defs and/or name lookup within the
        function's context argument. the names declared are based
        on the names that are referenced in the function body,
        which don't otherwise have any explicit assignment
        operation. names that are assigned within the body are
        assumed to be locally-scoped variables and are not
        separately declared.

        for def callable definitions, if the def is a top-level
        callable then a 'stub' callable is generated which wraps
        the current Context into a closure. if the def is not
        top-level, it is fully rendered as a local closure.

        c                 S   s   i | ]}|j |qS r   r7   r.   cr   r   r    
<dictcomp>      zA_GenerateRenderMethod.write_variable_declares.<locals>.<dictcomp>c                 S   rj   r   r   r   r   r   r    r0     rl   zA_GenerateRenderMethod.write_variable_declares.<locals>.<listcomp>r   FNr   z_import_ns = {}Tr   z:_mako_get_namespace(context, %r)._populate(_import_ns, %r)z\s*,\s*z%loop = __M_loop = runtime.LoopStack()r   z%%s = _mako_get_namespace(context, %r)z"%s = _import_ns.get(%r, UNDEFINED)zif %s is UNDEFINED:r   z%s = context[%r]r   z&raise NameError("'%s' is not defined")z3%s = _import_ns.get(%r, context.get(%r, UNDEFINED))z%s = context.get(%r, UNDEFINED)z__M_writer = context.writer())rI   rp   rq   
undeclaredclosuredefsr~   
differencer   locally_declaredr1   r   discardintersectiongetattrr   ru   has_importsr]   itemsr=   resplitr?   r@   write_def_declr   is_rootr   rO   )
r#   rU   r   limitcomp_identsto_writehas_loopidentnscompr   r   r    r     s   

	




z-_GenerateRenderMethod.write_variable_declaresc                 C   s   |j }| }|jdd}| js't| jjdks t| jjdkr'|dd n|dd | j	d|d
|f  | j	d|d
|f  | j	d	 d	S )
z>write a locally-available callable referencing a top-level defTas_callr   zcontext._locals(__M_locals)r   r   r   zreturn render_%s(%s)N)r7   r8   r6   r9   rU   r   r   insertr   ru   rv   )r#   r   rU   r7   	namedeclsnameargsr   r   r    r   B  s   z$_GenerateRenderMethod.write_def_declc           
   	   C   s  |  }|j}|r| jd|  | jd|jd|f  t|jjdk}t	|j
dd}t	|j
dd}| jdd	 |sF|sF|rL| jd
 |j||d}| | | j| |jD ]}	|	|  qa| j  | |||| | jd |r| j||j|d|ddd dS dS )z?write a locally-available def callable inside an enclosing def.z&@runtime._decorate_inline(context, %s)r   r   r   r(   r)   r*   r   r   r   r   NFT)inliner   )r8   r   r   ru   r7   rv   r9   r:   r;   r<   r=   r>   rO   rz   r   r2   rV   rn   ro   popr   r   )
r#   r   rU   r   r   r   rH   r(   r*   rk   r   r   r    r   U  sJ   



z&_GenerateRenderMethod.write_inline_defTc                 C   s   |s|s|s| j d |r| j ddd |s|s|rp|s |r(| j dd n| j dd |r7| j d d}|rD| |jj|d}| j d |rW|sW| | jj|d}|s[|re| j d	|  dS | j d
| d dS dS )aJ  write the end section of a rendering function, either outermost or
        inline.

        this takes into account if the rendering function was filtered,
        buffered, etc.  and closes the corresponding try: block if any, and
        writes code to retrieve captured content, apply filters, send proper
        return value.	return ''finally:z!context.caller_stack._pop_frame()Nz__M_buf = context._pop_buffer()6__M_buf, __M_writer = context._pop_buffer_and_writer()__M_buf.getvalue()Fz	return %s__M_writer(%s))r   ru   rO   create_filter_callabler:   r;   r1   r   )r#   r   r(   rH   r*   	callstackrX   r   r   r    r     s>   

z&_GenerateRenderMethod.write_def_finishc                    sZ  j d||f   jdt|}i }	jjdur*|	fddjjjD  |	 fdd jD  d|	v rEtt	|	d |	d< j d|d	
|f  d
d |D }
j||  d |rd||d	
|
d
dd |	 D |f }jj|d}j d| d dS j d||d	
|
d
dd |	 D |f dd dS )zhwrite a post-function decorator to replace a rendering
        callable with a cached version of itself.z__M_%s = %s	cache_keyNc                 3   s<    | ]}| d r|dkr|dd  jjj| fV  qdS cache_r      N)
startswithr1   rB   r   r.   parT   r   r    	<genexpr>  s    z>_GenerateRenderMethod.write_cache_decorator.<locals>.<genexpr>c                 3   s8    | ]}| d r|dkr|dd  j| fV  qdS r   )r   r   r   )node_or_pagetagr   r    r     s    timeoutr   r   c                 S   s0   g | ]}d |v rd| d d fd  n|qS )=r   r   rm   )r   r-   r   r   r    r0     s    $z?_GenerateRenderMethod.write_cache_decorator.<locals>.<listcomp>)r   r   z`context.get('local').cache._ctx_get_or_create(%s, lambda:__M_%s(%s),  context, %s__M_defname=%r)rf   c                 S      g | ]
\}}d ||f qS z%s=%s, r   r.   kvr   r   r    r0         Fzreturn zk__M_writer(context.get('local').cache._ctx_get_or_create(%s, lambda:__M_%s(%s), context, %s__M_defname=%r))c                 S   r   r   r   r   r   r   r    r0     r   r   )r   ru   r   r>   r   r1   rB   r|   intr<   rv   r   undeclared_identifiersr   r   r   rO   )r#   r   rG   r;   r(   rU   r   r   cachekey
cache_args	pass_argsrX   r   )r   r#   r    r     sj   
z+_GenerateRenderMethod.write_cache_decoratorc           
      C   s   dd }d|vr$|r$| j jr| j jjj| }| j jr$d|vr$| j j| }|D ]0}|dkr-q&td|}|rF|dd\}}||}	|	| }n
||}|dusPJ d||f }q&|S )	zwrite a filter-applying expression based on the filters
        present in the given filter names, adjusting for the global
        'default' filter aliases as needed.c                 S   s"   t d| r
d|  S tj| | S )Nz
decode\..+zfilters.)r   matchr   DEFAULT_ESCAPESr>   r\   r   r   r    locate_encode  s   zC_GenerateRenderMethod.create_filter_callable.<locals>.locate_encoderk   z(.+?)(\(.*\))r   rm   Nz%s(%s))r1   rB   r:   r;   r   r   r   group)
r#   r;   targetis_expressionr   emr   fargsr   r   r   r    r   	  s&   
z,_GenerateRenderMethod.create_filter_callablec                 C   s   | j |j t|js | jjd urt| jjjjs t| jj	r6| 
|jjd|j d}| j d|  d S | j d|j  d S )Nz%sTr   )r   r   rN   r9   escapesr1   rB   r:   r;   r   r   escapes_coder   ru   )r#   r   rX   r   r   r    visitExpression)  s   
z%_GenerateRenderMethod.visitExpressionc                    s   j r"| jd  jr | jd | jd | jd  d S d S | jj | jjr9jdkr9t	| j}nj
}| j|    fdd} retdd  D rbtfdd D se| rm| jd	 d S d S )
Nr   zloop = __M_loop._exit()forc                     s2    D ]} t | tjrqt | tjr dS  dS d S )NTFr3   r   CommentControlLine)r   )childrenr   r    _search_for_control_lineP  s   zH_GenerateRenderMethod.visitControlLine.<locals>._search_for_control_linec                 s   s"    | ]}t |tjtjfV  qd S r"   r
  r   r   r   r    r   Z  s
    
z9_GenerateRenderMethod.visitControlLine.<locals>.<genexpr>c                 3   s.    | ]}t |tjr |jp|jV  qd S r"   )r3   r   r  
is_ternarykeywordisendr   )r   r   r    r   ^  s    

r   )r  r   ru   has_loop_contextr   rN   r1   r   r  mangle_mako_loopr   get_childrenall)r#   r   r   r  r   )r  r   r    visitControlLine:  s8   	z&_GenerateRenderMethod.visitControlLinec                 C   s(   | j |j | j dt|j  d S )Nr   )r   r   rN   ru   r   contentr   r   r   r    	visitTextg  s   z_GenerateRenderMethod.visitTextc              
   C   sh   t |jjdk}|r| jdd |jD ]}||  q|r2| jddd| |jjdd d  d S d S )	Nr   z#__M_writer = context._push_writer()r   r   r   r   r   F)r9   r:   r;   r   rO   rn   ro   r   )r#   r   rH   rk   r   r   r    visitTextTagk  s&   

z"_GenerateRenderMethod.visitTextTagc                 C   sp   |j s2| jj|j|jd | js4t| jjdkr6| j	d | j	dd
dd | D   d S d S d S d S )Nr   r   z0__M_locals_builtin_stored = __M_locals_builtin()z__M_locals.update(__M_dict_builtin([(__M_key, __M_locals_builtin_stored[__M_key]) for __M_key in [%s] if __M_key in __M_locals_builtin_stored]))r   c                 S   s   g | ]}t |qS r   )r   r   r   r   r    r0     r   z3_GenerateRenderMethod.visitCode.<locals>.<listcomp>)r`   r   r   r   rN   r6   r9   rU   r   ru   rv   rr   r   r   r   r    rb   ~  s   z_GenerateRenderMethod.visitCodec                 C   sV   | j |j |jd}|r| j d|jd |f  d S | j d|jd   d S )Nr;   z5runtime._include_file(context, %s, _template_uri, %s)r   z1runtime._include_file(context, %s, _template_uri))r   r   rN   r=   r>   ru   r   )r#   r   r;   r   r   r    visitIncludeTag  s   z%_GenerateRenderMethod.visitIncludeTagc                 C      d S r"   r   r   r   r   r    r^        z'_GenerateRenderMethod.visitNamespaceTagc                 C   r  r"   r   r   r   r   r    r     r  z!_GenerateRenderMethod.visitDefTagc                 C   sp   |j r| jd|j  d S |jdd}|dg7 }| jd|j  | jd|jd|f  | jd d S )	Nz%s()Tr   r+   zOif 'parent' not in context._data or not hasattr(context._data['parent'], '%s'):zcontext['self'].%s(%s)r   rg   )r@   r   ru   r7   r8   rv   )r#   r   r   r   r   r    r     s   
z#_GenerateRenderMethod.visitBlockTagc                 C   r   r"   visitCallTagr   r   r   r    visitCallNamespaceTag  s   z+_GenerateRenderMethod.visitCallNamespaceTagc                    s`  j d dgjj|ddj|dd  d j  G  fddd}| }|jD ]}|| q5j	  |j
 }j d	d
|  d}|r]j dd   j  |jD ]}| qkj	  j||dddd j d dd
 d  j dd j |j j dg |jd ddd  d S )Nzdef ccall(caller):bodyTr   Fcallerc                       s.   e Zd Zdd Zdd Z fddZdS )z6_GenerateRenderMethod.visitCallTag.<locals>.DefVisitorc                 S   r   r"   r   rW   r   r   r    r     r[   zB_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitDefTagc                 S   r   r"   r   rW   r   r   r    r     r[   zD_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitBlockTagc                    s@   j |dd |js|j |j jv r j|j= d S d S )NFr   )r   r@   rV   r7   r   rW   body_identifierscallable_identifiersr   r#   r   r    r     s   zE_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitDefOrBaseNr   r   r"  r   r    
DefVisitor  s    r%  zdef body(%s):r   r   r   )r   r   zccontext.caller_stack.nextcaller = runtime.Namespace('caller', context, callables=ccall(__M_caller))r   r   z&context.caller_stack.nextcaller = None)r   ru   rU   rz   add_declaredr2   rV   rn   ro   r   rC   r8   rv   rO   r   r   r   rN   r   
expression)r#   r   r%  r   rk   bodyargsr(   r   r"  r    r    sH   






z"_GenerateRenderMethod.visitCallTag)FN)T)FF)r%   r&   r'   __doc__r$   rF   propertyrU   rA   rE   r   r   r   r   r   r   r   r   r   r  r  r  r  rb   r  r^   r   r   r  r  r   r   r   r    r   c   s>    1
m9
]~/
:
S -r   c                   @   s   e Zd ZdZd(ddZdd Zedd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' ZdS ))rs   zCtracks the status of identifier names as template code is rendered.NFc                 C   s
  |d urBt |tjrt | _t | _n7t|jdd |j	
 D |j|j| _|r7| j|j| _tjdi |j| _n	t | _t | _|| _t | _t | _t | _t | _t | _	|| _|d uro||  | jj| j}|rtdd| d S )Nc                 S   rj   r   r\   r   r   r   r    r0     rl   z)_Identifiers.__init__.<locals>.<listcomp>z'Reserved words declared in template: %sre   r   )r3   r   NamespaceTagrp   rt   r   SetLikeDictr{   rq   r   r~   r   r   r   r1   r   r   ro   r   r   r   NameConflictErrorrv   )r#   r1   r   parentr   illegal_namesr   r   r    r$     sB   


z_Identifiers.__init__c                 K   s   t | j|| fi |S )zUcreate a new Identifiers for a new Node, with
        this Identifiers as the parent.)rs   r1   )r#   r   rD   r   r   r    rz   H  s   z_Identifiers.branchc                 C   s   t | j| j S r"   )rp   r{   rq   r   r~   rT   r   r   r    rI   N  s   z_Identifiers.defsc                 C   sH   dt | jt | jt | jdd | j D dd | j D | jf S )NzrIdentifiers(declared=%r, locally_declared=%r, undeclared=%r, topleveldefs=%r, closuredefs=%r, argumentdeclared=%r)c                 S   rj   r   r\   r   r   r   r    r0   [  rl   z)_Identifiers.__repr__.<locals>.<listcomp>c                 S   rj   r   r\   r   r   r   r    r0   \  rl   )r   rt   r   r   r{   r~   r   r   rT   r   r   r    __repr__R  s   z_Identifiers.__repr__c                 C   sR   |  D ]}|dkr|| j| jvr| j| q| D ]}| j| qdS )zlupdate the state of this Identifiers with the undeclared
        and declared identifiers of the given node.r   N)r   rt   rq   r   r   r   rr   r#   r   r   r   r   r    check_declareda  s   z_Identifiers.check_declaredc                 C   s*   | j | || jv r| j| d S d S r"   )rt   r   r   remove)r#   r   r   r   r    r&  m  s   
z_Identifiers.add_declaredc                 C   r   r"   r2  r   r   r   r    r  r  r[   z_Identifiers.visitExpressionc                 C   r   r"   r4  r   r   r   r    r  u  r[   z_Identifiers.visitControlLinec                 C   s*   |j s| | | j| | _d S d S r"   )r`   r2  r   rq   rr   r   r   r   r    rb   x  s   

z_Identifiers.visitCodec                 C   s(   | j |u r|jD ]	}||  qd S d S r"   )r   rn   ro   r#   r   rk   r   r   r    r^     s
   

z_Identifiers.visitNamespaceTagc                 C   sX   | |j}|||j< |d ur&||ur(|js|jr*tjd|j fi |jd S d S d S )Nz<%%def or %%block named '%s' already exists in this template.)r>   r7   r?   r   r   r   )r#   
collectionr   existingr   r   r    _check_name_exists  s"   
z_Identifiers._check_name_existsc                 C   s   |  r|js| | j| n|| jur| | j| | D ]}|dkr4|| j| j	vr4| j
| q|| ju rR| D ]}| j| q>|jD ]	}||  qJd S d S Nr   )r   r@   r8  r{   r   r   r   rt   rq   r   r   r   rr   r   rn   ro   r#   r   r   rk   r   r   r    r     s"   


z_Identifiers.visitDefTagc                 C   s  || j ur8|js8t| j tjr tjd|j| j jf fi |jt| j tj	tj
fr8tjd|jf fi |j| D ]}|dkrQ|| j| jvrQ| j| q<|jsd| | j| | j|j n|| j urp| | j| | D ]}| j| qt|jD ]}||  qd S )Nz/Named block '%s' not allowed inside of def '%s'z3Named block '%s' not allowed inside of <%%call> tagr   )r   r@   r3   r   r4   r   r   rG   r   CallTagCallNamespaceTagr   rt   rq   r   r   r   r8  r{   r7   r   rr   r   rn   ro   r:  r   r   r    r     sF   

z_Identifiers.visitBlockTagc                 C   s8   |  D ]}|dkr|| j| jvr| j| qd S r9  )r   rt   rq   r   r   r   r1  r   r   r    r    s   z_Identifiers.visitTextTagc                 C   r   r"   r4  r   r   r   r    r    r[   z_Identifiers.visitIncludeTagc                 C   s(   |  D ]}| j| q| | d S r"   )rr   r   r   r2  r1  r   r   r    r_     s   z_Identifiers.visitPageTagc                 C   r   r"   r  r   r   r   r    r    r[   z"_Identifiers.visitCallNamespaceTagc                 C   s   || j u r9| D ]}|dkr|| j| jvr| j| q	| D ]}| j| q#|j	D ]}|
|  q/d S | D ]}|dkrR|| j| jvrR| j| q=d S r9  )r   r   rt   rq   r   r   r   rr   r   rn   ro   r:  r   r   r    r    s&   

z_Identifiers.visitCallTag)NNF)r%   r&   r'   r)  r$   rz   r*  rI   r0  r2  r&  r  r  rb   r^   r8  r   r   r  r  r_   r  r  r   r   r   r    rs     s*    
C
!rs   z^for\s+((?:\(?)\s*(?:\(?)\s*[A-Za-z_][A-Za-z_0-9]*(?:\s*,\s*(?:[A-Za-z_][A-Za-z_0-9]*),??)*\s*(?:\)?)(?:\s*,\s*(?:(?:\(?)\s*[A-Za-z_][A-Za-z_0-9]*(?:\s*,\s*(?:[A-Za-z_][A-Za-z_0-9]*),??)*\s*(?:\)?)),??)*\s*(?:\)?))\s+in\s+(.*):c                 C   sr   t  }| | |jr4d| jd _t| j}|r-|d|	d d d|	d }|S t
d| j | j}|S )	zconverts a for loop into a context manager wrapped around a for loop
    when access to the `loop` variable has been detected in the for loop body
    TrS   zloop = __M_loop._enter(%s)rm   r   zfor %s in loop:r   zCouldn't apply loop context: %s)LoopVariablero   detectedrn   r  	_FOR_LOOPr   r   rO   r   SyntaxError)r   r   loop_variabler   r   r   r   r    r    s   
r  c                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )r=  zQA node visitor which looks for the name 'loop' within undeclared
    identifiers.c                 C   s
   d| _ d S )NF)r>  rT   r   r   r    r$     s   
zLoopVariable.__init__c                 C   s2   d|  v rd| _d S | D ]}||  qd S )Nr   T)r   r>  r  ro   r5  r   r   r    _loop_reference_detected  s
   
z%LoopVariable._loop_reference_detectedc                 C   r   r"   rB  r   r   r   r    r     r[   zLoopVariable.visitControlLinec                 C   r   r"   rC  r   r   r   r    rb   #  r[   zLoopVariable.visitCodec                 C   r   r"   rC  r   r   r   r    r  &  r[   zLoopVariable.visitExpressionN)	r%   r&   r'   r)  r$   rB  r  rb   r  r   r   r   r    r=    s    r=  )r)  rP   r   rx   makor   r   r   r   r   
mako.pygenr   rw   r}   rq   RESERVED_NAMES	frozensetr!   r   r   rs   r?  r  r=  r   r   r   r    <module>   sP   
(       % o