U
C^% @ s0 d Z ddlmZmZ ddlZddlmZ ddlmZ dZ
dZ z>ddl Z dZ
ze
e jdd Z W n ek
rv Y nX W nn ek
r d Z
ddlZe @ ejd
d d zddlm mZ W n ek
r Y nX W 5 Q R X Y nX zLe
d krdd
lmZ ddlmZmZ dZ
n dd
l mZ ddl mZmZ W n^ ek
r z,ddlm Z ddlm Z ddlmZ dZ
W n ek
r dZ
Y nX Y nX zddl m Z W n ek
r Y nX zddlm Z W n" ek
r ddl!m Z Y nX ddddgiZ"ddd d gZ#G d d de Z$d d Z%e$Ze%Z&dS ) z
IPython/Jupyter Notebook progressbar decorator for iterators.
Includes a default (x)range iterator printing to stderr.
Usage:
>>> from tqdm.notebook import trange[, tqdm]
>>> for i in trange(10): #same as: for i in tqdm(xrange(10))
... ...
)divisionabsolute_importN )_range)tqdm . ignorez2.*The `IPython.html` package has been deprecated.*)message)
FloatProgress)HBoxHTML )FloatProgressWidget)ContainerWidget)r )display)escapez github.com/Zlrq3000Z casperdclZ alexanderkuk
tqdm_notebooktnranger trangec sr e Zd ZdZedddZdddZ fdd Z fd
d Z fd d
Z fddZ
dd Z d fdd Z Z
S )r zB
Experimental IPython/Jupyter Notebook widget using tqdm!
Nc C s z.|rt d|d}nt ddd}d|_d|_W n tk
rJ tdY nX |rf||_tdkrfd|j_t }t
||gd}|d k rt |}zt |dkr|d
7 }W n t
k
r Y nX d |j_||j_d |j_d
|j_t| |S )zY
Manage the printing of an IPython/Jupyter Notebook progress bar widget.
r )minmaxr infozFloatProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html initial)childrenNZpx2z inline-flexzrow wrap) IProgressvalue bar_style NameError ImportError descriptionIPYWstyledescription_widthr r
strint
ValueErrorZlayoutZflexwidthr Z flex_flow)_totaldescncolspbarptext container r3 //tmp/pip-install-6_kvzl1k/tqdm/tqdm/notebook.pystatus_printerQ s:
z tqdm_notebook.status_printerFc C s |s|s| }| jj\}}| j|_|rd|krHtt|dd\}}ndt| }}|rr|d dkrr|d d }|r|d dkr|dd }||_t dkrd|j
_ |r||_|r|j d kr|d
ks||_ |r
|j d kr
z| j
W n tk
r d | j_Y nX d S ) Nr |r r r dangersuccessF)__repr__r2 r nr mapr splitr$ r% r& r' r! closeAttributeErrorZvisible) selfmsgposr@ r! r0 r1 leftrightr3 r3 r4 r s2
ztqdm_notebook.displayc s | dtj}|tjks |d kr*tj|d< d|d< |dd |d dd|d< t| dd |d< tt| j || | j
s|d sd S | j rd
n
| d d | _ | j
dkrd n| j
pd }| jr| j| n| j}| | j|| j| j | _| j| _| j
s| d S )
NfileTguiZ
bar_formatz{l_bar}{bar}{r_bar}z{bar}r6 disableFz100%r/ r )getsysstderrstdout
setdefaultreplaceboolsuperr __init__rI Z
dynamic_ncolsr/
unit_scaler- r5 fpr. r2 r sp)rB argskwargsZ
file_kwargrS r- __class__r3 r4 rR s,
ztqdm_notebook.__init__c / sB z"t t| j||D ]
}|V qW n | jdd Y nX d S Nr: r! )rQ r __iter__rU )rB rV rW objrX r3 r4 r\ s ztqdm_notebook.__iter__c
sN zt t| j|| W n2 tk
rH } z| jdd |W 5 d }~X Y nX d S rZ )rQ r update ExceptionrU )rB rV rW excrX r3 r4 r^ s
ztqdm_notebook.updatec s` t t| j|| t| dr\| jr<| j| jk r<| jdd n | jrP| jdd n | jdd d S )NrU r: r[ r; T)r@ )rQ r r@ hasattrr- r= rU ZleaverB rV rW rX r3 r4 r@ s
ztqdm_notebook.closec O s d S )Nr3 rb r3 r3 r4 moveto s ztqdm_notebook.movetoc s, |dk r| j 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.
N)r- )r2 r r rQ r reset)rB r- r0 r, rX r3 r4 rd s
ztqdm_notebook.reset)NNN)NNFN)N)__name__
__module__ __qualname____doc__ staticmethodr5 r rR r\ r^ r@ rc rd
__classcell__r3 r3 rX r4 r L s 3
4
c O s t t| f|S )z}
A shortcut for `tqdm.notebook.tqdm(xrange(*args), **kwargs)`.
On Python3+, `range` is used instead of `xrange`.
)r r )rV rW r3 r3 r4 r s )'rh
__future__r r rK utilsr Zstdr Zstd_tqdmZIPYr% Z
ipywidgetsr) __version__r? rA r# warningscatch_warningsfilterwarningsZIPython.html.widgetshtmlZwidgetsr r r
r r r ZIPython.displayr r cgi
__author____all__r r r r3 r3 r3 r4 sn
A