B
    ×‹dÍ¸  ã            
   @   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é
   Ú	UNDEFINEDZ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   Z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   úY/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/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ršd|j	 }| 
¡ }t|jjƒdk}t|j dd¡ƒ}t|j dd¡ƒ}d }	d }
|jr˜|js˜|dg7 }n~|  ¡ }	| jj}
d}|
d k	r|
j 
¡ }|
jjsÔ|dg7 }t|
j dd¡ƒ}| jjpþt|
j dd¡ƒ| j_n
dg}d	}d	 }}|d kr*d
g}ndd„ d
g| D ƒ}|  |
pJ||||||¡ |	d k	r~x|	D ]}t|||ƒ qhW | jsŽ|  ¡  d S )Nz	render_%sr   ÚbufferedÚFalseÚcachedz
**pageargsZ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   ÚDefTagZ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   r+   r   Únamer4   Úfilteredr%   r'   Údefsr;   r   r   r   r!   j   sL    







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   Zline_mapz"""Z__M_BEGIN_METADATAz__M_END_METADATA
""")Úmaxr   Z
source_mapÚlinenor+   r   r   r   Ú
writelinesÚjsonÚdumps)r    Ústructr   r   r   r?   ›   s    z+_GenerateRenderMethod.write_metadata_structc             C   s
   | j d S )Néÿÿÿÿ)r,   )r    r   r   r   Úidentifiers¬   s    z!_GenerateRenderMethod.identifiersc                s¢  g ‰ i ‰g ‰dˆj _G ‡ ‡‡‡fdd„dƒ}|ƒ }xˆjjD ]}| |¡ q<W ˆˆj _tƒ }xˆD ]}| | ¡ ¡}qbW t	ˆj ƒ}||_
ˆj jr¬ˆj jr¬ˆj dˆj j ¡ ˆj jrÐˆ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rÐd}x*ˆj jD ]}||d 7 }ˆj |¡ q˜W t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 ˆ ƒr€ˆ "ˆ¡ ˆ #ˆ d ¡ nt ˆƒr”ˆ "ˆ¡ 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 )N)Úappend)Úsr   )Úinheritr   r   ÚvisitInheritTag¼   s    zJ_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitInheritTagc                s   |ˆ |j < d S )N)r@   )rL   r   )Ú
namespacesr   r   ÚvisitNamespaceTag¿   s    zL_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitNamespaceTagc                s   |ˆ j _d S )N)r+   r;   )rL   r   )r    r   r   ÚvisitPageTagÂ   s    zG_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitPageTagc                s   |j rˆ  |¡ d S )N)ÚismodulerK   )rL   r   )Úmodule_coder   r   Ú	visitCodeÅ   s    zD_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitCodeN)r"   r#   r$   rN   rP   rQ   rT   r   )rM   rS   rO   r    r   r   ÚFindTopLevel»   s   rU   z# -*- coding:%s -*-zfrom __future__ import %sz, 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)ÚsourcerD   Úposr   z_exports = %rc             S   s   g | ]
}|j ‘qS r   )r@   )r(   Únr   r   r   r*     s    z8_GenerateRenderMethod.write_toplevel.<locals>.<listcomp>é   rI   )%r+   r;   r   ÚnodesÚaccept_visitorrO   ÚsetÚunionÚdeclared_identifiersÚ_IdentifiersÚdeclaredr   r   r   Ú	writeliner   ÚjoinÚMAGIC_NUMBERÚtimer   r   r   r   r   Z
PythonCodeÚbranchÚtopleveldefsÚupdateÚTOPLEVEL_DECLAREDrJ   ÚvaluesÚwrite_blanksr2   Úwrite_module_codeÚwrite_namespacesÚwrite_inheritÚlist)r    rU   ÚfrZ   Zmodule_identZmodule_identifiersr   ÚimpZimpcodeZmain_identifiersZmitr   )rM   rS   rO   r    r   r:   °   s†    










