File: //home/arjun/.local/lib/python3.10/site-packages/tqdm/__pycache__/asyncio.cpython-310.pyc
o
���g�
� @ sZ d Z ddlZddlmZ ddlmZ ddgiZg d�ZG d d
� d
e�Z d d � Z
e Ze
Z dS )
z�
Asynchronous progressbar decorator for iterators.
Includes a default `range` iterator printing to `stderr`.
Usage:
>>> from tqdm.asyncio import trange, tqdm
>>> async for i in trange(10):
... ...
� N)� version_info� )�tqdmz github.com/� casperdcl)� tqdm_asyncio�taranger �trangec sf e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zedddd �d d
��Z edddd �dd��Z
� Z S )r z0
Asynchronous-friendly version of tqdm.
Nc sv t � j|g|�R i |�� d| _|d ur9t|d�r"|j| _d| _d S t|d�r-|j| _d S t|�| _| jj| _d S d S )NF� __anext__T�__next__) �super�__init__�iterable_awaitable�hasattrr �
iterable_nextr
�iter�iterable_iterator)�self�iterable�args�kwargs�� __class__� �?/home/arjun/.local/lib/python3.10/site-packages/tqdm/asyncio.pyr s
�ztqdm_asyncio.__init__c C s | S �Nr )r r r r � __aiter__$ s ztqdm_asyncio.__aiter__c � s` �z| j r
| �� I d H }n| �� }| �� |W S ty$ | �� t� ty/ | �� � w r )r
r �update�
StopIteration�close�StopAsyncIteration�
BaseException)r �resr r r r ' s � �ztqdm_asyncio.__anext__c O s | j j|i |��S r )r �send)r r r r r r r" 6 s ztqdm_asyncio.send��loop�timeout�totalc k s^ � |du r t |�}i }tdd� dk r||d< | tj|fd|i|��fd|i|��E dH dS )z5
Wrapper for `asyncio.as_completed`.
N� )� �
r$ r% r& )�lenr �asyncio� as_completed)�cls�fsr$ r% r& � tqdm_kwargsr r r r r, 9 s ���ztqdm_asyncio.as_completedc � s\ �dd� � � fdd�t |�D �}dd� | j|f|||d�|��D �I dH }dd� t|�D �S ) z/
Wrapper for `asyncio.gather`.
c � s �| |I d H fS r r )�i�fr r r �wrap_awaitableK s �z+tqdm_asyncio.gather.<locals>.wrap_awaitablec s g | ] \}}� ||��qS r r )�.0r0 r1 �r2 r r �
<listcomp>N s z'tqdm_asyncio.gather.<locals>.<listcomp>c � s �g | ]}|I d H �qS r r )r3 r1 r r r r5 O s � r# Nc S s g | ]\}}|�qS r r )r3 �_r0 r r r r5 Q s )� enumerater, �sorted)r- r$ r% r& r. r/ �ifsr! r r4 r �gatherF s ���ztqdm_asyncio.gatherr ) �__name__�
__module__� __qualname__�__doc__r r r r" � classmethodr, r: �
__classcell__r r r r r s
r c O s t t| � fi |��S )zE
A shortcut for `tqdm.asyncio.tqdm(range(*args), **kwargs)`.
)r �range)r r r r r r T s r ) r> r+ �sysr �stdr �std_tqdm�
__author__�__all__r r r r r r r �<module> s
A