a
�`) � @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z dd l m
Z
dd
l mZ d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Ze
dd� �Zdd � Zd0d d �Zd d!� Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z G d,d-� d-e
�Z d1d.d/�Z e � dS )2zWThe code for async support. Importing this patches Jinja on supported
Python versions.
� N��update_wrapper)�Markup� )�TemplateModule)� LoopContext��concat�� internalcode)�missingc � s& g �� �fdd�}|� I d H t ��S )Nc � s" � 2 z3 d H W } �� | � q6 d S �N)�append)�event�� async_gen�rv� �7/usr/lib/python3.9/site-packages/jinja2/asyncsupport.py�collect s z concat_async.<locals>.collectr )r r r r r � concat_async s r c O s\ t |i |��}z*| �| �|��2 z3 d H W }|V q 6 W n tyV | j�� V Y n0 d S r
)�dict�root_render_func� new_context� Exception� environment�handle_exception)�self�args�kwargs�varsr r r r �generate_async s r! c s dd� � � �fdd�}t |��S )Nc s s> | j |i |��}z|�|�� �V qW n ty8 Y n0 d S r
)r! �run_until_complete� __anext__�StopAsyncIteration)r �loopr r r r r r �_convert_generator' s
z.wrap_generate_func.<locals>._convert_generatorc s0 | j js �| g|�R i |��S � | t�� ||�S r
)r �is_async�asyncio�get_event_loop)r r r �r& �original_generater r �generate/ s z$wrap_generate_func.<locals>.generater )r+ r, r r* r �wrap_generate_func& s r- c � s` | j jstd��t|i |��}| �|�}zt| �|��I d H W S tyZ | j �� Y S 0 d S )Nz8The environment was not created with async mode enabled.) r r' � RuntimeErrorr r r r r r )r r r r �ctxr r r � render_async7 s
r0 c s � fdd�}t |� �S )Nc s< | j js � | g|�R i |��S t�� }|�| j|i |���S r
)r r' r( r) r" r0 )r r r r% ��original_renderr r �renderE s z wrap_render_func.<locals>.renderr )r2 r3 r r1 r �wrap_render_funcD s r4 c s( t dd� �� t � �fdd��}t|��S )Nc � s2 t | j| j | j��I d H }| jjjr.t|�}|S r
)r Z_stackZ_depth�_contextZeval_ctx�
autoescaper �r r r r r �
async_callO s
z-wrap_block_reference_call.<locals>.async_callc s | j jjs�| �S � | �S r
)r5 r r' �r �r8 �
original_callr r �__call__V s
z+wrap_block_reference_call.<locals>.__call__�r r )r; r< r r: r �wrap_block_reference_callN s
r> c s( t dd� �� t � �fdd��}t|��S )Nc � s | j |� I d H }|r t|�}|S r
)Z_funcr )r � argumentsr6 r r r r � async_invoke` s z'wrap_macro_invoke.<locals>.async_invokec s | j js�| ||�S � | ||�S r
)Z _environmentr' )r r? r6 �r@ �original_invoker r �_invokeg s z"wrap_macro_invoke.<locals>._invoker= )rB rC r rA r �wrap_macro_invoke_ s
rD c � s( | j d ur| j S | �� I d H | _ }|S r
)�_module�make_module_asyncr7 r r r �get_default_module_asyncp s
rG c s t � fdd��}|S )Nc s | j jrtd��� | �S )Nz6Template module attribute is unavailable in async mode)r r' r. r9 ��original_default_moduler r �_get_default_moduley s z0wrap_default_module.<locals>._get_default_moduler
)rI rJ r rH r �wrap_default_modulex s rK Fc � sB | � |||�}g }| �|�2 z3 d H W }|�|� q 6 t| ||�S r
)r r r r )r r Zshared�locals�contextZ body_stream�itemr r r rF � s
rF c C sd ddl m} t| j�| _tt| j�| _tt| j�| _t| j�| _t | j
�| _
t | _ tt
| j
�| _
d S )Nr ��Template)� rP r- r, r r! r0 r4 r3 rK rJ rG Z_get_default_module_asyncrF rO r r r �patch_template� s �rR c C s, ddl m} m} t| j�| _t|j�|_d S )Nr ��BlockReference�Macro)�runtimerT rU r> r< rD rC rS r r r �
patch_runtime� s rW c C s& ddl m} ddlm} | �|� d S )Nr )�FILTERS)�
ASYNC_FILTERS)�filtersrX Z asyncfiltersrY �update)rX rY r r r �
patch_filters� s r\ c C s t � t� t� d S r
)rR rW r\ r r r r � patch_all� s r] c � s t �| �r| I d H S | S r
)�inspectZ isawaitable)�valuer r r �
auto_await� s
r` c C s< t | d�r(| 2 z3 d H W }|V q6 d S | D ]
}|V q,d S )N� __aiter__)�hasattr)�iterablerN r r r �
auto_aiter� s
rd c @ sh e Zd Zee�Zedd� �Zedd� �Zedd� �Z dd� Z
ed d
� �Z ed d � �Z d
d� Z
dd� ZdS )�AsyncLoopContextc � sv | j d ur| j S zt| j�| _ W nN tyn dd� | j2 �I d H }| �|�| _t|�| j | jtu | _ Y n0 | j S )Nc � s g | z3 d H W }|�q6 S r
r )�.0�xr r r �
<listcomp>� � z+AsyncLoopContext.length.<locals>.<listcomp>) Z_length�lenZ _iterable� TypeError� _iterator� _to_iterator�index�_afterr )r rc r r r �length� s
zAsyncLoopContext.lengthc � s | j I d H | j S r
)rp rn r9 r r r � revindex0� s zAsyncLoopContext.revindex0c � s | j I d H | j S r
)rp �index0r9 r r r �revindex� s zAsyncLoopContext.revindexc � sF | j tur| j S z| j�� I d H | _ W n ty> t| _ Y n0 | j S r
)ro r rl r# r$ r9 r r r �
_peek_next� s
zAsyncLoopContext._peek_nextc � s | � � I d H tu S r
)rt r r9 r r r �last� s zAsyncLoopContext.lastc � s$ | � � I d H }|tu r | �d�S |S )Nzthere is no next item)rt r Z
_undefinedr7 r r r �nextitem� s
zAsyncLoopContext.nextitemc C s | S r
r r9 r r r ra � s zAsyncLoopContext.__aiter__c � sL | j tur| j }t| _ n| j�� I d H }| jd7 _| j| _|| _|| fS )Nr )ro r rl r# rr Z_currentZ_beforer7 r r r r# � s
zAsyncLoopContext.__anext__N)�__name__�
__module__� __qualname__� staticmethodrd rm �propertyrp rq rs rt ru rv ra r# r r r r re � s
re c � s& dd l }|jdtdd� t| |||�S )Nr zRThis template must be recompiled with at least Jinja 2.11, or it will fail in 3.0.� )�
stacklevel)�warnings�warn�DeprecationWarningre )rc Z undefinedZrecurseZdepth0r~ r r r �make_async_loop_context� s �r� )NFN)Nr )!�__doc__r( r^ � functoolsr Z
markupsafer r r rV r Zutilsr r r r r! r- r0 r4 r> rD rG rK rF rR rW r\ r] r` rd re r� r r r r �<module> s:
A