File: //usr/local/lib/python3.6/site-packages/tqdm/__pycache__/rich.cpython-36.pyc
3
LBj � @ s� d Z ddlmZ ddlmZ ddlmZmZmZm Z m
Z
m Z m Z ddl
mZ ddl
mZ ddlmZ d d
giZd d d
dgZG dd� de�ZG dd� de�ZG dd � d e�Zdd � ZeZeZdS )z~
`rich.progress` decorator for iterators.
Usage:
>>> from tqdm.rich import trange, tqdm
>>> for i in trange(10):
... ...
� )�absolute_import)�warn)� BarColumn�Progress�ProgressColumn�Text�TimeElapsedColumn�TimeRemainingColumn�filesize� )�TqdmExperimentalWarning)�tqdm)�_rangez github.com/Z casperdcl� tqdm_rich�trranger
�trangec s* e Zd ZdZd� fdd� Zdd� Z� ZS ) �FractionColumnz*Renders completed/total, e.g. '0.5/2.3 G'.F�� c s || _ || _t� j� d S )N)�
unit_scale� unit_divisor�super�__init__)�selfr r )� __class__� �)/tmp/pip-build-44150kxb/tqdm/tqdm/rich.pyr s zFractionColumn.__init__c C s� t |j�}t |j�}| jrBtj|ddddddddd g | j�\}}ntj|dgd
�\}}|d
krbd nd
}t|| d |� d
��d|| d |� d
��d|� �dd�S )z.Calculate common unit for completed and total.� �K�M�G�T�P�E�Z�Yr r z,.�f�/� zprogress.download)�style)�int� completed�totalr r
�pick_unit_and_suffixr r )r �taskr* r+ �unit�suffix� precisionr r r �render s
.zFractionColumn.render)Fr )�__name__�
__module__� __qualname__�__doc__r r1 �
__classcell__r r )r r r s r c s* e Zd ZdZd � fdd� Zdd� Z� ZS )
�
RateColumnz&Renders human readable transfer speed.r F�� c s || _ || _|| _t� j� d S )N)r. r r r r )r r. r r )r r r r 3 s zRateColumn.__init__c C s� |j }|dkr$td| j� d�dd�S | jrRtj|dddd d
d d d
dg | j�\}}ntj|dgd�\}}|dkrrdnd}t|| d|� d��d|� | j� d�dd�S )zShow data transfer speed.Nz? z/szprogress.data.speed)r( r r r r r r! r" r# r$ r r z,.r% r' )�speedr r. r r
r, r )r r- r9 r. r/ r0 r r r r1 9 s $zRateColumn.render)r Fr8 )r2 r3 r4 r5 r r1 r6 r r )r r r7 1 s r7 c sJ e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Zd
� fd d � Z� Z S )r z/Experimental rich.progress GUI version of tqdm!c
s |j � }d|d< t|jdd��|d< |jdd�}|jdi �j � }tt| �j||� | jr\dS tdt d d
� | j
}|dkr�d t dd �t |d
|d d�dt
� dt� dt|d |d
|d d�df
}|jd| j � t||�| _| jj� | jj| jp�df|�| _dS )aZ
This class accepts the following parameters *in addition* to
the parameters accepted by `tqdm`.
Parameters
----------
progress : tuple, optional
arguments for `rich.progress.Progress()`.
options : dict, optional
keyword arguments for `rich.progress.Progress()`.
T�gui�disableF�progressN�optionszrich is experimental/alpha� )�
stacklevelzU[progress.description]{task.description}[progress.percentage]{task.percentage:>4.0f}%)Z bar_widthr r )r r �[�<�,r. )r. r r �]Z transientr )�copy�bool�get�popr r r r; r r Z format_dictr r r r r7 �
setdefaultZleaver �_prog� __enter__Zadd_task�desc�_task_id)r �args�kwargsr< r= �d)r r r r N s,
ztqdm_rich.__init__c s, | j r
d S tt| �j� | jjd d d � d S )N)r; r r �closerI �__exit__)r )r r r rP w s ztqdm_rich.closec O s d S )Nr )r �_�__r r r �clear} s ztqdm_rich.clearc O s* t | d�sd S | jj| j| j| jd� d S )NrI )r* � description)�hasattrrI �updaterL �nrK )r rR rS r r r �display� s
ztqdm_rich.displayNc s. t | d�r| jj|d� tt| �j|d� dS )z�
Resets to 0 iterations for repeated use.
Parameters
----------
total : int or float, optional. Total to use for the new bar.
rI )r+ N)rV rI �resetr r )r r+ )r r r rZ � s
ztqdm_rich.reset)N)
r2 r3 r4 r5 r rP rT rY rZ r6 r r )r r r K s ) c O s t t| � f|�S )zy
A shortcut for `tqdm.rich.tqdm(xrange(*args), **kwargs)`.
On Python3+, `range` is used instead of `xrange`.
)r r )rM rN r r r r � s N)r5 �
__future__r �warningsr Z
rich.progressr r r r r r r
Zstdr r
Zstd_tqdm�utilsr �
__author__�__all__r r7 r r r r r r r �<module> s $
G