z$_GenerateRenderMethod.write_toplevelc       	      C   sh  | j r |j}|r | j d| ¡ | j |j¡ | j d|d |¡f dd¡ |sX|sX|rd| j d¡ | j 	| j
j | j¡¡ | j rŠ| jjr¤d|kr¤| jd j d	¡ | j sît| jjƒd
ksÊt| jjƒd
krî| j dd dd„ | jjD ƒ¡ ¡ | j| jdd x| jjD ]}| | ¡ qW |  | j|||¡ | j d¡ | j d¡ |rd| j||||| jdd 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)zdef %s(%s):ú,z/__M_caller = context.caller_stack._push_frame()ztry:zcontext._push_buffer()z
**pageargsrI   Zpageargsr   z!__M_locals = __M_dict_builtin(%s)c             S   s   g | ]}d ||f ‘qS )z%s=%sr   )r(   Úxr   r   r   r*   C  s   z?_GenerateRenderMethod.write_render_callable.<locals>.<listcomp>T)ÚtoplevelNr[   )r/   Ú	decoratorr   rc   Ústart_sourcerD   rE   rd   r,   rK   r+   rJ   rg   r   r8   Úargument_declaredÚaddr2   Úlocally_assignedÚwrite_variable_declaresr\   r]   Úwrite_def_finishrl   Úwrite_cache_decorator)	r    r   r@   r4   r%   rA   r'   rv   rZ   r   r   r   r>     sB    
z+_GenerateRenderMethod.write_render_callablec             C   s&   x |D ]}| j j|j|jd qW dS )zewrite module-level template code, i.e. that which
        is enclosed in <%! %> tags in the template.)Ústarting_linenoN)r   Úwrite_indented_blockÚtextrD   )r    rS   rZ   r   r   r   rm   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):z"_mako_generate_namespaces(context)z8return runtime._inherit_from(context, %s, _template_uri)ÚfileN)r   rE   Úparsed_attributes)r    r   r   r   r   ro   \  s    
z#_GenerateRenderMethod.write_inheritc          
      sÖ  ˆj  dddddddd¡ ˆj  d¡ x†| ¡ D ]x}d|jkrJd	ˆj_ˆj  |j¡ t	|j
ƒrêˆj  d
¡ g ‰ ˆjj |¡‰d	ˆ_G ‡ ‡‡fdd„dƒ}|ƒ }x|j
D ]}| |¡ qªW ˆj  dd ˆ ¡ ¡ ˆj  d¡ dˆ_d}nd}d|jkrˆj  d|j|j dd¡|f ¡ nFd|jkrNˆ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¡ q2W t	|ƒsÆˆj  d¡ ˆj  d¡ dS )z5write the module-level namespace-generating callable.z'def _mako_get_namespace(context, name):ztry:z+return context.namespaces[(__name__, name)]zexcept KeyError:z"_mako_generate_namespaces(context)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   s   |   |¡ d S )N)ÚvisitDefOrBase)rL   r   r   r   r   ÚvisitDefTag€  s    zH_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefTagc             S   s   |   |¡ d S )N)r„   )rL   r   r   r   r   ÚvisitBlockTagƒ  s    zJ_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitBlockTagc                s4   |j rtjd|jŽ‚ˆj|ˆdd ˆ  |j¡ d S )Nú.Can't put anonymous blocks inside <%namespace>F)Únested)r‡   )r9   r   ÚCompileExceptionÚexception_kwargsÚwrite_inline_defrK   r0   )rL   r   )ÚexportrJ   r    r   r   r„   †  s    zK_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefOrBaseN)r"   r#   r$   r…   r†   r„   r   )rŒ   rJ   r    r   r   ÚNSDefVisitor  s   r   zreturn [%s]rs   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   rE   rc   rk   r6   r+   Úhas_ns_importsrw   rD   r2   r\   rJ   rg   r/   r]   rd   r‚   r@   r7   r5   Úreprrl   )r    rO   r   r   ÚvisrZ   Zcallable_namer   )rŒ   rJ   r    r   rn   g  sf    


z&_GenerateRenderMethod.write_namespacesFNc       
      C   s^  dd„ |j D ƒ}tƒ }| |j¡}| dd„ |j ¡ D ƒ¡}| |j¡}| |j¡}| j	j
