관리-도구
편집 파일: compiler.cpython-37.pyc
B ף�a � @ sr d Z ddlZddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddlmZ dd lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl!m"Z" ej#�r ddl$Z%d dl&m'Z' ej(dej)dej*f d�Z+dd d d d d d!d"d#�Z,e+e+d$�d%d&�Z-e.ej)d'ej/d(gdf d)�d*d+�Z0e.ej)d'ej1d(gdf d)�d,d-�Z2dFej3d0ej4e. ej4e. ej4ej5 e6e6ej4e. d1�d2d3�Z7ej*e6d4�d5d6�Z8ej9ej: ej9e. ej;e. d7�d8d9�Z<G d:d;� d;�Z=G d<d(� d(�Z>G d=d>� d>e?�Z@G d?d@� d@e"�ZAG dAdB� dBe"�ZBG dCdD� dDeC�ZDG dEd'� d'e"�ZEdS )Gz0Compiles nodes from the parser into Python code.� N)�contextmanager)�update_wrapper)�StringIO)�chain)� iskeyword)�escape)�Markup� )�nodes)�TemplateAssertionError)�Symbols)�VAR_LOAD_ALIAS)�VAR_LOAD_PARAMETER)�VAR_LOAD_RESOLVE)�VAR_LOAD_UNDEFINED)� EvalContext)� Optimizer)�_PassArg)�concat)� NodeVisitor)� Environment�F.)�boundz==z!=�>z>=�<z<=�inznot in)�eq�ne�gtZgteq�ltZlteqr Znotin)�f�returnc s2 dt jdtjtjd�� fdd� }tt�t|�� �S )N� CodeGenerator�Frame)�self�node�frame�kwargsr! c sF | j d k r6|jjs6| j �||j�}||kr6| �||�S � | ||f|�S )N)� optimizer�eval_ctx�volatile�visit)r$ r% r& r' �new_node)r � �@/opt/alt/python37/lib/python3.7/site-packages/jinja2/compiler.py�new_func, s z optimizeconst.<locals>.new_func)r �Expr�t�Anyr �castr )r r/ r- )r r. � optimizeconst+ s r4 r" r# )�opr! c s" t dtjtd d�� fdd� �}|S )Nr" )r$ r% r& r! c s� | j jrN� | j jkrN| �d� �d�� | �|j|� | �d� | �|j|� n8| �d� | �|j|� | �d� � d�� | �|j|� | �d� d S )Nz environment.call_binop(context, z, �(� �))� environment� sandboxedZintercepted_binops�writer+ �left�right)r$ r% r& )r5 r- r. �visitor< s z _make_binop.<locals>.visitor)r4 r �BinExprr# )r5 r> r- )r5 r. � _make_binop; s r@ c s" t dtjtd d�� fdd� �}|S )Nr" )r$ r% r& r! c s` | j jr6� | j jkr6| �d� �d�� | �|j|� n | �d� � | �|j|� | �d� d S )Nz environment.call_unop(context, z, r6 r8 )r9 r: Zintercepted_unopsr; r+ r% )r$ r% r& )r5 r- r. r> T s z_make_unop.<locals>.visitor)r4 r � UnaryExprr# )r5 r> r- )r5 r. � _make_unopQ s rB FTr )r% r9 �name�filename�stream� defer_init� optimizedr! c C sH t | tj�std��|�||||||�}|�| � |dkrD|j�� S dS )z+Generate the python source for a node tree.z Can't compile non template nodesN)� isinstancer �Template� TypeErrorZcode_generator_classr+ rE �getvalue)r% r9 rC rD rE rF rG � generatorr- r- r. �generatee s rM )�valuer! c C s� | dks| t ks| tkr dS t| �ttttttt hkr:dS t| �t t t t hkr`tdd� | D ��S t| �tkr�tdd� | �� D ��S dS )z)Does the node have a safe representation?NTc s s | ]}t |�V qd S )N)� has_safe_repr)�.0�vr- r- r. � <genexpr>� s z has_safe_repr.<locals>.<genexpr>c s s" | ]\}}t |�ot |�V qd S )N)rO )rP �krQ r- r- r. rR � s F)�NotImplemented�Ellipsis�type�bool�int�float�complex�range�strr �tuple�list�set� frozenset�all�dict�items)rN r- r- r. rO } s rO )r �namesr! c C s@ t |�}y x| D ]}|�|� qW W n tk r8 Y nX |jS )z�Check if the names passed are accessed undeclared. The return value is a set of all the undeclared names from the sequence of names found. )�UndeclaredNameVisitorr+ � VisitorExit� undeclared)r rd r> r% r- r- r. �find_undeclared� s rh c @ s* e Zd Zejejejf dd�dd�ZdS )�MacroRefN)r% r! c C s || _ d| _d| _d| _d S )NF)r% �accesses_caller�accesses_kwargs�accesses_varargs)r$ r% r- r- r. �__init__� s zMacroRef.__init__) �__name__� __module__� __qualname__r1 �Unionr �Macro� CallBlockrm r- r- r- r. ri � s ri c @ sd e Zd ZdZdeejd eje dd�dd�Zd d�dd�Z de d d �d d �Z d d�d d�Z e Z dS )r# z&Holds compile time information for us.N)r) �parent�levelr! c C s~ || _ || _|d kr4t|d�| _d| _d | _d | _n(t|j|d�| _|j| _|j| _|j| _d| _d| _d| _ d| _ d| _ d S )N)ru F) r) rt r �symbols�require_output_check�buffer�block�toplevel� rootlevel� loop_frame� block_frame� soft_frame)r$ r) rt ru r- r- r. rm � s zFrame.__init__)r! c C s2 t �tt�| j��}|j�| j� | j� � |_|S )z!Create a copy of the current one.) r1 r3 r# �object�__new__� __class__�__dict__�updaterv �copy)r$ �rvr- r- r. r� � s z Frame.copyF)�isolatedr! c C s& |rt | j| jjd d�S t | j| �S )zReturn an inner frame.r )ru )r# r) rv ru )r$ r� r- r- r. �inner� s z Frame.innerc C s | � � }d|_d|_|S )a& Return a soft frame. A soft frame may not be modified as standalone thing as it shares the resources with the frame it was created of, but it's not a rootlevel frame any longer. This is only used to implement if-statements and conditional expressions. FT)r� r{ r~ )r$ r� r- r- r. �soft� s z Frame.soft)NN)F)rn ro rp �__doc__r r1 �OptionalrX rm r� rW r� r� �__copy__r- r- r- r. r# � s . c @ s e Zd ZdZdS )rf z?Exception used by the `UndeclaredNameVisitor` to signal a stop.N)rn ro rp r� r- r- r- r. rf � s rf c @ sT e Zd ZdZdd�dd�Zejdd�dd�Zejdd�d d �Z ej dd�d d �Z dS ) �DependencyFinderVisitorz.A visitor that collects filter and test calls.N)r! c C s t � | _t � | _d S )N)r_ �filters�tests)r$ r- r- r. rm � s z DependencyFinderVisitor.__init__)r% r! c C s | � |� | j�|j� d S )N)� generic_visitr� �addrC )r$ r% r- r- r. � visit_Filter s z$DependencyFinderVisitor.visit_Filterc C s | � |� | j�|j� d S )N)r� r� r� rC )r$ r% r- r- r. � visit_Test s z"DependencyFinderVisitor.visit_Testc C s dS )zStop visiting at blocks.Nr- )r$ r% r- r- r. � visit_Block s z#DependencyFinderVisitor.visit_Block) rn ro rp r� rm r �Filterr� �Testr� �Blockr� r- r- r- r. r� � s r� c @ sJ e Zd ZdZeje dd�dd�Zej dd�dd�Z ej dd�d d �Z dS ) re z�A visitor that checks if a name is accessed without being declared. This is different from the frame visitor as it will not stop at closure frames. N)rd r! c C s t |�| _t � | _d S )N)r_ rd rg )r$ rd r- r- r. rm s z UndeclaredNameVisitor.__init__)r% r! c C sJ |j dkr8|j| jkr8| j�|j� | j| jkrFt� �n| j�|j� d S )N�load)�ctxrC rd rg r� rf �discard)r$ r% r- r- r. � visit_Name s z UndeclaredNameVisitor.visit_Namec C s dS )zStop visiting a blocks.Nr- )r$ r% r- r- r. r� ! s z!UndeclaredNameVisitor.visit_Block) rn ro rp r� r1 �Iterabler\ rm r �Namer� r� r� r- r- r- r. re s re c @ s e Zd ZdZdS )� CompilerExitz�Raised if the compiler encountered a situation where it just doesn't make sense to further process the code. Any block that raises such an exception is not further processed. N)rn ro rp r� r- r- r- r. r� % s r� c @ sJ e Zd ZU d�deje eje ejej eedd�dd�Ze ed�d d ��Z ee d d �d d�Z ed�dd�Z edd�dd�Zd�eedd�dd�Zdd�dd�Zd�e dd�dd �Zd�eejej dd �d d �Zedd�d d!�Zd�eeejej dd"�d#d$�Zejej edd%�d&d'�Zedd(�d)d*�Zd�eejej e dd,�d-d.�Zd�ejej e dd/�d0d1�Z d�ej ej ej ej f eejej!eej"f dd2�d3d4�Z#ejej dd5�d6d7�Z$edd�d8d9�Z%d�eedd:�d;d<�Z&d�eeed?�d@dA�Z'eedB�dCdD�Z(ej ej)ej*f eej+ee,f dE�dFdG�Z-e,eddH�dIdJ�Z.ejedK�dLdM�Z/eed�dNdO�Z0dd�dPdQ�Z1edd�dRdS�Z2dd�dTdU�Z3eddV�dWdX�Z4eddV�dYdZ�Z5dd�d[d\�Z6ed�d]d^�Z7ed�d_d`�Z8eed�dadb�Z9eedV�dcdd�Z:dd�dedf�Z;edd�dgdh�Z<d�ej=eje ddE�didj�Z>ej?eddE�dkdl�Z@ejAeddE�dmdn�ZBejCeddE�dodp�ZDej ejEejFf eddE�dqdr�ZGejEeddE�dsdt�ZHejFeddE�dudv�ZIejJeddE�dwdx�ZKejLeddE�dydz�ZMej)eddE�d{d|�ZNej*eddE�d}d~�ZOejPeddE�dd��ZQejReddE�d�d��ZSejTeddE�d�d��ZUG d�d�� d�ejV�ZWeXej"ej"d��d�d���ZYdZZejeW e[d�< eWd�d�d��Z\ejej" ed��d�d��Z]ej^eeWed��d�d��Z_ej^eeWdd��d�d��Z`ej^eeWdd��d�d��ZaejbeddE�d�d��ZcejdeddE�d�d��ZeejfeddE�d�d��ZgejheddE�d�d��ZiejjeddE�d�d��ZkejleddE�d�d��ZmejneddE�d�d��Zoej+eddE�d�d��ZpejqeddE�d�d��ZrejseddE�d�d��Zteud��Zveud��Zweud��Zxeud��Zyeud��Zzeud��Z{eud��Z|eud��Z}eud��Z~ed��Z�ed��Z�ed��Z�e�ej�eddE�d�d���Z�e�ej�eddE�d�d���Z�ej�eddE�d�d��Z�e�ej�eddE�d�d���Z�e�ej�eddE�d�d���Z�ej�eddE�d�d��Z�e�ej ej ej f eeej�d d��d�dÄ�Z�e�ej eddE�d�dń�Z�e�ej eddE�d�dDŽ�Z�e�ej�eddE�d�dɄ�Z�e�d�ej eeddʜd�d̄�Z�ej�eddE�d�d΄Z�ej�eddE�d�dЄZ�ej�eddE�d�d҄Z�ej�eddE�d�dԄZ�ej�eddE�d�dքZ�ej�eddE�d�dZ�ej�eddE�d�dڄZ�ej�eddE�d�d܄Z�ej�eddE�d�dބZ�ej�eddE�d�d�Z�ej�eddE�d�d�Z�ej�eddE�d�d�Z�ej�eddE�d�d�Z�ej�eddE�d�d�Z�ej�eddE�d�d�Z�dS )�r" NFTr )r9 rC rD rE rF rG r! c C s� |d krt � }|| _|| _|| _|| _d| _|| _d | _|rFt|�| _i | _ i | _ d| _ d| _ d| _ i | _i | _g | _d | _d| _d| _d| _d| _d| _g | _g | _dg| _d S )NFr r T�context)r r9 rC rD rE Zcreated_block_contextrF r( r �import_aliases�blocks�extends_so_far�has_known_extends� code_linenor� r� � debug_info�_write_debug_info� _new_lines� _last_line� _first_write�_last_identifier� _indentation� _assign_stack�_param_def_block�_context_reference_stack)r$ r9 rC rD rE rF rG r- r- r. rm - s8 zCodeGenerator.__init__)r! c C s | j d k S )N)r( )r$ r- r- r. rG v s zCodeGenerator.optimizedz te.NoReturn)�msg�linenor! c C s t ||| j| j��dS )z*Fail with a :exc:`TemplateAssertionError`.N)r rC rD )r$ r� r� r- r- r. �fail| s zCodeGenerator.failc C s | j d7 _ d| j � �S )z Get a new unique identifier.r Zt_)r� )r$ r- r- r. �temporary_identifier� s z"CodeGenerator.temporary_identifier)r& r! c C s | � � |_| �|j� d�� dS )z7Enable buffering for the frame from that point onwards.z = []N)r� rx � writeline)r$ r& r- r- r. rx � s zCodeGenerator.buffer)r& �force_unescapedr! c C s� |s�|j jrl| �d� | �� | �d|j� d�� | �� | �d� | �� | �d|j� d�� | �� dS |j jr�| �d|j� d�� dS | �d|j� d�� dS )z(Return the buffer contents of the frame.z if context.eval_ctx.autoescape:zreturn Markup(concat(z))zelse:zreturn concat(r8 N)r) r* r� �indentrx �outdent� autoescape)r$ r& r� r- r- r. �return_buffer_contents� s z$CodeGenerator.return_buffer_contentsc C s | j d7 _ dS )zIndent by one.r N)r� )r$ r- r- r. r� � s zCodeGenerator.indentr )�stepr! c C s | j |8 _ dS )zOutdent by step.N)r� )r$ r� r- r- r. r� � s zCodeGenerator.outdent)r& r% r! c C s0 |j dkr| �d|� n| �|j � d�|� dS )z%Yield or write into the frame buffer.Nzyield z.append()rx r� )r$ r& r% r- r- r. � start_write� s zCodeGenerator.start_writec C s |j dk r| �d� dS )z1End the writing process started by `start_write`.Nr8 )rx r; )r$ r& r- r- r. � end_write� s zCodeGenerator.end_write)�sr& r% r! c C s$ | � ||� | �|� | �|� dS )z4Simple shortcut for start_write + write + end_write.N)r� r; r� )r$ r� r& r% r- r- r. � simple_write� s zCodeGenerator.simple_write)r r& r! c C sB y(| � d� x|D ]}| �||� qW W n tk r< Y nX dS )z�Visit a list of nodes as block in a frame. If the current frame is no buffer a dummy ``if 0: yield None`` is written automatically. �passN)r� r+ r� )r$ r r& r% r- r- r. � blockvisit� s zCodeGenerator.blockvisit)�xr! c C s� | j rp| jsR| j�d| j � | j| j 7 _| jdk rR| j�| j| jf� d| _d| _| j�d| j � d| _ | j�|� dS )z&Write a string into the output stream.� NFz r ) r� r� rE r; r� r� r� �appendr� )r$ r� r- r- r. r; � s zCodeGenerator.writer )r� r% �extrar! c C s | � ||� | �|� dS )z!Combination of newline and write.N)�newliner; )r$ r� r% r� r- r- r. r� � s zCodeGenerator.writeline)r% r� r! c C s: t | jd| �| _|dk r6|j| jkr6|j| _|j| _dS )z/Add one or more newlines before the next write.r N)�maxr� r� r� r� )r$ r% r� r- r- r. r� � s zCodeGenerator.newline)r% r& � extra_kwargsr! c C s� t dd� tdd� |jD �|p d�D ��}x$|jD ]}| �d� | �||� q0W |s�x$|jD ]}| �d� | �||� qZW |dk r�x*|�� D ] \}}| �d|� d|� �� q�W |jr�| �d� | �|j|� |�r�|jdk r�| �d � n | �d � x8|jD ].}| �|j �d �� | �|j |� | �d� q�W |dk �rbx,|�� D ] \}}| �|�d |� d�� �q>W |jdk �r�| �d � | �|j|� | �d � n | �d� n$|jdk �r�| �d� | �|j|� dS )a+ Writes a function call to the stream for the current node. A leading comma is added automatically. The extra keyword arguments may not include python keywords otherwise a syntax error could occur. The extra keyword arguments should be given as python dict. c s s | ]}t t�t|��V qd S )N)�is_python_keywordr1 r3 r\ )rP rS r- r- r. rR � s z*CodeGenerator.signature.<locals>.<genexpr>c s s | ] }|j V qd S )N)�key)rP r� r- r- r. rR � s r- z, N�=z, *z , **dict({z, **{z: z}, **r8 �}z, **) �anyr r' �argsr; r+ rc Zdyn_argsZ dyn_kwargsr� rN ) r$ r% r& r� Zkwarg_workaround�arg�kwargr� rN r- r- r. � signature� sF zCodeGenerator.signature)r r! c C s t � }x|D ]}|�|� q W x�| j|jdf| j|jdffD ]�\}}}x�t|�D ]�}||krh| �� ||< | �d� | �� | �|| � d|� d|�d�� | �� | �d� | �� | �d� | �d || � d �� | �� | �d |d d � � d|�d�� | �� | �� qPW q<W d S )a� Find all filter and test names used in the template and assign them to variables in the compiled namespace. Checking that the names are registered with the environment is done when compiling the Filter and Test nodes. If the node is in an If or CondExpr node, the check is done at runtime instead. .. versionchanged:: 3.0 Filters and tests in If and CondExpr nodes are checked at runtime instead of compile time. r� r� ztry:z = environment.�[�]zexcept KeyError:z @internalcodezdef z (*unused):z raise TemplateRuntimeError("No N���z named z found.")) r� r+ r� r� �sortedr� r� r� r� )r$ r r> r% Zid_maprd Z dependencyrC r- r- r. �pull_dependencies s. z CodeGenerator.pull_dependenciesc C s� g }x�|j j�� D ]z\}\}}|tkr(q|tkrR| �|� d| �� � d|�d�� q|tkrp| �|� d|� �� q|tkr�|� |� qt d��qW |r�| �d� |�� d�� d S )Nz = r6 r8 zunknown load instructionz = missing) rv �loadsrc r r r� �get_resolve_funcr r r� �NotImplementedError�join)r$ r& �undefs�target�action�paramr- r- r. � enter_frameE s " zCodeGenerator.enter_frame)r& �with_python_scoper! c C sB |s>g }x|j jD ]}|�|� qW |r>| �d�|�� d�� d S )Nz = z = missing)rv r� r� r� r� )r$ r& r� r� r� r- r- r. � leave_frameU s zCodeGenerator.leave_frame�async � )� async_value� sync_valuer! c C s | j jr |S |S )N)r9 �is_async)r$ r� r� r- r- r. � choose_async] s zCodeGenerator.choose_async)rC r! c C s | � � � d|� �S )Nzdef )r� )r$ rC r- r- r. �func` s zCodeGenerator.func)r% r& r! c C s� |� � }|j�|� t|�}d}t� }g }xNt|j�D ]@\}}|jdkrL|}|jdkrb|�|j� |� |j� |j�� q6W t |j d�} d| kr�|dk r�y|j |t|j� W q� tk r� | �d|j� Y q�X n|� |j�d�� d|_d| k�rd|k�r|� |j�d�� d|_d| k�rDd|k�rD|� |j�d�� d|_d |_|j�|� | �| �d �� d d �|�� d �|� | �� | �|� | � |� | � |� x�t|j�D ]�\}}|j� |j�} | �d| � d�� | �� y|j |t|j� } W n6 tk �r* | �| � d|j�d|j�d�� Y n X | �| � d�� | � | |� | � | � | � � �q�W | �!� | �"|j |� | j#|dd� | j$|dd� | � � ||fS )z/Dump the function def of a macro or call block.N�caller)r' �varargs)r� r' r� zhWhen defining macros or call blocks the special "caller" argument must be omitted or be given a default.Tr' r� F�macror6 z, z):zif z is missing:z = undefined("parameter z was not provided", name=r8 z = )r� )r� )%r� rv � analyze_noderi r_ � enumerater� rC r� r� �refrh �body�defaults�len� IndexErrorr� r� �declare_parameterrj rk rl rw r� r� r� r� rx r� �push_parameter_definitionsr+ �mark_parameter_storedr� �pop_parameter_definitionsr� r� r� ) r$ r% r& � macro_refZexplicit_callerZskip_special_paramsr� �idxr� rg r� �defaultr- r- r. � macro_bodyc sn $ " zCodeGenerator.macro_body)r� r& r! c C sr d� dd� |jjD ��}t|jdd�}t|jj�dkr>|d7 }| �d|�d |� d |j�d|j�d|j�d � � dS ) z<Dump the macro definition for the def created by macro_body.z, c s s | ]}t |j�V qd S )N)�reprrC )rP r� r- r- r. rR � s z*CodeGenerator.macro_def.<locals>.<genexpr>rC Nr �,zMacro(environment, macro, z, (z), z , context.eval_ctx.autoescape)) r� r% r� �getattrr� r; rk rl rj )r$ r� r& Z arg_tuplerC r- r- r. � macro_def� s zCodeGenerator.macro_def)r% r! c C s* d|j � �}| jdk r&|� d| j��}|S )z.Return a human readable position for the node.zline Nz in )r� rC )r$ r% r� r- r- r. �position� s zCodeGenerator.positionc C s* d� dd� |j�� �� D ��}d|� d�S )Nz, c s s | ]\}}|�d |� �V qdS )z: Nr- )rP rC r� r- r- r. rR � s z3CodeGenerator.dump_local_context.<locals>.<genexpr>�{r� )r� rv Z dump_storesrc )r$ r& Zitems_kvr- r- r. �dump_local_context� s z CodeGenerator.dump_local_contextc C s, | � d� | � d� | � d� | � d� dS )z�Writes a common preamble that is used by root and block functions. Primarily this sets up common local helpers and enforces a generator through a dead branch. z$resolve = context.resolve_or_missingz!undefined = environment.undefinedz cond_expr_undefined = Undefinedzif 0: yield NoneN)r� )r$ r- r- r. � write_commons� s zCodeGenerator.write_commonsc C s | j �|j�� � dS )aQ Pushes all parameter targets from the given frame into a local stack that permits tracking of yet to be assigned parameters. In particular this enables the optimization from `visit_Name` to skip undefined expressions for parameters in macros as macros can reference otherwise unbound parameters. N)r� r� rv Zdump_param_targets)r$ r& r- r- r. r� � s z(CodeGenerator.push_parameter_definitionsc C s | j �� dS )z+Pops the current parameter definitions set.N)r� �pop)r$ r- r- r. r� � s z'CodeGenerator.pop_parameter_definitions)r� r! c C s | j r| j d �|� dS )z�Marks a parameter in the current parameter definitions as stored. This will skip the enforced undefined checks. r� N)r� r� )r$ r� r- r- r. r� � s z#CodeGenerator.mark_parameter_storedc C s | j �|� d S )N)r� r� )r$ r� r- r- r. �push_context_reference� s z$CodeGenerator.push_context_referencec C s | j �� d S )N)r� r )r$ r- r- r. �pop_context_reference� s z#CodeGenerator.pop_context_referencec C s | j d S )Nr� )r� )r$ r- r- r. �get_context_ref� s z CodeGenerator.get_context_refc C s | j d }|dkrdS |� d�S )Nr� r� �resolvez.resolve)r� )r$ r� r- r- r. r� � s z CodeGenerator.get_resolve_funcc C s | � � � d| �|�� d�S )Nz .derived(r8 )r r� )r$ r& r- r- r. �derive_context� s z CodeGenerator.derive_contextc C s | j s dS || j d kS )z4Checks if a given target is an undeclared parameter.Fr� )r� )r$ r� r- r- r. �parameter_is_undeclared s z%CodeGenerator.parameter_is_undeclaredc C s | j �t� � dS )z+Pushes a new layer for assignment tracking.N)r� r� r_ )r$ r- r- r. �push_assign_tracking s z"CodeGenerator.push_assign_trackingc C s� | j �� }|js |js |jr |s$dS dd� |D �}t|�dkr�tt|��}|j� |�}|jrv| � d|�d|� �� dS |jr�| � d|�d|� �� dS | � d|�d|� �� n||jr�| � d � n |jr�| � d � n | � d � xBt |�D ]6\}}|r�| � d � |j� |�}| � |�d |� �� q�W | � d� |j�s�|j�s�|�r�t|�dk�rf| � d|d �d�� n"d � tt|��}| � d|� d�� dS )zoPops the topmost level for assignment tracking and updates the context variables if necessary. Nc S s g | ]}|d d� dkr|�qS )Nr �_r- )rP r� r- r- r. � <listcomp> s z5CodeGenerator.pop_assign_tracking.<locals>.<listcomp>r z _loop_vars[z] = z _block_vars[z context.vars[z_loop_vars.update({z_block_vars.update({zcontext.vars.update({z, z: z})zcontext.exported_vars.add(r r8 z context.exported_vars.update((z)))r� r r} r| rz r� �next�iterrv r� r� r� r; r� �mapr� )r$ r& �varsZ public_namesrC r� r� � names_strr- r- r. �pop_assign_tracking sB z!CodeGenerator.pop_assign_trackingc C s$ |d kst d��t| j| j�}ddlm}m} | jjrDt|| �}nt|�}| � d� | � dd� |� � | j rtdnd}|� t j�d k }xD|�t j�D ]4} | j| jkr�| �d | j�d �| j� | | j| j< q�W x�|�t j�D ]v} | j| jkr�| j} | �� | j| < } d | k�r<| �d d�\} }| � d | � d |� d| � �� q�| � d| � d| � �� q�W | � d| j��� | j | �d�� d|� d�dd� | �� | �� t |�}dt |j d�k�r�|j � d�}| � |� d�� |j �!|� d |_"|_#|�o�| j$ |_%|�r| � d� | �&|� | �'|j � | �(|j |� | j)|dd� | �*� |�r�| j$�s`| �� | � d� | �� | jj�s~| � d � n$| � d � | �� | � d � | �*� | �*d| j$ � �x| j�+� D �]\}} | � | �d | �� d|� d�| d� | �� | �� t |�}d|_,t | j d �}d|k�r@|j � d�}| � |� d�� d!|k�rr|j � d!�}| � |� d"|�d#|� d$�� |j �!| � ||_-| � d%� | �&|� | �'| j � | �(| j |� | j)|dd� | �*� �q�W d� d&d'� | jD ��}| j d(|� d)�dd� d*� d+d'� | j.D ��}| � d,|��� d S )-Nzno root frame allowedr )�exported�async_exportedz%from __future__ import generator_stopzfrom jinja2.runtime import z, r� z, environment=environmentzblock z defined twice�.zfrom z import z as zimport zname = �rootz(context, missing=missingz):)r� r$ )r$ z = TemplateReference(context)Tzparent_template = None)r� z if parent_template is not None:z4yield from parent_template.root_render_func(context)z=async for event in parent_template.root_render_func(context):z yield eventZblock_)r$ �superr z = context.super(z, block_r8 z_block_vars = {}c s s | ]}|�d |� �V qdS )z: block_Nr- )rP r� r- r- r. rR � s z/CodeGenerator.visit_Template.<locals>.<genexpr>z blocks = {r� �&c s s | ]\}}|� d |� �V qdS )r� Nr- )rP rS rQ r- r- r. rR � s z debug_info = )/�AssertionErrorr r9 rC Zruntimer r r� r� r� r� rF �findr �Extends�find_allr� r� r� r� � ImportedName� importnamer� r� �rsplitr� r� r� r# rh r� rv r� r� rz r{ r� rw r� r� r� r� r� rc r} ry r� )r$ r% r& r) r r Zexported_namesZenvenvZ have_extendsry �import_�imp�alias�module�objr� rC r} rg Z blocks_kv_strZ debug_kv_strr- r- r. �visit_Template9 s� z CodeGenerator.visit_Templatec C s d}|j r8| jrdS | jdkr8| �d� | �� |d7 }|jrJ| �|�}n| �� }|jr�| �d|j �d�|� | �� | �d|j �d�|� | � � | j j s�|j dkr�| �d |j �d |� d �|� n@| �| �� � d |j �d |� d �|� | �� | �d|� | � � | � |� dS )z.Call a block and register it for the template.r Nzif parent_template is None:r zif len(context.blocks[z]) <= 1:z+raise TemplateRuntimeError("Required block z not found")zyield from context.blocks[z][0](r8 z for event in context.blocks[z):�event)rz r� r� r� r� �scopedr r �requiredrC r� r9 r� rx r� r� )r$ r% r& ru r� r- r- r. r� � s8 zCodeGenerator.visit_Blockc C s� |j s| �d|j� | jdkrV| js6| �d� | �� | �d� | jrNt� �n| �� | �d|� | � |j |� | � d| j �d�� | �d� | �� | �d � | �� |j r�d | _| jd 7 _d S ) zCalls the extender.z,cannot use extend from a non top-level scoper z if parent_template is not None:z5raise TemplateRuntimeError("extended multiple times")z+parent_template = environment.get_template(z, r8 z9for name, parent_block in parent_template.blocks.items():z8context.blocks.setdefault(name, []).append(parent_block)Tr N)rz r� r� r� r� r� r� r� r� r+ �templater; rC r{ )r$ r% r& r- r- r. � visit_Extends� s( zCodeGenerator.visit_Extendsc C sp |j r| �d� | �� d}t|jtj�rVt|jjt�r>d}qnt|jjt t f�rnd}nt|jtj tj f�rnd}| �d|� d�|� | � |j|� | �d| j�d�� |j r�| �� | �d � | �� | �d � | �� | �d � | �� d }|j�r| �| �� � d | �|�� d�� n$| jj�r,| �d� n| �d� d}|�s\| �� | �d|� | �� |j �rl| �� dS )zHandles includes.ztry:Zget_or_select_templateZ get_templateZselect_templateztemplate = environment.r6 z, r8 zexcept TemplateNotFound:r� zelse:FzUfor event in template.root_render_func(template.new_context(context.get_all(), True, z)):zGfor event in (await template._get_default_module_async())._body_stream:z6yield from template._get_default_module()._body_streamTr# N)Zignore_missingr� r� rH r& r �ConstrN r\ r] r^ �Tuple�Listr+ r; rC r� � with_contextr� r� r9 r� r� )r$ r% r& � func_nameZskip_event_yieldr- r- r. � visit_Include sJ zCodeGenerator.visit_Includec C s� | � | �d�� d�� | �|j|� | � d| j�d�� |jrld| �d�� �}| � |� d| �|�� d�� n| � d | �d�� d �� d S ) Nzawait zenvironment.get_template(z, z).Z make_moduleZ_asyncz(context.get_all(), True, r8 Z_get_default_modulez (context))r; r� r+ r& rC r+ r� )r$ r% r& �f_namer- r- r. �_import_common9 s z CodeGenerator._import_commonc C sl | � |j�|j�� d�|� |jr6| �d|j�d�� | �||� |jrh|j�d�sh| � d|j�d�� dS )zVisit regular imports.z = z context.vars[z] = r z context.exported_vars.discard(r8 N)r� rv r� r� rz r; r/ � startswith)r$ r% r& r- r- r. � visit_ImportH s zCodeGenerator.visit_Importc s� | � |� | �d� | �|� � g }g }x�|jD ]�}t|t�rH|\}}n|}| �� j�|�� d|�d�� | �d� j�|�� d�� | � � d| � |�� d|��}| �� j�|�� d|�d |�d �� | � � � j r0|� |� |�d �s0|� |� q0W |�r\t|�d k�r2|d }| �d|�d� j�|�� �� n*d�� fdd�|D ��}| �d|� d�� |�r�t|�d k�r�| �d|d �d �� n"d�tt|��} | �d| � d�� dS )zVisit named imports.zincluded_template = z = getattr(included_template, z , missing)zif z is missing:z9the template {included_template.__name__!r} (imported on z%) does not export the requested name z = undefined(fz, name=r8 r r r z context.vars[z] = z, c 3 s$ | ] }|�d � j �|�� �V qdS )z: N)rv r� )rP rC )r& r- r. rR y s z1CodeGenerator.visit_FromImport.<locals>.<genexpr>zcontext.vars.update({z})z context.exported_vars.discard(z)context.exported_vars.difference_update((z))N)r� r; r/ rd rH r] r� rv r� r� r� r� rz r� r0 r� r� r r� ) r$ r% r& Z var_namesZdiscarded_namesrC r �messageZnames_kvr r- )r& r. �visit_FromImportS sD z CodeGenerator.visit_FromImportc C s� |� � }d|_|� � }|� � }|jpRdt|jdd�d�kpRtdd� |�tj�D ��}d }|rh|j � d�}|j j |dd � |j r�|j j |d d � |j �r�| �� }|j j |d d � | �| �|�� d �|j � | �� | �|� | �| �d d�� | �|j|� | �d� | �| �dd�� | �d� | �� | �d|j � | �|j |� | �d� | �� | �d� | �|j|� | �d� | j|dd� |j�r�| �| �d�� d�|� | �� | �|� |j|_|�r�| �|� d�� x<|�tj�D ],} | jdk�r�| j dk�r�| � d| j � �q�W |j �r4| �� } | �| � d�� | �| �d d�|� | �|j|� |�r|| �d |� d| �d �� d �� n | �d� |j �r�| �|� d �� |j�r�| �d � nB| j j �r�|�s�| �d!� | �|j!|� | j j �r�|�s�| �d"� |j �r| �d"� |j�r| �d#� n| �|�r(d$nd� | �� | �|� | �d%� | �"|j#|� |j �rp| �| � d&�� | �� | j||j�o�|j d� |j �r�| �d| � d�� | �� | �|� | �"|j |� | �|� | �� |j�rd| �$|� | �� | �%||� | �| �d'�� d(�� | j j �r.| �d!� | �|j!|� | j j �rP| �d"� | �d)� | �&|� | j'�r�| j'd* �(|j j)� d S )+NT�loop)r� )�only)r4 c s s | ] }|j V qd S )N)r$ )rP ry r- r- r. rR � s z*CodeGenerator.visit_For.<locals>.<genexpr>r� )Z for_branch�else�testz(fiter):z async for zfor z in zauto_aiter(fiter)Zfiter�:zif zyield � )r� z%(reciter, loop_render_func, depth=0):z = missing�storez8Can't assign to special loop variable in for-loop targetz = 1z, ZAsyncz LoopContext(r6 Zreciterz auto_aiter(r8 z&, undefined, loop_render_func, depth):z , undefined):z_loop_vars = {}z = 0zawait zloop(z, loop)r� )*r� r| � recursiverh �iter_child_nodesr� r r r� rv r� r� �else_r7 r� r� r� r� r� r� r+ r� r; r� r� rx r� r� rC r� r� r9 r� r r� r� r� r� r� r� �difference_updateZstores) r$ r% r& r| Z test_frameZ else_frameZ extended_loopZloop_refZloop_filter_funcrC Ziteration_indicatorr- r- r. � visit_For� s� zCodeGenerator.visit_Forc C s� |� � }| �d|� | �|j|� | �d� | �� | �|j|� | �� xP|j D ]F}| �d|� | �|j|� | �d� | �� | �|j|� | �� qRW |j r�| �d� | �� | �|j |� | �� d S )Nzif r8 zelif zelse:) r� r� r+ r7 r; r� r� r� r� �elif_r= )r$ r% r& Zif_framer@ r- r- r. �visit_If s& zCodeGenerator.visit_Ifc C s| | � ||�\}}| �� |jrR|j�d�s>| �d|j�d�� | �d|j�d�� | �|j�|j�� d�� | � ||� d S )Nr zcontext.exported_vars.add(r8 z context.vars[z] = z = ) r� r� rz rC r0 r; r� rv r� r� )r$ r% r& Z macro_framer� r- r- r. � visit_Macro% s zCodeGenerator.visit_Macroc C sR | � ||�\}}| �d� | �||� | �||� | j|j|dd� | �|� d S )Nz caller = T)�forward_caller)r� r� r� r� � visit_Call�callr� )r$ r% r& Z call_framer� r- r- r. �visit_CallBlock/ s z CodeGenerator.visit_CallBlockc C sh |� � }|j�|� | �|� | �|� | �|j|� | �||� | �|j |� | � |� | � |� d S )N) r� rv r� r� rx r� r� r� r� �filterr� r� )r$ r% r& Z filter_framer- r- r. �visit_FilterBlock7 s z CodeGenerator.visit_FilterBlockc C s� |� � }|j�|� | �|� xDt|j|j�D ]2\}}| �� | �||� | � d� | �||� q.W | � |j |� | � |� d S )Nz = ) r� rv r� r� �zip�targets�valuesr� r+ r; r� r� r� )r$ r% r& Z with_framer� �exprr- r- r. � visit_WithB s zCodeGenerator.visit_Withc C s | � |� | �|j|� d S )N)r� r+ r% )r$ r% r& r- r- r. �visit_ExprStmtN s z CodeGenerator.visit_ExprStmtc @ s4 e Zd ZU ejejdef ed< eje ed< dS )zCodeGenerator._FinalizeInfo.�const�srcN)rn ro rp r1 r� �Callabler\ �__annotations__r- r- r- r. � _FinalizeInfoR s rS )rN r! c C s t | �S )z�The default finalize function if the environment isn't configured with one. Or, if the environment has one, this is called on that function's output for constants. )r\ )rN r- r- r. �_default_finalizeV s z CodeGenerator._default_finalize� _finalizec s� �j dk r�j S �j }� d}�jjr�d}�jj�tjdtjdtjdi�t����}d}|dkr|t j t j d�� �fdd� }n2|� |� d �}|dkr�t j t j d�� ��fd d� }�� ||��_ �j S ) a� Build the finalize function to be used on constants and at runtime. Cached so it's only created once for all output nodes. Returns a ``namedtuple`` with the following attributes: ``const`` A function to finalize constant data at compile time. ``src`` Source code to output around nodes to be evaluated at runtime. Nzenvironment.finalize(r� zcontext.eval_ctxr9 )rN r! c s � �| ��S )Nr- )rN )r� � env_finalizer- r. �finalize� s z.CodeGenerator._make_finalize.<locals>.finalizez, c s � ��j | ��S )N)r9 )rN )r� rV r$ r- r. rW � s ) rU rT r9 rW r r� � eval_context�get�from_objr1 r2 rS )r$ rW rP �pass_argr- )r� rV r$ r. �_make_finalize` s& z CodeGenerator._make_finalize)�groupr! c C s t t|��S )z�Given a group of constant values converted from ``Output`` child nodes, produce a string to write to the template module source. )r� r )r$ r] r- r- r. �_output_const_repr� s z CodeGenerator._output_const_repr)r% r& rW r! c C s: |� |j�}|jjr t|�}t|tj�r0t|�S |�|�S )aC Try to optimize a child of an ``Output`` node by trying to convert it to constant, finalized data at compile time. If :exc:`Impossible` is raised, the node is not constant and will be evaluated at runtime. Any other exception will also be evaluated at runtime for easier debugging. ) �as_constr) r� r rH r � TemplateDatar\ rO )r$ r% r& rW rO r- r- r. �_output_child_to_const� s z$CodeGenerator._output_child_to_constc C sL |j jr| �d� n |j jr(| �d� n | �d� |jdk rH| �|j� dS )zXOutput extra source code before visiting a child of an ``Output`` node. z1(escape if context.eval_ctx.autoescape else str)(zescape(zstr(N)r) r* r; r� rP )r$ r% r& rW r- r- r. �_output_child_pre� s z CodeGenerator._output_child_prec C s" | � d� |jdk r | � d� dS )zWOutput extra source code after visiting a child of an ``Output`` node. r8 N)r; rP )r$ r% r& rW r- r- r. �_output_child_post� s z CodeGenerator._output_child_postc C s� |j r"| jrd S | �d� | �� | �� }g }x�|jD ]�}y,|jsVt|tj�sVt� � �| � |||�}W n& tj t fk r� |� |� w6Y nX |r�t|d t �r�|d � |� q6|� |g� q6W |jd k �rt|�dkr�| �|j� d�� n| �|j� d�� | �� x�|D ]�}t|t ��rR| �|�}|jd k�rB| �d| � n| �|d � nb|jd k�rl| �d|� n | �|� | �|||� | �||� | �|||� |jd k �r | �d� �q W |jd k �r�| �� | �t|�dk�r�dnd � |j �r�| �� d S ) Nzif parent_template is None:r� r z.append(z .extend((zyield r� r8 z)))rw r� r� r� r\ r rO rH r` � Impossiblera � Exceptionr� r^ rx r� r^ r� rb r+ rc r; r� ) r$ r% r& rW r� �childrO �item�valr- r- r. � visit_Output� sV zCodeGenerator.visit_Outputc C sF | � � | �|� | �|j|� | �d� | �|j|� | �|� d S )Nz = )r r� r+ r� r; r% r )r$ r% r& r- r- r. � visit_Assign s zCodeGenerator.visit_Assignc C s� | � � |�� }d|_|j�|� | �|� | �|� | �|j|� | � |� | � |j |� | � d� |j d k r�| �|j |� n| � d|j� d�� | � d� | �|� | �|� d S )NFz9 = (Markup if context.eval_ctx.autoescape else identity)(zconcat(r8 )r r� rw rv r� r� rx r� r� r� r+ r� r; rG r� r r� )r$ r% r& r} r- r- r. �visit_AssignBlock s z CodeGenerator.visit_AssignBlockc C s� |j dkr4|js |js |jr4| jr4| jd �|j� |j�|j�}|j dkr�|j� |�}|d k rv|d t krv| � |�r�| � d|j�d|� d|� d�� d S | � |� d S ) Nr: r� r� r z(undefined(name=z) if z is missing else r8 ) r� rz r| r} r� r� rC rv r� Z find_loadr r r; )r$ r% r& r� r� r- r- r. r� 6 s zCodeGenerator.visit_Namec C sV |j �|j�}| �d|� d�� | �� | �d� | �� | �|� d|j�d�� d S )Nzif not isinstance(z , Namespace):zMraise TemplateRuntimeError("cannot assign attribute on non-namespace object")r� r� )rv r� rC r� r� r� �attr)r$ r% r& r� r- r- r. � visit_NSRefO s zCodeGenerator.visit_NSRefc C s8 |� |j�}t|t�r&| �t|�� n| �t|�� d S )N)r_ r) rH rY r; r\ r� )r$ r% r& rh r- r- r. � visit_Const] s zCodeGenerator.visit_Constc C sJ y| � t|�|j��� W n* tjk rD | � d|j�d�� Y nX d S )Nz6(Markup if context.eval_ctx.autoescape else identity)(r8 )r; r� r_ r) r rd �data)r$ r% r& r- r- r. �visit_TemplateDatad s z CodeGenerator.visit_TemplateDatac C sZ | � d� d}x0t|j�D ]"\}}|r0| � d� | �||� qW | � |dkrPdnd� d S )Nr6 r� z, r z,)r8 )r; r� rc r+ )r$ r% r& r� rg r- r- r. � visit_Tuplel s zCodeGenerator.visit_Tuplec C sJ | � d� x0t|j�D ]"\}}|r,| � d� | �||� qW | � d� d S )Nr� z, r� )r; r� rc r+ )r$ r% r& r� rg r- r- r. � visit_Listu s zCodeGenerator.visit_Listc C sd | � d� xJt|j�D ]<\}}|r,| � d� | �|j|� | � d� | �|j|� qW | � d� d S )Nr� z, z: r� )r; r� rc r+ r� rN )r$ r% r& r� rg r- r- r. � visit_Dict} s zCodeGenerator.visit_Dict�+�-�*�/z//z**�%�and�orznot c C sd |j jrd}n|j jr d}nd}| �|� d�� x$|jD ]}| �||� | �d� q8W | �d� d S )Nz8(markup_join if context.eval_ctx.volatile else str_join)Z markup_joinZstr_joinz((z, z)))r) r* r� r; r r+ )r$ r% r& r, r� r- r- r. � visit_Concat� s zCodeGenerator.visit_Concatc C sB | � d� | �|j|� x|jD ]}| �||� q W | � d� d S )Nr6 r8 )r; r+ rL Zops)r$ r% r& r5 r- r- r. � visit_Compare� s zCodeGenerator.visit_Comparec C s* | � dt|j � d�� | �|j|� d S )Nr7 )r; � operatorsr5 r+ rL )r$ r% r& r- r- r. � visit_Operand� s zCodeGenerator.visit_Operandc C sT | j jr| �d� | �d� | �|j|� | �d|j�d�� | j jrP| �d� d S )Nz(await auto_await(zenvironment.getattr(z, r8 z)))r9 r� r; r+ r% rl )r$ r% r& r- r- r. � visit_Getattr� s zCodeGenerator.visit_Getattrc C s� t |jtj�r@| �|j|� | �d� | �|j|� | �d� n^| jjrR| �d� | �d� | �|j|� | �d� | �|j|� | �d� | jjr�| �d� d S )Nr� r� z(await auto_await(zenvironment.getitem(z, r8 z))) rH r� r �Slicer+ r% r; r9 r� )r$ r% r& r- r- r. � visit_Getitem� s zCodeGenerator.visit_Getitemc C s` |j d k r| �|j |� | �d� |jd k r:| �|j|� |jd k r\| �d� | �|j|� d S )Nr8 )�startr+ r; �stopr� )r$ r% r& r- r- r. � visit_Slice� s zCodeGenerator.visit_Slice)r% r& � is_filterr! c c s | j jr| �d� |r@| �| j|j � d�� | j j�|j�}n(| �| j|j � d�� | j j�|j�}|d kr�|js�|r~dnd}| �d|� d|j�d�|j � t j dt j d t j d i�t � |��}|d k r�| �|� d �� d V | �||� | �d � | j j�r | �d � d S ) Nzawait auto_await(r6 rG r7 zNo z named r r� zcontext.eval_ctxr9 z, r8 )r9 r� r; r� rC rY r� r~ r� r� r r� rX rZ r� )r$ r% r& r� r� Z type_namer[ r- r- r. �_filter_test_common� s* z!CodeGenerator._filter_test_commonc C s� | � ||d��x |jd k r*| �|j|� nX|jjrP| �d|j� d|j� d�� n2|jjrn| �d|j� d�� n| �d|j� d�� W d Q R X d S )NTz(Markup(concat(z.)) if context.eval_ctx.autoescape else concat(z))zMarkup(concat(zconcat(r8 )r� r% r+ r) r* r; rx r� )r$ r% r& r- r- r. r� s zCodeGenerator.visit_Filterc C s, | � ||d�� | �|j|� W d Q R X d S )NF)r� r+ r% )r$ r% r& r- r- r. r� s zCodeGenerator.visit_Testc sl � � � � d d�� ��fdd� }��d� ���j� � ��d� ���j� � ��d� |� ��d� d S )N)r! c s8 �j d k r ���j � � d S ��d����� d�� d S )Nz1cond_expr_undefined("the inline if-expression on z6 evaluated to false and no else section was defined."))Zexpr2r+ r; r� r- )r& r% r$ r- r. � write_expr2 s z1CodeGenerator.visit_CondExpr.<locals>.write_expr2r6 z if z else r8 )r� r; r+ Zexpr1r7 )r$ r% r& r� r- )r& r% r$ r. �visit_CondExpr s z CodeGenerator.visit_CondExpr)r% r& rC r! c C s� | j jr| �d� | j jr&| �d� n | �d� | �|j|� |rJddind }|jr\ddini }|jrnddini }|r�|j|f|� n|s�|r�t |f|�}| � |||� | �d� | j jr�| �d� d S )Nzawait auto_await(zenvironment.call(context, z context.call(r� Z _loop_varsZ _block_varsr8 ) r9 r� r; r: r+ r% r| r} r� rb r� )r$ r% r& rC r� Z loop_kwargsZ block_kwargsr- r- r. rD 0 s" zCodeGenerator.visit_Callc C s" | � |jd � | �|j|� d S )Nr� )r; r� r+ rN )r$ r% r& r- r- r. � visit_KeywordG s zCodeGenerator.visit_Keywordc C s& | � d� | �|j|� | � d� d S )NzMarkup(r8 )r; r+ rL )r$ r% r& r- r- r. �visit_MarkSafeM s z CodeGenerator.visit_MarkSafec C s& | � d� | �|j|� | � d� d S )Nz6(Markup if context.eval_ctx.autoescape else identity)(r8 )r; r+ rL )r$ r% r& r- r- r. �visit_MarkSafeIfAutoescapeR s z(CodeGenerator.visit_MarkSafeIfAutoescapec C s | � d|j � d S )Nz environment.)r; rC )r$ r% r& r- r- r. �visit_EnvironmentAttributeY s z(CodeGenerator.visit_EnvironmentAttributec C s | � d|j�d|j� �� d S )Nzenvironment.extensions[z].)r; � identifierrC )r$ r% r& r- r- r. �visit_ExtensionAttribute^ s z&CodeGenerator.visit_ExtensionAttributec C s | � | j|j � d S )N)r; r� r )r$ r% r& r- r- r. �visit_ImportedNamec s z CodeGenerator.visit_ImportedNamec C s | � |j� d S )N)r; rC )r$ r% r& r- r- r. �visit_InternalNamef s z CodeGenerator.visit_InternalNamec C s | � d� d S )Nr� )r; )r$ r% r& r- r- r. �visit_ContextReferencei s z$CodeGenerator.visit_ContextReferencec C s | � | �|�� d S )N)r; r )r$ r% r& r- r- r. � visit_DerivedContextReferencen s z+CodeGenerator.visit_DerivedContextReferencec C s | � d|� d S )N�continue)r� )r$ r% r& r- r- r. �visit_Continues s z CodeGenerator.visit_Continuec C s | � d|� d S )N�break)r� )r$ r% r& r- r- r. � visit_Breakv s zCodeGenerator.visit_Breakc C s: |� � }|j�|� | �|� | �|j|� | �|� d S )N)r� rv r� r� r� r� r� )r$ r% r& � scope_framer- r- r. � visit_Scopey s zCodeGenerator.visit_Scopec C s� | � � }| �|� d| �|�� �� | �|� d�� | �|j|� | �|� |jdd�}|j�|� | � |� | � |j |� | � |� | � � d S )Nz = z.vars = T)r� )r� r� r r+ r� r r� rv r� r� r� r� r� r )r$ r% r& r� r� r- r- r. �visit_OverlayScope� s z CodeGenerator.visit_OverlayScopec C sx xr|j D ]h}| �d|j� d�� | �|j|� y|j�|j�}W n tjk r^ d|j_ Y qX t |j|j|� qW d S )Nzcontext.eval_ctx.z = T) �optionsr� r� r+ rN r_ r) r rd r* �setattr)r$ r% r& �keywordrh r- r- r. �visit_EvalContextModifier� s z'CodeGenerator.visit_EvalContextModifierc C sl | � � }|j�� }| �|� d�� | �||� x|jD ]}| �||� q6W |j�|� | �d|� d�� d S )Nz = context.eval_ctx.save()zcontext.eval_ctx.revert(r8 )r� r) Zsaver� r� r� r+ Zrevert)r$ r% r& Z old_ctx_nameZ saved_ctxrf r- r- r. � visit_ScopedEvalContextModifier� s z-CodeGenerator.visit_ScopedEvalContextModifier)NFT)F)r )N)N)Nr )Nr )N)F)r� r� )N)F)�rn ro rp r1 r� r\ �TextIOrW rm �propertyrG rX r� r� r# rx r� r� r� r �Noder� r� r� r� r� r; r� r� rq �Callr� r� �Mappingr2 r� r� r� r� r� r� rr rs r) ri r� r� r� r� r� r� r� r� r r r r� r r r r rI r" r� r� r r' �Includer- �ImportZ FromImportr/ r1 r3 �Forr? �IfrA rB rF Z FilterBlockrH �WithrM ZExprStmtrN � NamedTuplerS � staticmethodrT rU rR r\ r^ r0 ra rb rc ZOutputri �Assignrj Z AssignBlockrk r� r� ZNSRefrm r( rn r` rp rq r* rr �Dictrs r@ Z visit_AddZ visit_SubZ visit_MulZ visit_DivZvisit_FloorDivZ visit_PowZ visit_ModZ visit_AndZvisit_OrrB Z visit_PosZ visit_NegZ visit_Notr4 ZConcatr{ �Comparer| ZOperandr~ ZGetattrr ZGetitemr� r� r� r �Iteratorr� r� r� ZCondExprr� rD �Keywordr� ZMarkSafer� ZMarkSafeIfAutoescaper� ZEnvironmentAttributer� ZExtensionAttributer� r r� Z InternalNamer� ZContextReferencer� ZDerivedContextReferencer� �Continuer� �Breakr� ZScoper� Z OverlayScoper� ZEvalContextModifierr� ZScopedEvalContextModifierr� r- r- r- r. r" , s "A ,4+Q .u++4 2 0 O ' )NFT)Fr� �typingr1 � contextlibr � functoolsr �ior � itertoolsr r� r r� Z markupsafer r r� r � exceptionsr Z idtrackingr r r r r r r( r �utilsr r r> r � TYPE_CHECKINGZtyping_extensions�ter9 r �TypeVarrQ r2 r r} r4 r\ r? r@ rA rB rI r� r� rW rM rO r� r� �Setrh ri r# � RuntimeErrorrf r� re re r� r" r- r- r- r. �<module> sd " & S