3
sݨ\� � @ s d Z ddlZddlZddlZddlmZ ddlmZmZm Z ddl
m Z ddl m
Z
mZ dd� Zd d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zedd� �Zdd� Zd.dd�Zd d � Zd d � Zd d!� Zd"d#� Z d$d%� Z d&d'� Z G d(d)� d)e
�Z G d*d+� d+e �Z!d/d,d-�Z"dS )0z�
jinja2.asyncsupport
~~~~~~~~~~~~~~~~~~~
Has all the code for async support which is implemented as a patch
for supported Python versions.
:copyright: (c) 2017 by the Jinja Team.
:license: BSD, see LICENSE for more details.
� N)�update_wrapper)�concat� internalcode�Markup)�TemplateModule)�LoopContextBase�_last_iterationc � s& g �� �fdd�}|� I d H t ��S )Nc � s> x8� 2 d H y 3 d H } W n
t k
s.X �j| � q
Y W d S )N)�StopAsyncIteration�append)�event)� async_gen�rv� �"/usr/lib/python3.6/asyncsupport.py�collect s "z concat_async.<locals>.collect)r )r r r )r r
r � concat_async s r c O s� t ||�}yFx@| j| j|��2 d H y 3 d H }W n
tk
sBX |V q" Y W W n tk
rl tj� }Y nX d S | jj|d�V d S )NT) �dict�root_render_func� new_contextr � Exception�sys�exc_info� environment�handle_exception)�self�args�kwargs�varsr r r r r �generate_async s
.r c s dd� � � �fdd�}t |��S )Nc s s@ | j ||�}yx|j|j� �V qW W n tk
r: Y nX d S )N)r �run_until_complete� __anext__r )r �loopr r r r r r �_convert_generator, s z.wrap_generate_func.<locals>._convert_generatorc s* | j js�| f|�|�S � | tj� ||�S )N)r �is_async�asyncio�get_event_loop)r r r )r" �original_generater r �generate3 s z$wrap_generate_func.<locals>.generate)r )r& r' r )r" r& r �wrap_generate_func+ s r( c � sd | j jstd��t||�}| j|�}yt| j|��I d H S tk
rT tj � }Y nX | j j
|d�S )Nz8The environment was not created with async mode enabled.T) r r# � RuntimeErrorr r r r r r r r )r r r r �ctxr r r r � render_async: s
r+ c s � fdd�}t |� �S )Nc s2 | j js� | f|�|�S tj� }|j| j||��S )N)r r# r$ r% r r+ )r r r r! )�original_renderr r �renderJ s z wrap_render_func.<locals>.render)r )r, r- r )r, r �wrap_render_funcI s r. c s( t dd� �� t � �fdd��}t|��S )Nc � s2 t | j| j | j��I d H }| jjjr.t|�}|S )N)r Z_stackZ_depth�_contextZeval_ctx�
autoescaper )r r
r r r �
async_callS s
z-wrap_block_reference_call.<locals>.async_callc s | j jjs�| �S � | �S )N)r/ r r# )r )r1 �
original_callr r �__call__Z s
z+wrap_block_reference_call.<locals>.__call__)r r )r2 r3 r )r1 r2 r �wrap_block_reference_callR s r4 c s( t dd� �� t � �fdd��}t|��S )Nc � s | j |� I d H }|r t|�}|S )N)Z_funcr )r � argumentsr0 r
r r r � async_invoked s z'wrap_macro_invoke.<locals>.async_invokec s | j js�| ||�S � | ||�S )N)Z _environmentr# )r r5 r0 )r6 �original_invoker r �_invokek s z"wrap_macro_invoke.<locals>._invoke)r r )r7 r8 r )r6 r7 r �wrap_macro_invokec s r9 c � s( | j d k r| j S | j� I d H | _ }|S )N)�_module�make_module_async)r r
r r r �get_default_module_asyncs s
r< c s t � fdd��}|S )Nc s | j jrtd��� | �S )Nz6Template module attribute is unavailable in async mode)r r# r) )r )�original_default_moduler r �_get_default_module| s z0wrap_default_module.<locals>._get_default_module)r )r= r> r )r= r �wrap_default_module{ s r? Fc � s^ | j |||�}g }x>| j|�2 d H y 3 d H }W n
tk
sFX |j|� q" Y W t| ||�S )N)r r r r
r )r r Zshared�locals�contextZ body_stream�itemr r r r; � s
(r; c C sd ddl m} t| j�| _tt| j�| _tt| j�| _t| j�| _t | j
�| _
t | _ tt
| j
�| _
d S )Nr )�Template)Zjinja2rC r( r' r r r+ r. r- r? r> r< Z_get_default_module_asyncr; )rC r r r �patch_template� s
rD c C s, ddl m} m} t| j�| _t|j�|_d S )Nr )�BlockReference�Macro)�jinja2.runtimerE rF r4 r3 r9 r8 )rE rF r r r �
patch_runtime� s
rH c C s& ddl m} ddlm} | j|� d S )Nr )�FILTERS)�
ASYNC_FILTERS)Zjinja2.filtersrI Zjinja2.asyncfiltersrJ �update)rI rJ r r r �
patch_filters� s rL c C s t � t� t� d S )N)rD rH rL r r r r � patch_all� s rM c � s t j| �r| I d H S | S )N)�inspectZ isawaitable)�valuer r r �
auto_await� s
rP c C s\ t | d�rDx4| 2 d H y 3 d H }W n
tk
s4X |V q Y W d S x| D ]
}|V qJW d S )N� __aiter__)�hasattrr )�iterablerB r r r �
auto_aiter� s
"
rT c @ s* e Zd Zd dd�Zedd� �Zdd� ZdS )
�AsyncLoopContextNr c C s& t j| |||� || _|| _|| _d S )N)r �__init__�_async_iterator�_after�_length)r �async_iterator� undefined�after�length�recurse�depth0r r r rV � s zAsyncLoopContext.__init__c C s | j d krtd��| j S )NzHLoop length for some iterators cannot be lazily calculated in async mode)rY � TypeError)r r r r r] � s
zAsyncLoopContext.lengthc C s t | �S )N)�AsyncLoopContextIterator)r r r r rQ � s zAsyncLoopContext.__aiter__)Nr )�__name__�
__module__� __qualname__rV �propertyr] rQ r r r r rU � s
rU c @ s( e Zd Zd Zdd� Zdd� Zdd� ZdS )
ra rA c C s
|| _ d S )N)rA )r rA r r r rV � s z!AsyncLoopContextIterator.__init__c C s | S )Nr )r r r r rQ � s z"AsyncLoopContextIterator.__aiter__c � sp | j }| jd7 _|jtkr$t� �|j|_|j|_y|jj� I d H |_W n tk
rd t|_Y nX |j|fS )N� ) rA Zindex0rX r r Z_currentZ_beforerW r )r r* r r r r � s
z"AsyncLoopContextIterator.__anext__N)rA )rb rc rd � __slots__rV rQ r r r r r ra � s ra c � s� y t | �}W n8 ttfk
rD t| d�s<t| �} t | �}nd }Y nX t| �}y|j� I d H }W n tk
rx t}Y nX t ||||||�S )NrQ )
�lenr` �AttributeErrorrR �tuplerT r r r rU )rS r[ r^ r_ r] rZ r\ r r r �make_async_loop_context� s
rk )NFN)Nr )#�__doc__r r$ rN � functoolsr Z jinja2.utilsr r r Zjinja2.environmentr rG r r r r r( r+ r. r4 r9 r< r? r; rD rH rL rM rP rT rU �objectra rk r r r r �<module> s4