rpd|k}| d¡ nd}|dk	r†| |¡}|ròt| j	ddƒrò| j d	¡ d
| j	_xD| j	j ¡ D ]4\}}d|jkrº| j d|t d|jd ¡f ¡ qºW |r| j d¡ xF|D ]<}||kr~|| }	|	jrT|	jsB|  |	|¡ n| j|	|d
d n(|	 ¡ rl|  |	|¡ n| j|	|d
d nÊ|| j	jkr¢| j d||f ¡ n¦t| j	ddƒr| j	jrð| j d||f d| dd||f dd| dd¡ n| j d|||f ¡ n@| j	jr4| j dd||f dd| d¡ n| j d||f ¡ qW | 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   )r0   )r(   Úcr   r   r   ú
<dictcomp>Ø  s    zA_GenerateRenderMethod.write_variable_declares.<locals>.<dictcomp>c             S   s   g | ]
}|j ‘qS r   )r0   )r(   r–   r   r   r   r*   â  s    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:ztry:z%s = context[%r]zexcept KeyError: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())rB   r^   r_   Ú
undeclaredÚclosuredefsrk   Ú
differencerx   Úlocally_declaredr+   r   ÚdiscardÚintersectionÚgetattrr   rc   Zhas_importsrO   Úitemsr6   ÚreÚsplitr8   r9   Úwrite_def_declr‹   Úis_rootr   rE   )
r    rJ   ru   ÚlimitZcomp_identsZto_writeZhas_loopÚidentÚnsÚcompr   r   r   r{   Ä  s|    







z-_GenerateRenderMethod.write_variable_declaresc             C   sž   |j }| ¡ }|jdd}| jsNt| jjƒdks@t| jjƒdkrN| 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
   zdef %s(%s):rs   zreturn render_%s(%s)N)r0   r1   r/   r2   rJ   rz   rx   Úinsertr   rc   rd   )r    r   rJ   r0   Ú	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	¡ |sŒ|sŒ|r˜| j d
¡ |j||d}|  |¡ | j |¡ x|jD ]}	|	 | ¡ qÄW | j ¡  |  ||||¡ | j d¡ |r| j||j|d|ddd dS )z?write a locally-available def callable inside an enclosing def.z&@runtime._decorate_inline(context, %s)zdef %s(%s):rs   r   r%   r&   r'   z/__M_caller = context.caller_stack._push_frame()ztry:zcontext._push_buffer())rˆ   NFT)Úinlineru   )r1   rv   r   rc   r0   rd   r2   r3   r4   r5   r6   r7   rE   rg   r{   r,   rK   r\   r]   Úpopr|   r}   )
r    r   rJ   rˆ   rª   rv   rA   r%   r'   rZ   r   r   r   r‹   U  s@    


z&_GenerateRenderMethod.write_inline_defTc             C   sÞ   |s,|s,|s,| j  d¡ |r,| j  ddd¡ |s8|s8|rÚ|s@|rP| j  dd¡ n| j  dd¡ |rn| j  d¡ d}|rˆ|  |jj|d¡}| j  d¡ |r®|s®|  | jj|d¡}|s¶|rÈ| j  d	| ¡ n| j  d
| d¡ 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.z	return ''zfinally:z!context.caller_stack._pop_frame()Nz__M_buf = context._pop_buffer()z6__M_buf, __M_writer = context._pop_buffer_and_writer()z__M_buf.getvalue()Fz	return %sz__M_writer(%s))r   rc   rE   Úcreate_filter_callabler3   r4   r+   r   )r    r   r%   rA   r'   Ú	callstackrL   r   r   r   r|   ƒ  s2    

