File: //opt/saltstack/salt/lib/python3.10/site-packages/jinja2/__pycache__/async_utils.cpython-310.pyc
o
�N�g � @ s� d dl Z d dlZd dlmZ d dlmZ ddlmZ ddlmZ ej r'd dl
Z e� d�Z
dd � Zeeeeeeeed�hZd
ejejd df d dfd d
�ZG dd� deje
�Z ddd�Z d
dd ej d fdd�Z dS )� N)�WRAPPER_ASSIGNMENTS)�wraps� )�_PassArg)�pass_eval_context�Vc s � fdd�}|S )Nc s� t ���}|d u �|t ju rdtjdtfdd��n
dtjdtfdd��d}ttt�� |��}t
�|d�t
� |dd �� ���fd
d ���}�rJt |�}d |_ |S )
N�args�returnc S s t �t| d j�S �Nr )�t�cast�bool�is_async�r � r �F/opt/saltstack/salt/lib/python3.10/site-packages/jinja2/async_utils.pyr s z2async_variant.<locals>.decorator.<locals>.is_asyncc S s t �t| d jj�S r
)r r r
� environmentr r r r r r s )�
__module__�__name__� __qualname__)�assignedr )r �updatedc s8 �| �}�r | dd � } |r� | i |��S �| i |��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
�tuple�setr �
differencer r Zjinja_async_variant)r Zpass_argZasync_func_attrsZnormal_func_attrsr �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 | �tv r
t�d| �S t�| �rt�d| �I d H S | S )Nr zt.Awaitable[V])�type�_common_primitivesr r �inspectZ isawaitable�r$ r r r �
auto_await> s �
r) c @ s. e Zd Zd dd�Zd
dd�Zdefd d
�Zd S )�_IteratorToAsyncIterator�iterator�
t.Iterator[V]c C s
|| _ d S �N)� _iterator)�selfr+ r r r �__init__J s
z!_IteratorToAsyncIterator.__init__r �te.Selfc C s | S r- r )r/ r r r � __aiter__M s z"_IteratorToAsyncIterator.__aiter__c
� s4 �zt | j�W S ty } zt|j�|�d }~ww r- )�nextr. �
StopIteration�StopAsyncIterationr$ )r/ �er r r � __anext__P s � ��z"_IteratorToAsyncIterator.__anext__N)r+ r, )r r1 )r r r r0 r2 r r7 r r r r r* I s
r* �iterable�*t.Union[t.AsyncIterable[V], t.Iterable[V]]�t.AsyncIterator[V]c C s t | d�r | �� S tt| ��S )Nr2 )�hasattrr2 r* �iter)r8 r r r �
auto_aiterW s
r= c � s �dd� t | �2 �I d H S )Nc � s �g | z3 d H W }|�q6 S r- r )�.0�xr r r �
<listcomp>c s � z auto_to_list.<locals>.<listcomp>)r= r( r r r � auto_to_list` s �rA )r8 r9 r r: ) r' �typingr � functoolsr r �utilsr r Z
TYPE_CHECKINGZtyping_extensions�te�TypeVarr r# �int�floatr
�str�list�dictr r% r& �Union� Awaitabler) ZGenericr* r= ZListrA r r r r �<module> s.
," �
� ��