관리-도구
편집 파일: codegen.cpython-37.pyc
B �C]� � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd l m Z d Z ed d g�Zed dg��e�Zdddddddddde� f dd�ZG dd� de�ZG dd� de�ZG dd� de�Ze�d�Zdd� ZG d d � d e�ZdS ) zVprovides functionality for rendering a parsetree constructing into module source code.� N)�ast)�compat)� exceptions)�filters)� parsetree)�util)� PythonPrinter� Z UNDEFINEDZSTOP_RENDERING�context�loopTFc C sZ t js t|t j�r |�|�}t�� } t| �}t|t ||||||||| | | | � | � | � � S )z[Generate module source code given a parsetree node, uri, and optional source filename) r Zpy3k� isinstanceZ text_type�encoder ZFastEncodingBufferr �_GenerateRenderMethod�_CompileContext�getvalue)�node�uri�filename�default_filters�buffer_filters�imports�future_imports�source_encoding�generate_magic_comment�disable_unicode�strict_undefined� enable_loop�reserved_names�buf�printer� r �=/opt/alt/python37/lib/python3.7/site-packages/mako/codegen.py�compile s* r" c @ s e Zd Zdd� ZdS )r c C sL || _ || _|| _|| _|| _|| _|| _|| _| | _| | _ | | _ | | _ d S )N) r 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 r! �__init__R s z_CompileContext.__init__N)�__name__� __module__� __qualname__r$ r r r r! r Q s r c @ s� e Zd ZdZdd� Zdd� Zedd� �Zdd � Zd d � Z d d � Z dd� Z dd� Z d7dd�Z dd� Zdd� Zd8dd �Zd9d d �Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z dS ):r z]A template visitor object which generates the full module source for a template. c C s� || _ || _|| _d g| _t|tjtjf�| _| jr�d|j }|� � }t |j j �dk}t|j�dd��}t|j�dd��}d } d } |jr�|js�|dg7 }n~| �� } | jj} d}| d k �r| j� � }| jjs�|dg7 }t| j�dd��}| jjp�t| j�dd��| j_n dg}d }d }}|d k�r*d g}nd 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 | ]}|�qS r r )�.0�ar r r! � <listcomp>� s z2_GenerateRenderMethod.__init__.<locals>.<listcomp>)r �compilerr �identifier_stackr r �DefTagZBlockTag�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 �namer6 �filteredr( r* �defsr= r r r! r$ v sL z _GenerateRenderMethod.__init__c C sR t | jj�| jj| jj< | jj| jj| jj| jjd�}| j�ddt � |�d� d S )N)r r r Zline_mapz"""Z__M_BEGIN_METADATAz__M_END_METADATA """) �maxr Z source_map�linenor. r r r � writelines�json�dumps)r# �structr r r! rA � 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 � ���fdd�dt�}|� }x�jjD ]}|�|� q>W ��j _t� }x�D ]}|�|� � �}qdW t �j �}||_ �j j r��j j r��j�d�j j � �j jr҈j�dd��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 jD ] }||d 7 }�j�|� �q�W tj|ddddd�}nd}|��j�} |j} | �| j�|_|j � t � |�r|j � |j � |�j _ �j�ddd� | j� � D � � �j� d� t!���r^��"�� t!� ��r���#�� ��$� d � nt!���r���#�� t%| j� � �S ) zxTraverse a template structure for module-level directives and generate the start of module-level code. Nc s< e Zd Z� fdd�Z�fdd�Z�fdd�Z�fdd�Zd 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)rB )rN r )� namespacesr r! �visitNamespaceTag� s zL_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitNamespaceTagc s |� j _d S )N)r. r= )rN r )r# r r! � visitPageTag� s zG_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitPageTagc s |j r� �|� d S )N)�ismodulerM )rN r )� module_coder r! � visitCode� s zD_GenerateRenderMethod.write_toplevel.<locals>.FindTopLevel.visitCodeN)r% r& r' rP rR rS rV r )rO rU rQ r# r r! � FindTopLevel� s rW z# -*- coding:%s -*-zfrom __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 ztemplate defined imports)�sourcerF �posr z _exports = %rc S s g | ] }|j �qS r )rB )r+ �nr r r! r- s z8_GenerateRenderMethod.write_toplevel.<locals>.<listcomp>� rK )&r. r= �objectr �nodes�accept_visitorrQ �set�union�declared_identifiers� _Identifiers�declaredr r r � writeliner �join� MAGIC_NUMBER�timer r r r r Z PythonCode�branch� topleveldefs�update�TOPLEVEL_DECLAREDrL �values� write_blanksr4 �write_module_code�write_namespaces� write_inherit�list) r# rW �fr\ Z module_identZmodule_identifiersr �impZimpcodeZmain_identifiersZmitr )rO rU rQ r# r! r< � s� z$_GenerateRenderMethod.write_toplevelc C sh | j r |j}|r | j�d| � | j�|j� | j�d|d�|�f dd� |sX|sX|rd| j�d� | j� | j j � | j �� | j r�| j jr�d|kr�| jd j�d � | j s�t| j j�d ks�t| j j�d kr�| j�d d�d d � | j jD �� � | j| j dd� x | j jD ]}|�| � �qW | �| j |||� | j�d� | j�d� |�rd| j||||| j dd� dS )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()ztry:zcontext._push_buffer()z **pageargsrK Zpageargsr z!__M_locals = __M_dict_builtin(%s)c S s g | ]}d ||f �qS )z%s=%sr )r+ �xr r r! r- O s z?_GenerateRenderMethod.write_render_callable.<locals>.<listcomp>T)�toplevelNr] )r1 � decoratorr rf � start_sourcerF rG rg r/ rM r. rL rj r r: �argument_declared�addr4 �locally_assigned�write_variable_declaresr_ r` �write_def_finishro �write_cache_decorator) r# r rB r6 r( rC r* ry r\ r r r! r@ ) sB z+_GenerateRenderMethod.write_render_callablec C s& x |D ]}| j j|j|jd� qW dS )zewrite module-level template code, i.e. that which is enclosed in <%! %> tags in the template.)�starting_linenoN)r �write_indented_block�textrF )r# rU r\ r r r! rp b s z'_GenerateRenderMethod.write_module_codec C s | j �ddd|jd d� dS )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 rG �parsed_attributes)r# r r r r! rr h s z#_GenerateRenderMethod.write_inheritc s� �j �dddddddd� �j �d� �x�|�� D �]z}d|jkrJd �j_�j �|j� t |j �r�j �d � g � �jj � |��d �_ G � ��fd d �d t�}|� }x|j D ]}|�|� q�W �j �d d�� � � �j �d� d�_ d}nd}d|jk�r �j �d|j|j�dd�|f � nFd|jk�rP�j �d|j||j�dd�f � n�j �d|j|f � t|j�dd���r��j �d|j � �j �dt|j� � �j �d� q2W t |��sȈj �d � �j �d� dS ) z5write the module-level namespace-generating callable.z'def _mako_get_namespace(context, name):ztry:z+return context.namespaces[(__name__, name)]zexcept KeyError:z"_mako_generate_namespaces(context)Nz'def _mako_generate_namespaces(context):�importTzdef make_namespace():c s, e Zd Zdd� Zdd� Z� ��fdd�ZdS )z<_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitorc S s | � |� d S )N)�visitDefOrBase)rN r r r r! � visitDefTag� s zH_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefTagc S s | � |� d S )N)r� )rN r r r r! � visitBlockTag� s zJ_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitBlockTagc s4 |j rtjd|j���j|�dd� � �|j� d S )N�.Can't put anonymous blocks inside <%namespace>F)�nested)r� )r; r �CompileException�exception_kwargs�write_inline_defrM r2 )rN r )�exportrL r# r r! r� � s zK_GenerateRenderMethod.write_namespaces.<locals>.NSDefVisitor.visitDefOrBaseN)r% r& r' r� r� r� r )r� rL r# r r! � NSDefVisitor� s r� z return [%s]rv Fzmake_namespace()�Noner� z�ns = 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) zcontext['self'].%s = nsz'context.namespaces[(__name__, %s)] = ns� �pass)r rG rf rn r8 r. �has_ns_importsrz rF r4 r_ rL rj r1 r^ r` rg r� rB r9 r7 �reprro )r# rQ r r� �visr\ Z callable_namer )r� rL r# r! rq s sf z&_GenerateRenderMethod.write_namespacesFNc C sb t dd� |jD ��}t� }|�|j�}|�dd� |j�� D ��}|�|j�}|�|j �}| j j rtd|k}|� d� nd}|dk r�|� |�}|r�t| j dd�r�| j�d� d | j _xD| j j�� D ]4\}}d |jkr�| j�d |t�d |jd �f � q�W |�r| j�d � �xF|D �]<}||k�r�|| } | j�rX| j�sF| �| |� n| j| |d d� n(| �� �rp| �| |� n| j| |d d� n�|| j jk�r�| j�d||f � n�t| j dd��r | j j �r�| j� d||f d| dd||f dd| dd� n| j�d|||f � n@| j j �r8| j� dd||f dd| d� n| j�d||f � �qW | j�d� dS )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 g | ]}|j |f�qS r )r2 )r+ �cr r r! r- � s zA_GenerateRenderMethod.write_variable_declares.<locals>.<listcomp>c S s g | ] }|j �qS r )r2 )r+ r� r r r! r- � s 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)zif %s is UNDEFINED:ztry:z%s = context[%r]zexcept 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()) �dictrD ra rb � undeclared� closuredefsrn � differencer{ �locally_declaredr. r �discard� intersection�getattrr rf Z has_importsrQ �itemsr8 �re�splitr: r; �write_def_declr� �is_rootr rG ) r# rL rx �limitZ comp_identsZto_writeZhas_loop�ident�ns�compr r r! r~ � s| z-_GenerateRenderMethod.write_variable_declaresc C s� |j }|�� }|jdd�}| jsNt| jj�dks@t| jj�dkrN|�dd� n |�dd� | 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 zcontext._locals(__M_locals)r z def %s(%s):rv zreturn render_%s(%s)N) r2 r3 r1 r4 rL r} r{ �insertr rf rg )r# r rL r2 � namedecls�nameargsr r r! r� N s z$_GenerateRenderMethod.write_def_declc C s |� � }|j}|r"| j�d| � | j�d|jd�|�f � t|jj�dk}t |j � dd��}t |j � dd��}| j� dd � |s�|s�|r�| j� d � |j ||d �}| �|� | j�|� x|jD ]} | �| � q�W | j�� | �||||� | j�d � |�r | j||j|d |dd d� d S )z?write a locally-available def callable inside an enclosing def.z&@runtime._decorate_inline(context, %s)z def %s(%s):rv r r( r) r* z/__M_caller = context.caller_stack._push_frame()ztry:zcontext._push_buffer())r� NFT)�inlinerx )r3 ry r rf r2 rg r4 r5 r6 r7 r8 r9 rG rj r~ r/ rM r_ r` �popr r� ) r# r rL r� r� ry rC r( r* r\ r r r! r� a s@ z&_GenerateRenderMethod.write_inline_defTc C s� |s,|s,|s,| j �d� |r,| j �ddd� |s8|s8|r�|s@|rP| j �dd� n| j �dd� |rn| j �d� d}|r�| �|jj|d�}| j �d� |r�|s�| �| jj|d�}|s�|r�| j �d | � n| j �d | d� dS ) 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 ''zfinally: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 rf rG �create_filter_callabler5 r6 r. r )r# r r( rC r* � callstackrN r r r! r � s2 z&_GenerateRenderMethod.write_def_finishc sZ �j �d||f � � j�dt|��}i } �jjdk rT| ��fdd��jjjD �� | �� fdd�� jD �� d| kr�tt | d ��| d< �j �d|d � |�f � d d � |D �} �j ||� � � d � |�r d ||d � | �d� dd � | � � D ��|f } ���jj| d�} �j �d| d� n8�j �d||d � | �d� dd � | � � D ��|f dd� dS )zlwrite 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|dkr|dd� � jjj| fV qdS )�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|dkr|dd� � j| fV qdS )r� r� r� N)r� r� )r+ r� )�node_or_pagetagr r! r� � s �timeoutz def %s(%s):rv c S s0 g | ](}d |kr(d|� d �d fd n|�qS )�=z%s=%sr r] )r� )r+ r, r r r! r- � s z?_GenerateRenderMethod.write_cache_decorator.<locals>.<listcomp>)rx r� z`context.get('local').cache._ctx_get_or_create(%s, lambda:__M_%s(%s), context, %s__M_defname=%r)rX c S s g | ]\}}d ||f �qS )z%s=%s, r )r+ �k�vr r r! r- � s Fzreturn 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 �qS )z%s=%s, r )r+ r� r� r r r! r- s z return '')r rf r� r9 r� r. r= rl �intr7 rg r~ �undeclared_identifiersr� r� r rG ) r# r� rB r6 r( rL r� rx ZcachekeyZ cache_argsZ pass_argsrN r )r� r# r! r� � sP z+_GenerateRenderMethod.write_cache_decoratorc s� � fdd�}d|krL|rL� j jr0� j jjj| }� j jrLd|krL� j j| }xh|D ]`}|dkr`qRt�d|�}|r�|�dd�\}}||�} | | }n||�}|dk s�t�d||f }qRW |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< t �d| �rd| S � jjr*tj�| | �S tj�| | �S d S )Nz decode\..+zfilters.)r� �matchr. r r ZNON_UNICODE_ESCAPESr9 ZDEFAULT_ESCAPES)rB )r# r r! � locate_encode s zC_GenerateRenderMethod.create_filter_callable.<locals>.locate_encoder\ z (.+?)(\(.*\))r� r] Nz%s(%s)) r. r= r5 r6 r r� r� �group�AssertionError) r# r6 �targetZ is_expressionr� �e�mr� Zfargsrt r )r# r! r� s& z,_GenerateRenderMethod.create_filter_callablec C s� | j �|j� t|j�s@| jjd k r4t| jjjj�s@t| jj �rj| � |j jd|j d�}| j � d| � n| j � d|j � d S )Nz%sTz__M_writer(%s))r rz rF r4 Zescapesr. r= r5 r6 r r� Z escapes_coder� rf )r# r rN r r r! �visitExpression7 s z%_GenerateRenderMethod.visitExpressionc s� � j r>| j�d � � jr�| j�d� | j�d� | j�d � n�| j�� j� | jjrl� jdkrlt � | j�}n� j }| j�|� � � � }|r�t � dd� |D ��r�t � � fdd�|D ��r�| j�d� d S )Nzfinally:zloop = __M_loop._exit()�forc s s | ]}t |tjtjf�V qd S )N)r r �Comment� ControlLine)r+ r� r r r! r� _ s z9_GenerateRenderMethod.visitControlLine.<locals>.<genexpr>c 3 s, | ]$}t |tj�r� �|j�p"|jV qd S )N)r r r� Z is_ternary�keyword�isend)r+ r� )r r r! r� c s r� )r� r rf �has_loop_contextrz rF r. r r� �mangle_mako_loopr� � get_childrenr �all)r# r r� �childrenr )r r! �visitControlLineI s( z&_GenerateRenderMethod.visitControlLinec C s( | j �|j� | j �dt|j� � d S )Nz__M_writer(%s))r rz rF rf r� �content)r# r r r r! � visitTextj s z _GenerateRenderMethod.visitTextc C sh t |jj�dk}|r"| j�dd� x|jD ]}|�| � q*W |rd| j�ddd| �|jjdd� d � d S ) Nr z#__M_writer = context._push_writer()ztry:zfinally:z6__M_buf, __M_writer = context._pop_buffer_and_writer()z__M_writer(%s)z__M_buf.getvalue()F)r4 r5 r6 r rG r_ r` r� )r# r rC r\ r r r! � visitTextTagn s z"_GenerateRenderMethod.visitTextTagc C sd |j s`| jj|j|jd� | js`t| jj�dkr`| j� d� | j� dd� dd� |� � 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]))rv c S s g | ] }t |��qS r )r� )r+ rw r r r! r- � s z3_GenerateRenderMethod.visitCode.<locals>.<listcomp>) rT r r� r� rF r1 r4 rL r} rf rg rc )r# r r r r! rV � s z _GenerateRenderMethod.visitCodec C sT | j �|j� |j�d�}|r:| j �d|jd |f � n| j �d|jd � d S )Nr6 z5runtime._include_file(context, %s, _template_uri, %s)r� z1runtime._include_file(context, %s, _template_uri))r rz rF r8 r9 rf r� )r# r r6 r r r! �visitIncludeTag� s z%_GenerateRenderMethod.visitIncludeTagc C s d S )Nr )r# r r r r! rR � 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|jdd�}|dg7 }| j�d|j � | j�d|jd�|�f � | j�d� d S ) Nz%s()T)r� z **pageargszOif 'parent' not in context._data or not hasattr(context._data['parent'], '%s'):zcontext['self'].%s(%s)rv rY )r; r rf r2 r3 rg )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 sj �j �d� dg��jj|dd���j|dd�� � �d� �j�� � G � ���fdd�dt�}|� }x|jD ]}|� |� qnW �j� � |j � � }�j �d d � |� � d}|r��j �d d � ��� � �j�� � x|jD ]}|� �� q�W �j� � �j||dddd � �j �d dd � �� d � �j �dd � �j �|j� �j �d��g |jd� ddd � d S )Nzdef ccall(caller):�bodyT)r� F�callerc s. e Zd Zdd� Zdd� Z� ���fdd�ZdS )z6_GenerateRenderMethod.visitCallTag.<locals>.DefVisitorc S s | � |� d S )N)r� )rN r r r r! r� � s zB_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitDefTagc S s | � |� d S )N)r� )rN r r r r! r� � s zD_GenerateRenderMethod.visitCallTag.<locals>.DefVisitor.visitBlockTagc s< �j |�dd� |js"��|j� |j� jkr8� j|j= d S )NF)r� )r� r; rM r2 r� )rN 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):rv zcontext._push_buffer()ztry:)r� z return [%s]zccontext.caller_stack.nextcaller = runtime.Namespace('caller', context, callables=ccall(__M_caller))z__M_writer(%s)zfinally:z&context.caller_stack.nextcaller = None)r rf rL rj � add_declaredr/ rM r^ r_ r` r� r> r3 rg rG r~ r rz rF r� Z expression)r# r r� r� r\ Zbodyargsr( r )r� r� r� r# r! r� � sB z"_GenerateRenderMethod.visitCallTag)FN)T)FF) r% r& r' �__doc__r$ rA �propertyrL r< r@ rp rr rq r~ r� r� r r� r� r� r� r� r� rV r� rR r� r� r� r� r r r r! r o s6 1 m9 ] ~/ 9 K"!r c @ s� e Zd ZdZd(dd�Zdd� Zedd � �Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� ZdS ))rd zCtracks the status of identifier names as template code is rendered.NFc C s |d k r�t |tj�r(t� | _t�� | _q�t|j��dd� |j � � D ���|j ��|j �| _|rn| j�|j �| _tjf |j�| _nt� | _t�� | _|| _t� | _ t� | _ t� | _t� | _ t�� | _ || _|d k r�|�| � | jj�| j �}|�rt�dd�|� ��d S )Nc S s g | ] }|j �qS r )rB )r+ r� r r r! r- s z)_Identifiers.__init__.<locals>.<listcomp>z'Reserved words declared in template: %sz, )r r Z NamespaceTagra re r Z SetLikeDictrk rb r� rn r� r{ r� r. r} r r` r r� r ZNameConflictErrorrg )r# r. r �parentr� Z illegal_namesr r r! r$ s8 z_Identifiers.__init__c K s t | j|| f|�S )zWcreate a new Identifiers for a new Node, with this Identifiers as the parent.)rd r. )r# r r? r r r! rj K s z_Identifiers.branchc C s t | j�| j��� �S )N)ra rk rb r� rn )r# r r r! rD Q s z_Identifiers.defsc C sH dt | j�t | j�t | j�dd� | j�� D �dd� | j�� D �| jf S )NzrIdentifiers(declared=%r, locally_declared=%r, undeclared=%r, topleveldefs=%r, closuredefs=%r, argumentdeclared=%r)c S s g | ] }|j �qS r )rB )r+ r� r r r! r- ^ s z)_Identifiers.__repr__.<locals>.<listcomp>c S s g | ] }|j �qS r )rB )r+ r� r r r! r- _ s )rs re r� r� rk rn r� r{ )r# r r r! �__repr__U s z_Identifiers.__repr__c C sZ x6|� � D ]*}|dkr || j�| j�kr | j�|� q W x |�� D ]}| j�|� qBW dS )zpupdate the state of this Identifiers with the undeclared and declared identifiers of the given node.r N)r� re rb r� r� r| rc )r# r r� r r r! �check_declaredd s z_Identifiers.check_declaredc C s& | j �|� || jkr"| j�|� d S )N)re r| r� �remove)r# r� r r r! r� p s z_Identifiers.add_declaredc C s | � |� d S )N)r� )r# r r r r! r� u s z _Identifiers.visitExpressionc C s | � |� d S )N)r� )r# r r r r! r� x s z _Identifiers.visitControlLinec C s& |j s"| �|� | j�|�� �| _d S )N)rT r� r} rb rc )r# r r r r! rV { s z_Identifiers.visitCodec C s( | j |kr$x|jD ]}|�| � qW d S )N)r r_ r` )r# r r\ r r r! rR � s z _Identifiers.visitNamespaceTagc C sL |� |j�}|||j< |d k rH||k rH|js2|jrHtjd|j f|j��d S )Nz<%%def or %%block named '%s' already exists in this template.)r9 r2 r: r r� r� )r# Z collectionr �existingr r r! �_check_name_exists� s z _Identifiers._check_name_existsc C s� |� � r |js | �| j|� n|| jk r6| �| j|� x6|�� D ]*}|dkr@|| j�| j �kr@| j � |� q@W || jkr�x |� � D ]}| j � |� q�W x|jD ]}|�| � q�W d S )Nr )r� r; r� rk r r� r� re rb r� r� r| rc r{ r_ r` )r# r r� r\ r r r! r� � s z_Identifiers.visitDefTagc C s || j k rj|jsjt| j tj�r>tjd|j| j jf f|j��n,t| j tj tj f�rjtjd|jf f|j��x6|� � D ]*}|dkrt|| j � | j�krt| j�|� qtW |js�| �| j|� | j�|j� n|| j k r�| �| j|� x |�� D ]}| j�|� q�W x|jD ]}|�| � �qW d S )Nz/Named block '%s' not allowed inside of def '%s'z3Named block '%s' not allowed inside of <%%call> tagr )r r; r r r0 r r� rB r� ZCallTagZCallNamespaceTagr� re rb r� r� r| r� rk r2 r� rc r{ r_ r` )r# r r� r\ r r r! r� � s2 z_Identifiers.visitBlockTagc C s< x6|� � D ]*}|dkr || j�| j�kr | j�|� q W d S )Nr )r� re rb r� r� r| )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)rc r{ r| r� )r# r r� r r r! rS � s z_Identifiers.visitPageTagc C s | � |� d S )N)r� )r# r r r r! r� � s z"_Identifiers.visitCallNamespaceTagc C s� || j kr|x6|�� D ]*}|dkr|| j�| j�kr| j�|� qW x |�� D ]}| j�|� qLW xR|j D ]}|� | � qhW n8x6|�� D ]*}|dkr�|| j�| j�kr�| j�|� q�W d S )Nr ) r r� re rb r� r� r| rc r{ r_ r` )r# r r� r\ r r r! r� � s z_Identifiers.visitCallTag)NNF)r% r& r' r� r$ rj r� rD r� r� r� r� r� rV rR r� r� r� r� r� rS r� r� r r r r! rd s& C "rd zi^for\s+((?:\(?)\s*[A-Za-z_][A-Za-z_0-9]*(?:\s*,\s*(?:[A-Za-z_][A-Za-z0-9_]*),??)*\s*(?:\)?))\s+in\s+(.*):c C sr t � }| �|� |jrhd| jd _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 TrK zloop = __M_loop._enter(%s)r] ztry:zfor %s in loop:r� z Couldn't apply loop context: %s) � LoopVariabler` �detectedr_ r� � _FOR_LOOPr� r� rG r� � SyntaxError)r r Z loop_variabler� r� r r r! r� � s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd 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 zLoopVariable.__init__c C s4 d|� � krd| _n x|�� D ]}|�| � q W d S )Nr T)r� r� r� r` )r# r r\ 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! rV " s zLoopVariable.visitCodec C s | � |� d S )N)r� )r# r r r r! r� % s z LoopVariable.visitExpressionN) r% r& r' r� r$ r� r� rV r� r r r r! r� s r� )r� rH r� ri �makor r r r r r Z mako.pygenr rh ra rm rb ZRESERVED_NAMES� frozensetr" r^ r r rd r� r� r� r r r r! �<module> sN $ p