z&_GenerateRenderMethod.write_def_finishc                sZ  ˆj  d||f ¡ ˆ j dt|ƒ¡}i }	ˆjjdk	rT|	 ‡fdd„ˆjjjD ƒ¡ |	 ‡ fdd„ˆ jD ƒ¡ d|	krŠtt	|	d ƒƒ|	d< ˆj  d|d	 
|¡f ¡ d
d„ |D ƒ}
ˆj||ˆ  ¡ d |rd||d	 
|
¡d 
dd„ |	 ¡ D ƒ¡|f }ˆ ˆjj|d¡}ˆj  d| d¡ n8ˆ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:   | ]2}|  d ¡r|dkr|dd… ˆ jjj| fV  qdS )Úcache_r°   é   N)Ú
startswithr+   r;   r‚   )r(   Úpa)r    r   r   ú	<genexpr>Ë  s   z>_GenerateRenderMethod.write_cache_decorator.<locals>.<genexpr>c             3   s6   | ].}|  d ¡r|dkr|dd… ˆ j| fV  qdS )r±   r°   r²   N)r³   r‚   )r(   r´   )Únode_or_pagetagr   r   rµ   Ð  s   Útimeoutzdef %s(%s):rs   c             S   s0   g | ](}d |kr(d|  d ¡d fd  n|‘qS )ú=z%s=%sr   r[   )r¡   )r(   r)   r   r   r   r*   Û  s    z?_GenerateRenderMethod.write_cache_decorator.<locals>.<listcomp>)ru   r¤   z`context.get('local').cache._ctx_get_or_create(%s, lambda:__M_%s(%s),  context, %s__M_defname=%r)rV   c             S   s   g | ]\}}d ||f ‘qS )z%s=%s, r   )r(   ÚkÚvr   r   r   r*   í  s    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   s   g | ]\}}d ||f ‘qS )z%s=%s, r   )r(   r¹   rº   r   r   r   r*     s    z	return '')r   rc   r‚   r7   r”   r+   r;   ri   Úintr5   rd   r{   Úundeclared_identifiersrŸ   r®   r   rE   )r    r¶   r@   r4   r%   rJ   r¬   ru   ZcachekeyZ
cache_argsZ	pass_argsrL   r   )r¶   r    r   r}   ¶  sP    

z+_GenerateRenderMethod.write_cache_decoratorc       
      C   s¶   dd„ }d|krH|rH| j jr,| j jjj| }| j jrHd|krH| j j| }xh|D ]`}|dkr\qNt d|¡}|rŽ| dd¡\}}||ƒ}	|	| }n||ƒ}|dk	s¢t‚d||f }qNW |S )	z§write 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 d S )Nz
decode\..+zfilters.)r    Úmatchr   ZDEFAULT_ESCAPESr7   )r@   r   r   r   Úlocate_encode  s    zC_GenerateRenderMethod.create_filter_callable.<locals>.locate_encoderZ   z(.+?)(\(.*\))r‘   r[   Nz%s(%s))	r+   r;   r3   r4   r   r    r½   ÚgroupÚAssertionError)
r    r4   ÚtargetZis_expressionr¾   ÚeÚmr¥   Zfargsrq   r   r   r   r®   	  s&    

z,_GenerateRenderMethod.create_filter_callablec             C   s€   | j  |j¡ t|jƒs@| jjd k	r4t| jjjjƒs@t| jj	ƒrj|  
|jjd|j d¡}| j  d| ¡ n| j  d|j ¡ d S )Nz%sTz__M_writer(%s))r   rw   rD   r2   Úescapesr+   r;   r3   r4   r   r®   Zescapes_coder€   rc   )r    r   rL   r   r   r   ÚvisitExpression)  s    
z%_GenerateRenderMethod.visitExpressionc                sÂ   ˆ j r>| j d ¡ ˆ jr¾| j d¡ | j d¡ | j d ¡ n€| j ˆ j¡ | jjrlˆ jdkrlt	ˆ | jƒ}nˆ j
}| j |¡ ˆ  ¡ }|r²tdd„ |D ƒƒr¾t‡ fdd„|D ƒƒr¾| j d¡ d S )Nzfinally:zloop = __M_loop._exit()Úforc             s   s    | ]}t |tjtjfƒV  qd S )N)r-   r   ÚCommentÚControlLine)r(   r–   r   r   r   rµ   Q  s   z9_GenerateRenderMethod.visitControlLine.<locals>.<genexpr>c             3   s,   | ]$}t |tjƒrˆ  |j¡p"|jV  qd S )N)r-   r   rÈ   Z
is_ternaryÚkeywordÚisend)r(   r–   )r   r   r   rµ   U  s   r’   )rÊ   r   rc   Úhas_loop_contextrw   rD   r+   r   rÉ   Úmangle_mako_loopr€   Úget_childrenÚall)r    r   r€   Úchildrenr   )r   r   ÚvisitControlLine;  s(    


z&_GenerateRenderMethod.visitControlLinec             C   s(   | j  |j¡ | j  dt|jƒ ¡ d S )Nz__M_writer(%s))r   rw   rD   rc   r”   Úcontent)r    r   r   r   r   Ú	visitText\  s    z_GenerateRenderMethod.visitTextc          
   C   sh   t |jjƒdk}|r"| j dd¡ x|jD ]}| | ¡ q*W |rd| j ddd|  |jjdd¡ d ¡ d S )	Nr   z#__M_writer = context._push_writer()ztry:zfinally:z6__M_buf, __M_writer = context._pop_buffer_and_writer()z__M_writer(%s)z__M_buf.getvalue()F)r2   r3   r4   r   rE   r\   r]   r®   )r    r   rA   rZ   r   r   r   ÚvisitTextTag`  s    z"_GenerateRenderMethod.visitTextTagc             C   sd   |j s`| jj|j|jd | js`t| jjƒdkr`| j 	d¡ | j 	dd 
dd„ | ¡ D ƒ¡ ¡ d S )N)r~   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]))rs   c             S   s   g | ]}t |ƒ‘qS r   )r”   )r(   rt   r   r   r   r*   …  s    z3_GenerateRenderMethod.visitCode.<locals>.<listcomp>)rR   r   r   r€   rD   r/   r2   rJ   rz   rc   rd   r`   )r    r   r   r   r   rT   s  s    z_GenerateRenderMethod.visitCodec             C   sT   | j  |j¡ |j d¡}|r:| j  d|jd |f ¡ n| j  d|jd  ¡ d S )Nr4   z5runtime._include_file(context, %s, _template_uri, %s)r   z1runtime._include_file(context, %s, _template_uri))r   rw   rD   r6   r7   rc   r‚   )r    r   r4   r   r   r   ÚvisitIncludeTagˆ  s    z%_GenerateRenderMethod.visitIncludeTagc             C   s   d S )Nr   )r    r   r   r   r   rP   –  s    z'_GenerateRenderMethod.visitNamespaceTagc             C   s   d S )Nr   )r    r   r   r   r   r…   ™  s    z!_GenerateRenderMethod.visitDefTagc             C   sn   |j r| j d|j ¡ nP|jdd}|dg7 }| j d|j ¡ | j d|jd |¡f ¡ | j d¡ d S )	Nz%s()T)r¨   z
**pageargszOif 'parent' not in context._data or not hasattr(context._data['parent'], '%s'):zcontext['self'].%s(%s)rs   rW   )r9   r   rc   r0   r1   rd   )r    r   r«   r   r   r   r†   œ  s    

