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.ddZd 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 internalcodeMarkup)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)StopAsyncIterationappend)event) async_genrv "/usr/lib/python3.6/asyncsupport.pycollect s "z concat_async..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|dV d S )NT) dictroot_render_func new_contextr Exceptionsysexc_info environmenthandle_exception)selfargskwargsvarsr 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.._convert_generatorc s* | j js| f||S | tj ||S )N)r is_asyncasyncioget_event_loop)r r r )r" original_generater r generate3 s z$wrap_generate_func..generate)r )r& r' r )r" r& r wrap_generate_func+ s r( c sd | j jstdt||}| j|}yt| j|I d H S tk
rT tj }Y nX | j j
|dS )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..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 | jI 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..async_callc s | j jjs| S | S )N)r/ r r# )r )r1
original_callr r __call__Z s
z+wrap_block_reference_call..__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..async_invokec s | j js| ||S | ||S )N)Z _environmentr# )r r5 r0 )r6 original_invoker r _invokek s z"wrap_macro_invoke.._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)_modulemake_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.._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 ZsharedlocalscontextZ body_streamitemr 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 )BlockReferenceMacro)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 | drDx4| 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 ddZedd Zdd ZdS )
AsyncLoopContextNr c C s& t j| ||| || _|| _|| _d S )N)r __init___async_iterator_after_length)r async_iterator undefinedafterlengthrecursedepth0r 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| ds s4