File: //usr/local/lib/python3.6/site-packages/tqdm/__pycache__/asyncio.cpython-36.pyc
3
LBj�
� @ s^ d Z ddlZddlmZ ddlmZ ddgiZdd d
d gZG d d� de�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/Z casperdcl� tqdm_asyncio�taranger �trangec sf e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zedddd �d d
��Z edddd �dd��Z
� Z S )r z>
Asynchronous-friendly version of tqdm (Python 3.6+).
Nc sl t t| �j|f|�|� d| _|d k rht|d�r@|j| _d| _n(t|d�rT|j| _nt|�| _ | j j| _d S )NF� __anext__T�__next__)
�superr �__init__�iterable_awaitable�hasattrr �
iterable_nextr �iterZiterable_iterator)�self�iterable�args�kwargs)� __class__� �,/tmp/pip-build-44150kxb/tqdm/tqdm/asyncio.pyr s
ztqdm_asyncio.__init__c C s | S )Nr )r r r r � __aiter__$ s ztqdm_asyncio.__aiter__c � sl y*| j r| j� I d H }n| j� }| j� |S tk
rJ | j� t�Y n tk
rf | j� � Y nX d S )N)r r �update�
StopIteration�close�StopAsyncIteration�
BaseException)r �resr r r r ' s ztqdm_asyncio.__anext__c O s | j j||�S )N)r �send)r r r r r r r 6 s ztqdm_asyncio.send)�loop�timeout�totalc k s\ |dkrt |�}i }tdd� dk r,||d< | tj|fd|i|��fd|i|��E dH dS ) z5
Wrapper for `asyncio.as_completed`.
N� � �
r r r! )r# r$ )�lenr �asyncio� as_completed)�cls�fsr r r! � tqdm_kwargsr r r r r' 9 s ztqdm_asyncio.as_completedc � sZ 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 )Nr )�i�fr r r �wrap_awaitableK s z+tqdm_asyncio.gather.<locals>.wrap_awaitablec s g | ]\}}� ||��qS r r )�.0r+ r, )r- r r �
<listcomp>N s z'tqdm_asyncio.gather.<locals>.<listcomp>c � s g | ]}|I d H �qS )Nr )r. r, r r r r/ O s )r r r! Nc S s g | ] \}}|�qS r r )r. �_r+ r r r r/ Q s )� enumerater' �sorted)r( r r r! r) r* Zifsr r )r- r �gatherF s
ztqdm_asyncio.gather)N) �__name__�
__module__� __qualname__�__doc__r r r r � classmethodr' r3 �
__classcell__r r )r r r s
c O s t t| � f|�S )zE
A shortcut for `tqdm.asyncio.tqdm(range(*args), **kwargs)`.
)r �range)r r r r r r T s ) r7 r& �sysr Zstdr Zstd_tqdm�
__author__�__all__r r r r r r r �<module> s
A