z#_GenerateRenderMethod.visitBlockTagc             C   s   |   |¡ d S )N)ÚvisitCallTag)r    r   r   r   r   ÚvisitCallNamespaceTag«  s    z+_GenerateRenderMethod.visitCallNamespaceTagc                sh  ˆj  d¡ dg‰ˆjj|dd‰ˆj|dd‰ ˆ  d¡ ˆj ˆ ¡ G ‡ ‡‡‡fdd„dƒ}|ƒ }x|jD ]}| |¡ qlW ˆj 	¡  |j
 ¡ }ˆj  d	d
 |¡ ¡ d}|r¾ˆj  dd¡ ˆ ˆ ¡ ˆj ˆ ¡ x|jD ]}| ˆ¡ qÜW ˆ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):ÚbodyT)rˆ   FZcallerc                   s.   e Zd Zdd„ Zdd„ Z‡ ‡‡‡fdd„ZdS )z6_GenerateRenderMethod.visitCallTag.<locals>.DefVisitorc             S   s   |   |¡ d S )N)r„   )rL   r   r   r   r   r…   ¿  s    zB_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitDefTagc             S   s   |   |¡ d S )N)r„   )rL   r   r   r   r   r†   Â  s    zD_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitBlockTagc                s<   ˆj |ˆdd |js"ˆ |j¡ |jˆ jkr8ˆ j|j= d S )NF)rˆ   )r‹   r9   rK   r0   r™   )rL   r   )Úbody_identifiersÚcallable_identifiersrŒ   r    r   r   r„   Å  s
    zE_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitDefOrBaseN)r"   r#   r$   r…   r†   r„   r   )rØ   rÙ   rŒ   r    r   r   Ú
DefVisitor¾  s   rÚ   zdef body(%s):rs   zcontext._push_buffer()ztry:)r¯   zreturn [%s]zccontext.caller_stack.nextcaller = runtime.Namespace('caller', context, callables=ccall(__M_caller))z__M_writer(%s)zfinally:z&context.caller_stack.nextcaller = None)r   rc   rJ   rg   Úadd_declaredr,   rK   r\   r]   r­   r<   r1   rd   rE   r{   r|   rw   rD   r®   Z
expression)r    r   rÚ   r•   rZ   Zbodyargsr%   r   )rØ   rÙ   rŒ   r    r   rÕ   ±  sB    




