U ܟj@sdZddlmZddlmZmZmZmZmZm Z m Z ddl m Z ddl m ZddgiZd d d d gZGd ddeZGdddeZGdd d eZdd ZeZ eZdS)z~ `rich.progress` decorator for iterators. Usage: >>> from tqdm.rich import trange, tqdm >>> for i in trange(10): ... ... )warn) BarColumnProgressProgressColumnTextTimeElapsedColumnTimeRemainingColumnfilesize)TqdmExperimentalWarning)tqdmz github.com/Z casperdcl tqdm_richtrranger trangecs*eZdZdZdfdd ZddZZS) FractionColumnz*Renders completed/total, e.g. '0.5/2.3 G'.Fcs||_||_tdSN) unit_scale unit_divisorsuper__init__)selfrr __class__-/tmp/pip-unpacked-wheel-ktn_y856/tqdm/rich.pyrszFractionColumn.__init__c Cst|j}|jdk rt|jnd}|dk r.|n|}|jr`t|dddddddd d g |j\}}nt|dgd \}}|d krd nd }|dk r||d |dnd}t||d |dd|d|ddS)z.Calculate common unit for completed and total.NKMGTPEZYr r,.f?/ zprogress.downloadstyle)int completedtotalrr pick_unit_and_suffixrr) rtaskr-r.baseunitsuffix precisionZ total_strrrrrenders     zFractionColumn.render)Fr__name__ __module__ __qualname____doc__rr5 __classcell__rrrrrsrcs*eZdZdZd fdd ZddZZS) RateColumnz&Renders human readable transfer speed.rFrcs ||_||_||_tdSr)r2rrrr)rr2rrrrrr3szRateColumn.__init__c Cs|j}|dkr$td|jdddS|jrRt|dddd d d d d dg |j\}}nt|dgd\}}|dkrrdnd}t||d|dd||jdddS)zShow data transfer speed.Nz? z/szprogress.data.speedr*rrrrr r!r"r#r$r rr%r&r))speedrr2rr r/r)rr0r=r2r3r4rrrr59s $zRateColumn.render)rFrr6rrrrr<1sr<csJeZdZdZfddZfddZddZdd Zd fd d ZZ S)r z/Experimental rich.progress GUI version of tqdm!c s|}d|d<t|dd|d<|dd}|di}tj|||jrXdStdtd d |j }|dkrd t dd t |d |dddt dt dt|d|d |dddf }|d|j t|||_|j|jj|jpdf||_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()`. TguidisableFprogressNoptionszrich is experimental/alpha) stacklevelzU[progress.description]{task.description}[progress.percentage]{task.percentage:>4.0f}%)Z bar_widthrr)rr[<,r2)r2rr]Z transientr)copyboolgetpoprrr?rr Z format_dictrrrrr< setdefaultZleaver_prog __enter__Zadd_taskdesc_task_id)rargskwargsr@rAdrrrrNs>     ztqdm_rich.__init__cs0|jr dS|t|jddddSr)r?displayrcloserM__exit__)rrrrrUws  ztqdm_rich.closecOsdSrrr___rrrclear~sztqdm_rich.clearcOs*t|dsdS|jj|j|j|jddS)NrP)r- description)hasattrrMupdaterPnrOrWrrrrTs ztqdm_rich.displayNcs*t|dr|jj|dtj|ddS)z Resets to 0 iterations for repeated use. Parameters ---------- total : int or float, optional. Total to use for the new bar. rM)r.N)r\rMresetr)rr.rrrr_s ztqdm_rich.reset)N) r7r8r9r:rrUrZrTr_r;rrrrr Ks  ) cOstt|f|S)z6Shortcut for `tqdm.rich.tqdm(range(*args), **kwargs)`.)r range)rQrRrrrrsN)r:warningsrZ rich.progressrrrrrrr Zstdr r Zstd_tqdm __author____all__rr<r rrrrrrs $    H