3
U'íc ã @ s¨ d dl Z d dlZd dlmZ ddlmZ ddlmZ ejdZ dd Z
e e e
eeeeedhZejejd df dd d
d Zd d
dddZd ejd d ddZdS )é N)Úwrapsé )Ú_PassArg)Úpass_eval_contextÚVc s fdd}|S )Nc sr t j}|d k|t jkr0tjtdddntjtdddt fdd}rht|}d|_|S )N)ÚargsÚreturnc S s t jt| d jS )Nr )ÚtÚcastÚboolÚis_async)r © r
ú4/tmp/pip-build-gk9425m9/Jinja2/jinja2/async_utils.pyr s z2async_variant.<locals>.decorator.<locals>.is_asyncc S s t jt| d jjS )Nr )r r
r Ú environmentr )r r
r
r r s c s0 | }r| dd
} |r& | |S | |S )Nr r
)r ÚkwargsÚb)Ú
async_funcr Úneed_eval_contextÚ normal_funcr
r Úwrapper s
z1async_variant.<locals>.decorator.<locals>.wrapperT) r Zfrom_objr r ZAnyr r r Zjinja_async_variant)r Zpass_argr )r )r r r r Ú decorator s
z async_variant.<locals>.decoratorr
)r r r
)r r Ú
async_variant s r )Úvaluer c à s@ t | tkrtjd| S tj| r4tjd| I d H S tjd| S )Nr zt.Awaitable[V])ÚtypeÚ_common_primitivesr r
ÚinspectÚ isawaitable)r r
r
r Ú
auto_await2 s
r z*t.Union[t.AsyncIterable[V], t.Iterable[V]]zt.AsyncIterator[V])Úiterabler c C sj t | drJxZtjd| 2 d H y 3 d H }W n
tk
s<X |V q Y W n xtjd| D ]
}|V qXW d S )NÚ __aiter__zt.AsyncIterable[V]z
t.Iterable[V])Úhasattrr r
ÚStopAsyncIteration)r Úitemr
r
r Ú
auto_aiter= s
*r# c às dd t | 2 d H I d H S )Nc
Ó s. g | y 3 d H }W n
t k
s"X |q Y S )N)r! )Ú.0Úxr
r
r ú
<listcomp>K s z auto_to_list.<locals>.<listcomp>)r# )r r
r
r Ú auto_to_listH s r' )r Útypingr Ú functoolsr Úutilsr r ZTypeVarr r ÚintÚfloatr ÚstrÚlistÚdictÚtupler r ÚUnionÚ Awaitabler r# ÚListr' r
r
r
r Ú<module> s
$