File: //opt/saltstack/salt/lib/python3.10/site-packages/jinja2/__pycache__/async_utils.cpython-310.pyc
o
;j � @ 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 �Anyr
�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 )r0 r r r � __aiter__M s z"_IteratorToAsyncIterator.__aiter__c
� s4 �zt | j�W S ty } zt|j�|�d }~ww r. )�nextr/ �
StopIteration�StopAsyncIterationr% )r0 �er r r � __anext__P s � ��z"_IteratorToAsyncIterator.__anext__N)r, r- )r r2 )r r r r1 r3 r r8 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 )Nr3 )�hasattrr3 r+ �iter)r9 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 �rB )r9 r: r r; ) r( �typingr � functoolsr r �utilsr r �
TYPE_CHECKINGZtyping_extensions�te�TypeVarr r$ �int�floatr
�str�list�dictr r& r' �Union� Awaitabler* �Genericr+ r> �ListrB r r r r �<module> s.
," �
� ��