U
Z$d)+ @ s d Z ddlZddlZddlmZ ddlmZ dZzddl Z dZW nn e
k
r dZddl Z e @ e j
dd d
zddlm mZ W n e
k
r Y nX W 5 Q R X Y nX zZedkrdd lmZ dd lmZ dd
lmZ dZn$dd l mZ dd l mZ dd
l mZ W nf e
k
rp z,dd lmZ ddlmZ ddlmZ dZW n" e
k
rj dZdZeZY nX Y nX zddlmZ W n e
k
r Y nX zddlmZ W n" e
k
r ddlmZ Y nX ddddgiZ ddddgZ d Z G d d d eZ G d d deZ d d Z!e Ze!Z"dS )!z
IPython/Jupyter Notebook progressbar decorator for iterators.
Includes a default `range` iterator printing to `stderr`.
Usage:
>>> from tqdm.notebook import trange, tqdm
>>> for i in trange(10):
... ...
N)proxy )tqdm ignorez2.*The `IPython.html` package has been deprecated.*)message)HTML)
FloatProgress)HBox )ContainerWidget)FloatProgressWidget )display)escapez github.com/Zlrq3000Z casperdclZ alexanderkuk
tqdm_notebooktnranger trangez|IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.htmlc s4 e Zd ZdZd
ddZd fdd Zdd Z ZS ) TqdmHBoxz.`ipywidgets.HBox` with a pretty representationNc C s4 t | dd }|d kri S |j}|d k r0| |d< |S )Npbarascii)getattr format_dict)selfprettyr d r 1/tmp/pip-unpacked-wheel-6ucdd40t/tqdm/notebook.py_json_L s
zTqdmHBox._json_Fc s4 t | dd }|d kr"tt| S |jf | |S )Nr )r superr __repr__ format_meterr )r r r __class__r r r! U s zTqdmHBox.__repr__c O s | | d d S )NT)textr! )r pp___r r r
_repr_pretty_[ s zTqdmHBox._repr_pretty_)N)F)__name__
__module__ __qualname____doc__r r! r)
__classcell__r r r# r r J s
r c s e Zd ZdZedddZdddZed d
Zej d d
Z fd d
Z
fddZ d fdd Z fddZ
dd Zd fdd Z ZS ) r zB
Experimental IPython/Jupyter Notebook widget using tqdm!
Nc C s t dkrtt|r"t d|d}n(t ddd}d|_d|_|dkrJd|j_t }t }|r`||_t|||gd}|dk rt |}zt
|dkr|d7 }W n t k
r Y nX d |j_ ||j_d
|j_
d |j_|S ) zY
Manage the printing of an IPython/Jupyter Notebook progress bar widget.
Nr )minmaxr infoZ20px)childrenZpx2z inline-flexzrow wrap) IProgress ImportError WARN_NOIPYWvalue bar_stylelayoutwidthr r strint
ValueErrorZflexr Z flex_flow)r' totaldescncolsr ltextrtext containerr r r status_printerc s4
z tqdm_notebook.status_printerFTc C s |s2|s2| j }|d pddd|d< | jf |}| jj\}}} | j|_|rd|krnttt
d|d\}
} ndt| }
} |
|_| r| | _|r|j dks|d kr||_ |r|j dkrz| j W n t
k
r d
| j_Y nX d | jj_|r| jd kr| jst| j d
| _d S )NZ
bar_formatz{l_bar}{r_bar}z{bar}zz \|?\|?r dangersuccessFhiddenr T)r replacer" rC r2 nr7 mapr resplitr8 closeAttributeErrorZvisibler9 Z
visibilitydelay displayedr ) r msgposrN r8 check_delayr rA r rB leftrightr r r r s8
ztqdm_notebook.displayc C s t | dr| jjd jjS d S NrC hasattrrC r2 style bar_colorr r r r colour s
ztqdm_notebook.colourc C s t | dr|| jjd j_d S rW rY )r r\ r r r r^ s
c sL | }|dtj}|tjks(|dkr2tj|d< d|d< t|dd|d< |dd}|dd}tt| j || | j
s|d sd d
| _ dS | j rd n
|d d| _
| jdkrd
n| jpd
}| jr| j| n| j}| | j|| j| j
| _t| | j_d| _|r&| jdkr&t| j d| _| j| _ || _| j
sH| jdd dS )z
Supports the usual `tqdm.tqdm` parameters as well as those listed below.
Parameters
----------
display : Whether to call `display(self.container)` immediately
[default: True].
fileNTguidisableFr^ r c _ s d S Nr )r' r( r r r z(tqdm_notebook.__init__..z100%r@ r r )rT )copygetsysstderrstdoutboolpopr r __init__ra dispZ
dynamic_ncolsr@
unit_scaler> rD fpr? rC r r rQ rP r r^ )r argskwargsZ
file_kwargr^ Z display_herern r> r# r r rl s2
ztqdm_notebook.__init__c # sB z"t t| }|D ]
}|V qW n | jdd Y nX d S )NrF r8 )r r __iter__rm )r itobjr# r r rs s ztqdm_notebook.__iter__r c s4 zt t| j|dW S | jdd Y nX d S )N)rJ rF rr )r r updaterm )r rJ r# r r rv s
ztqdm_notebook.updatec sb | j r
d S tt| | jr:| j| jk r:| jddd n$| jrP| jddd n| jddd d S )NrF F)r8 rT rG T)rN rT )ra r r rN r> rJ rm Zleaver] r# r r rN s ztqdm_notebook.closec O s d S rb r )r r' r( r r r clear" s ztqdm_notebook.clearc sd | j rtt| j|dS | jj\}}}d|_|dk rR||_| jsR| j dkrRd|j
_ tt| j|dS )z
Resets to 0 iterations for repeated use.
Consider combining with `leave=True`.
Parameters
----------
total : int or float, optional. Total to use for the new bar.
)r> rE N) ra r r resetrC r2 r8 r0 r> r@ r9 r: )r r> r' r r# r r rx % s
ztqdm_notebook.reset)NNN)NNFNT)r )N)r* r+ r, r- staticmethodrD r propertyr^ setterrl rs rv rN rw rx r. r r r# r r _ s$ ,
4
-
c O s t t| f|S )z:Shortcut for `tqdm.notebook.tqdm(range(*args), **kwargs)`.)r range)rp rq r r r r : s )#r- rL rg weakrefr Zstdr Zstd_tqdmZIPYZ
ipywidgetsr5 warningscatch_warningsfilterwarningsZIPython.html.widgetshtmlZwidgetsr r
r4 r r
r objectZIPython.displayr r cgi
__author____all__r6 r r r r r r r r sp
\