z"_GenerateRenderMethod.visitCallTag)FN)T)FF)r"   r#   r$   Ú__doc__r!   r?   ÚpropertyrJ   r:   r>   rm   ro   rn   r{   r¢   r‹   r|   r}   r®   rÅ   rÐ   rÒ   rÓ   rT   rÔ   rP   r…   r†   rÖ   rÕ   r   r   r   r   r   c   s6   1m9]
~/
9 
K !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 ))ra   zCtracks the status of identifier names as template code is rendered.NFc             C   s  |d k	r€t |tjƒr(tƒ | _t ¡ | _q’t|jƒ dd„ |j	 
¡ D ƒ¡ |j¡ |j¡| _|rn| j |j¡| _tjf |jŽ| _ntƒ | _t ¡ | _|| _tƒ | _tƒ | _tƒ | _tƒ | _t ¡ | _	|| _|d k	rÚ| | ¡ | jj | j¡}|rt dd |¡ ¡‚d S )Nc             S   s   g | ]
}|j ‘qS r   )r@   )r(   r–   r   r   r   r*     s    z)_Identifiers.__init__.<locals>.<listcomp>z'Reserved words declared in template: %sz, )r-   r   ZNamespaceTagr^   rb   r   ZSetLikeDictrh   r_   r™   rk   r›   rx   r˜   r+   rz   r   r]   r   r   r   ZNameConflictErrorrd   )r    r+   r   Úparentrˆ   Zillegal_namesr   r   r   r!   ú  s8    




z_Identifiers.__init__c             K   s   t | j|| f|ŽS )zUcreate a new Identifiers for a new Node, with
        this Identifiers as the parent.)ra   r+   )r    r   r=   r   r   r   rg   =  s    z_Identifiers.branchc             C   s   t | j | j¡ ¡ ƒS )N)r^   rh   r_   r™   rk   )r    r   r   r   rB   C  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   s   g | ]
}|j ‘qS r   )r@   )r(   r–   r   r   r   r*   P  s    z)_Identifiers.__repr__.<locals>.<listcomp>c             S   s   g | ]
}|j ‘qS r   )r@   )r(   r–   r   r   r   r*   Q  s    )rp   rb   r›   r˜   rh   rk   r™   rx   )r    r   r   r   Ú__repr__G  s    z_Identifiers.__repr__c             C   sZ   x6|  ¡ D ]*}|dkr
|| j | j¡kr
| j |¡ q
W x| ¡ D ]}| j |¡ qBW dS )zlupdate the state of this Identifiers with the undeclared
        and declared identifiers of the given node.r
   N)r¼   rb   r_   r›   r˜   ry   r`   )r    r   r¥   r   r   r   Úcheck_declaredV  s    
z_Identifiers.check_declaredc             C   s&   | j  |¡ || jkr"| j |¡ d S )N)rb   ry   r˜   Úremove)r    r¥   r   r   r   rÛ   b  s    
z_Identifiers.add_declaredc             C   s   |   |¡ d S )N)rà   )r    r   r   r   r   rÅ   g  s    z_Identifiers.visitExpressionc             C   s   |   |¡ d S )N)rà   )r    r   r   r   r   rÐ   j  s    z_Identifiers.visitControlLinec             C   s&   |j s"|  |¡ | j | ¡ ¡| _d S )N)rR   rà   rz   r_   r`   )r    r   r   r   r   rT   m  s    
z_Identifiers.visitCodec             C   s(   | j |kr$x|jD ]}| | ¡ qW d S )N)r   r\   r]   )r    r   rZ   r   r   r   rP   t  s    
z_Identifiers.visitNamespaceTagc             C   sL   |  |j¡}|||j< |d k	rH||k	rH|js2|jrHtjd|j f|jŽ‚d S )Nz<%%def or %%block named '%s' already exists in this template.)r7   r0   r8   r   r‰   rŠ   )r    Z
collectionr   Úexistingr   r   r   Ú_check_name_exists|  s    
z_Identifiers._check_name_existsc             C   s´   |  ¡ r|js|  | j|¡ n|| jk	r6|  | j|¡ x6| ¡ D ]*}|dkr@|| j | j	¡kr@| j
 |¡ q@W || jkr°x| ¡ D ]}| j |¡ q‚W x|jD ]}| | ¡ qžW d S )Nr
   )r£   r9   rã   rh   r   r™   r¼   rb   r_   r›   r˜   ry   r`   rx   r\   r]   )r    r   r¥   rZ   r   r   r   r…   Š  s    


z_Identifiers.visitDefTagc             C   s  || j k	rj|jsjt| j tjƒr>tjd|j| j jf f|jŽ‚n,t| j tj	tj
fƒrjtjd|jf f|jŽ‚x6| ¡ D ]*}|dkrt|| j | j¡krt| j |¡ qtW |jsÆ|  | j|¡ | j |j¡ n|| j k	rÞ|  | j|¡ x| ¡ D ]}| j |¡ qèW x|jD ]}| | ¡ qW d S )Nz/Named block '%s' not allowed inside of def '%s'z3Named block '%s' not allowed inside of <%%call> tagr
   )r   r9   r-   r   r.   r   r‰   r@   rŠ   ZCallTagZCallNamespaceTagr¼   rb   r_   r›   r˜   ry   rã   rh   r0   r™   r`   rx   r\   r]   )r    r   r¥   rZ   r   r   r   r†   ž  s2    



