U
    )d+                     @   s>  d Z ddlmZmZ ddlZddlZddlmZ ddlm	Z
 ddlmZ dZzddlZdZW nn ek
r   d	ZddlZe @ ejd
dd zddlm  mZ W n ek
r   Y nX W 5 Q R X Y nX z\ed	krddlmZ ddlmZ ddlmZ dZn$ddlmZ ddlmZ ddlmZ W nf ek
r   z,ddlmZ ddlmZ ddlmZ dZW n" ek
r   dZdZeZY nX Y nX zddlmZ W n ek
r   Y nX zddlmZ W n" ek
r   ddl mZ Y nX ddddgiZ!ddddgZ"dZ#G dd  d eZ$G d!d de
Z%d"d Z&e%Z	e&Z'dS )#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):
...     ...
    )absolute_importdivisionN)proxy   )tqdm)_range       ignorez2.*The `IPython.html` package has been deprecated.*)message)HTML)FloatProgress)HBox   )ContainerWidget)FloatProgressWidget   )display)escapezgithub.com/Zlrq3000Z	casperdclZalexanderkuktqdm_notebooktnranger   trangez|IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.htmlc                       s4   e Zd ZdZd
ddZd fdd	Zdd	 Z  ZS )TqdmHBoxz.`ipywidgets.HBox` with a pretty representationNc                 C   s4   t | dd }|d kri S |j}|d k	r0| |d< |S )Npbarascii)getattrformat_dict)selfprettyr   d r    1/tmp/pip-unpacked-wheel-u8lm3wqg/tqdm/notebook.py_repr_json_Q   s    
zTqdmHBox._repr_json_Fc                    s4   t | dd }|d kr"tt|  S |jf | |S )Nr   )r   superr   __repr__format_meterr"   )r   r   r   	__class__r    r!   r$   Z   s    zTqdmHBox.__repr__c                 O   s   | | d d S )NT)textr$   )r   pp___r    r    r!   _repr_pretty_`   s    zTqdmHBox._repr_pretty_)N)F)__name__
__module____qualname____doc__r"   r$   r,   __classcell__r    r    r&   r!   r   O   s   
	r   c                       s   e Zd ZdZedddZdddZed	d
 Zej	dd
 Z fddZ
 fddZd fdd	Z fddZdd Zd fdd	Z  ZS )r   zB
    Experimental IPython/Jupyter Notebook widget using tqdm!
    Nc                 C   s   t dkrtt|r"t d|d}n(t ddd}d|_d|_|dkrJd|j_t }t }|r`||_t|||gd}|dk	rt	|}zt
|dkr|d7 }W n tk
r   Y nX d	|j_||j_d
|j_d|j_|S )zY
        Manage the printing of an IPython/Jupyter Notebook progress bar widget.
        Nr   )minmaxr   infoZ20px)childrenZpx2zinline-flexzrow wrap)	IProgressImportErrorWARN_NOIPYWvalue	bar_stylelayoutwidthr   r   strint
ValueErrorZflexr   Z	flex_flow)r*   totaldescncolsr   ltextrtext	containerr    r    r!   status_printerh   s4    ztqdm_notebook.status_printerFTc                 C   s  |s2|s2| j }|d pddd|d< | jf |}| jj\}}}	| j|_|rd|krnttt	
d|d\}
}ndt| }
}|
|_|r||	_|r|jdks|d	kr||_|r|jdkrz| j  W n tk
r   d
| j_Y nX |r| jdkr| jst| j d| _d S )NZ
bar_formatz{l_bar}<bar/>{r_bar}z{bar}z<bar/>z\|?<bar/>\|?r    dangersuccessFr   T)r   replacer%   rF   r5   nr:   mapr   resplitr;   closeAttributeErrorZvisibledelay	displayedr   )r   msgposrP   r;   check_delayr   rD   r   rE   leftrightr    r    r!   r      s6    
 
ztqdm_notebook.displayc                 C   s   t | dr| jjd jjS d S NrF   hasattrrF   r5   style	bar_colorr   r    r    r!   colour   s    
ztqdm_notebook.colourc                 C   s   t | dr|| jjd j_d S rY   r[   )r   r^   r    r    r!   r`      s    
c                    sL  |  }|dtj}|tjks(|dkr2tj|d< d|d< t|dd|d< |dd}|dd}tt| j	|| | j
s|d sd	d
 | _dS | jrdn
|dd| _| jdkrdn| jpd}| jr| j| n| j}| | j|| j| j| _t| | j_d| _|r&| jdkr&t| j d| _| j| _|| _| j
sH| jdd dS )z
        Supports the usual `tqdm.tqdm` parameters as well as those listed below.

        Parameters
        ----------
        display  : Whether to call `display(self.container)` immediately
            [default: True].
        fileNTguidisableFr`   r   c                  _   s   d S Nr    )r*   r+   r    r    r!   <lambda>       z(tqdm_notebook.__init__.<locals>.<lambda>z100%rC   r   r   )rV   )copygetsysstderrstdoutboolpopr#   r   __init__rc   dispZdynamic_ncolsrC   
unit_scalerA   rG   fprB   rF   r   r   rS   rR   r   r`   )r   argskwargsZ
file_kwargr`   Zdisplay_hererp   rA   r&   r    r!   rn      s2    	


ztqdm_notebook.__init__c                 #   sB   z"t t|  }|D ]
}|V  qW n   | jdd  Y nX d S )NrI   r;   )r#   r   __iter__ro   )r   itobjr&   r    r!   ru      s    ztqdm_notebook.__iter__r   c                    s4   zt t| j|dW S    | jdd  Y nX d S )N)rL   rI   rt   )r#   r   updatero   )r   rL   r&   r    r!   rx     s
    ztqdm_notebook.updatec                    sb   | j r
d S tt|   | jr:| j| jk r:| jddd n$| jrP| jddd n| jddd d S )NrI   F)r;   rV   rJ   T)rP   rV   )rc   r#   r   rP   rA   rL   ro   Zleaver_   r&   r    r!   rP     s    ztqdm_notebook.closec                 O   s   d S rd   r    )r   r*   r+   r    r    r!   clear&  s    ztqdm_notebook.clearc                    sd   | j rtt| j|dS | jj\}}}d|_|dk	rR||_| jsR| j	dkrRd|j
_tt| j|dS )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.
        )rA   rH   N)rc   r#   r   resetrF   r5   r;   r3   rA   rC   r<   r=   )r   rA   r*   r   r&   r    r!   rz   )  s    
ztqdm_notebook.reset)NNN)NNFNT)r   )N)r-   r.   r/   r0   staticmethodrG   r   propertyr`   setterrn   ru   rx   rP   ry   rz   r1   r    r    r&   r!   r   d   s$   ,      
3

-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   )rr   rs   r    r    r!   r   >  s    )(r0   
__future__r   r   rN   ri   weakrefr   Zstdr   Zstd_tqdmutilsr   ZIPYZ
ipywidgetsr8   warningscatch_warningsfilterwarningsZIPython.html.widgetshtmlZwidgetsr   r   r7   r   r   r   objectZIPython.displayr   r   cgi
__author____all__r9   r   r   r   r   r    r    r    r!   <module>   st   
 
 [	