z_Identifiers.visitBlockTagc             C   s<   x6|  ¡ D ]*}|dkr
|| j | j¡kr
| j |¡ q
W d S )Nr
   )r¼   rb   r_   r›   r˜   ry   )r    r   r¥   r   r   r   rÓ   À  s    
z_Identifiers.visitTextTagc             C   s   |   |¡ d S )N)rà   )r    r   r   r   r   rÔ   Ç  s    z_Identifiers.visitIncludeTagc             C   s,   x|  ¡ D ]}| j |¡ q
W |  |¡ d S )N)r`   rx   ry   rà   )r    r   r¥   r   r   r   rQ   Ê  s    z_Identifiers.visitPageTagc             C   s   |   |¡ d S )N)rÕ   )r    r   r   r   r   rÖ   Ï  s    z"_Identifiers.visitCallNamespaceTagc             C   s¸   || j kr|x6| ¡ D ]*}|dkr|| j | j¡kr| j |¡ qW x| ¡ D ]}| j |¡ qLW xR|j	D ]}| 
| ¡ qhW n8x6| ¡ D ]*}|dkr†|| j | j¡kr†| j |¡ q†W d S )Nr
   )r   r¼   rb   r_   r›   r˜   ry   r`   rx   r\   r]   )r    r   r¥   rZ   r   r   r   rÕ   Ò  s    


z_Identifiers.visitCallTag)NNF)r"   r#   r$   rÜ   r!   rg   rÝ   rB   rß   rà   rÛ   rÅ   rÐ   rT   rP   rã   r…   r†   rÓ   rÔ   rQ   rÖ   rÕ   r   r   r   r   ra   ö  s&   
C"ra   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rhd| jd _t | j¡}|rX| d| 	d¡ d¡ d| 	d¡ }qnt
d| j ƒ‚n| j}|S )	z—converts 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
    TrI   zloop = __M_loop._enter(%s)r[   ztry:zfor %s in loop:r‘   zCouldn't apply loop context: %s)ÚLoopVariabler]   Údetectedr\   rË   Ú	_FOR_LOOPr½   r€   rE   r¿   ÚSyntaxError)r   r   Z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å   )r    r   r   r   r!     s    zLoopVariable.__init__c             C   s4   d|  ¡ krd| _nx| ¡ D ]}| | ¡ qW d S )Nr   T)r¼   rå   rÍ   r]   )r    r   rZ   r   r   r   Ú_loop_reference_detected  s    z%LoopVariable._loop_reference_detectedc             C   s   |   |¡ d S )N)rè   )r    r   r   r   r   rÐ     s    zLoopVariable.visitControlLinec             C   s   |   |¡ d S )N)rè   )r    r   r   r   r   rT     s    zLoopVariable.visitCodec             C   s   |   |¡ d S )N)rè   )r    r   r   r   r   rÅ     s    zLoopVariable.visitExpressionN)	r"   r#   r$   rÜ   r!   rè   rÐ   rT   rÅ   r   r   r   r   rä     s   rä   )rÜ   rF   r    rf   Úmakor   r   r   r   r   Z
mako.pygenr   re   rj   r_   ZRESERVED_NAMESÚ	frozensetr   r   r   ra   ræ   rÌ   rä   r   r   r   r   Ú<module>   sJ           p
