a
    hbòfVŒ  ã                   @   s$  d dl mZmZ d dlmZ d dlmZ d dlmZm	Z	 d dl
mZ d dlmZ d dlmZmZmZ d dlmZ d d	lmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z" d
dl#m$Z$m%Z% d
dl&m'Z'm(Z(m)Z)m*Z* d
dl+m,Z, d
dl-m.Z. d
dl/m0Z0 d
dl1m2Z2 d
dl3m4Z4 d
dl5m6Z6 d
dl7m8Z8m9Z9 d
dl:m;Z;m<Z< ede=ƒZ>e!dƒZ?eg e@f ZAG dd„ deƒZBdie"ee? ee? f eCee@ eDee' eDeeg e@f  e@e6e6e6e6e@eDee? d#œd$d%„ZEG d&d'„ d'eƒZFG d(d)„ d)eFƒZGG d*d+„ d+eFƒZHG d,d-„ d-eFƒZIG d.d/„ d/eFƒZJG d0d1„ d1eFƒZKG d2d3„ d3eFƒZLG d4d5„ d5eFƒZMG d6d7„ d7eFƒZNG d8d9„ d9eFƒZOG d:d;„ d;eFƒZPG d<d=„ d=eƒZQeG d>d?„ d?ƒƒZRG d@dA„ dAe.ƒZSeTdBk�r d dlUZUd dlVZVd
dClWmXZX d
dDlYmZZZ d
dEl[m\Z\ d
dFl7m9Z9 e\dGdHddI�Z[e9dJdKdLƒZ7e7 ]dMdNdO¡ dPeXdQƒdRe7dSdTdUdVœdWe[eZdXƒg	Z^d dYl_m`Z` e`e^ƒZae'ddZ�Z&eSeHƒ eId[ƒeJƒ eId\ƒeLƒ eKƒ e&dd]��ŽZbebjcd^d_d`�Zdebjcdad_d`�Zeebjcdbd_ddc�Zfebjg�s ebjhedddde� ebjheedfde� eV idg¡ eU jd dh¡d
k �r®eb keleaƒ¡ �q®W d  ƒ n1 �s0    Y  dS )jé    )ÚABCÚabstractmethod)Údeque)ÚSized)Ú	dataclassÚfield)Ú	timedelta)Úceil)ÚEventÚRLockÚThread)ÚTracebackType)ÚAnyÚCallableÚDequeÚDictÚIterableÚListÚ
NamedTupleÚNewTypeÚOptionalÚSequenceÚTupleÚTypeÚTypeVarÚUnioné   )ÚfilesizeÚget_console)ÚConsoleÚJustifyMethodÚRenderableTypeÚGroup)ÚHighlighter)ÚJupyterMixin)ÚLive)ÚProgressBar)ÚSpinner)Ú	StyleType)ÚColumnÚTable)ÚTextÚTextTypeÚTaskIDÚProgressTypec                       sj   e Zd ZdZddedœ‡ fdd„Zddœd	d
„Zd dœdd„Zee	e
  ee
 ee ddœdd„Z‡  ZS )Ú_TrackThreadz)A thread to periodically update progress.ÚProgressr-   )ÚprogressÚtask_idÚupdate_periodc                    s.   || _ || _|| _tƒ | _d| _tƒ  ¡  d S )Nr   )r1   r2   r3   r
   ÚdoneÚ	completedÚsuperÚ__init__)Úselfr1   r2   r3   ©Ú	__class__© úV/home/httpd/docs/test/DocsMgr/lib/python3.9/site-packages/pip/_vendor/rich/progress.pyr7   /   s    z_TrackThread.__init__N©Úreturnc                 C   sd   | j }| jj}| j}d}| jj}||ƒsJ| j}||kr |||| ƒ |}q | jj| j | jdd� d S )Nr   T)r5   Úrefresh)r2   r1   Úadvancer3   r4   Úwaitr5   Úupdate)r8   r2   r@   r3   Úlast_completedrA   r5   r;   r;   r<   Úrun8   s    z_TrackThread.runc                 C   s   |   ¡  | S ©N©Ústart©r8   r;   r;   r<   Ú	__enter__F   s    z_TrackThread.__enter__©Úexc_typeÚexc_valÚexc_tbr>   c                 C   s   | j  ¡  |  ¡  d S rE   )r4   ÚsetÚjoin©r8   rK   rL   rM   r;   r;   r<   Ú__exit__J   s    
z_TrackThread.__exit__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úfloatr7   rD   rI   r   r   ÚBaseExceptionr   rQ   Ú__classcell__r;   r;   r9   r<   r/   ,   s   	
ûr/   ú
Working...NTFé
   úbar.backúbar.completeúbar.finishedú	bar.pulseçš™™™™™¹?)ÚsequenceÚdescriptionÚtotalÚauto_refreshÚconsoleÚ	transientÚget_timeÚrefresh_per_secondÚstyleÚcomplete_styleÚfinished_styleÚpulse_styler3   Údisabler>   c              	   c   s�   |rt dƒgng }| t||	|
|d�t dƒtƒ f¡ t||||||pFd|dœŽ}|�( |j| |||d�E dH  W d  ƒ n1 s‚0    Y  dS )a�  Track progress by iterating over a sequence.

    Args:
        sequence (Iterable[ProgressType]): A sequence (must support "len") you wish to iterate over.
        description (str, optional): Description of task show next to progress bar. Defaults to "Working".
        total: (float, optional): Total number of steps. Default is len(sequence).
        auto_refresh (bool, optional): Automatic refresh, disable to force a refresh after each iteration. Default is True.
        transient: (bool, optional): Clear the progress on exit. Defaults to False.
        console (Console, optional): Console to write to. Default creates internal Console instance.
        refresh_per_second (float): Number of times per second to refresh the progress information. Defaults to 10.
        style (StyleType, optional): Style for the bar background. Defaults to "bar.back".
        complete_style (StyleType, optional): Style for the completed bar. Defaults to "bar.complete".
        finished_style (StyleType, optional): Style for a finished bar. Defaults to "bar.done".
        pulse_style (StyleType, optional): Style for pulsing bars. Defaults to "bar.pulse".
        update_period (float, optional): Minimum time (in seconds) between calls to update(). Defaults to 0.1.
        disable (bool, optional): Disable display of progress.
    Returns:
        Iterable[ProgressType]: An iterable of the values in the sequence.

    ú([progress.description]{task.description})rh   ri   rj   rk   ú-[progress.percentage]{task.percentage:>3.0f}%rZ   )rc   rd   re   rf   rg   rl   )rb   ra   r3   N)Ú
TextColumnÚextendÚ	BarColumnÚTimeRemainingColumnr0   Útrack)r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   r3   rl   Úcolumnsr1   r;   r;   r<   rs   T   s4    &ÿüøÿù
ÿrs   c                   @   sj   e Zd ZU dZdZee ed< dee ddœdd„Z	edœdd	„Z
d
edœdd„Zed
edœdd„ƒZdS )ÚProgressColumnz3Base class for a widget to use in progress display.NÚmax_refresh)Útable_columnr>   c                 C   s   || _ i | _d | _d S rE   )Ú_table_columnÚ_renderable_cacheÚ_update_time)r8   rw   r;   r;   r<   r7   �   s    zProgressColumn.__init__r=   c                 C   s   | j p
tƒ S )z.Get a table column, used to build tasks table.)rx   r)   rH   r;   r;   r<   Úget_table_column¢   s    zProgressColumn.get_table_columnÚTask©Útaskr>   c                 C   sp   |  ¡ }| jdurR|jsRz| j|j \}}W n ty>   Y n0 || j |krR|S |  |¡}||f| j|j< |S )a  Called by the Progress object to return a renderable for the given task.

        Args:
            task (Task): An object containing information regarding the task.

        Returns:
            RenderableType: Anything renderable (including str).
        N)rf   rv   r5   ry   ÚidÚKeyErrorÚrender)r8   r~   Úcurrent_timeÚ	timestampÚ
renderabler;   r;   r<   Ú__call__¦   s    	
zProgressColumn.__call__c                 C   s   dS )z"Should return a renderable object.Nr;   ©r8   r~   r;   r;   r<   r�   ½   s    zProgressColumn.render)N)rR   rS   rT   rU   rv   r   rV   Ú__annotations__r)   r7   r{   r!   r…   r   r�   r;   r;   r;   r<   ru   ˜   s   
ru   c                       sD   e Zd ZdZdddœeee dœ‡ fdd„Zded	œd
d„Z‡  Z	S )ÚRenderableColumnz�A column to insert an arbitrary column.

    Args:
        renderable (RenderableType, optional): Any renderable. Defaults to empty string.
    Ú N©rw   )r„   rw   c                   s   || _ tƒ j|d� d S ©NrŠ   )r„   r6   r7   )r8   r„   rw   r9   r;   r<   r7   É   s    zRenderableColumn.__init__r|   r}   c                 C   s   | j S rE   )r„   r†   r;   r;   r<   r�   Ï   s    zRenderableColumn.render)r‰   )
rR   rS   rT   rU   r!   r   r)   r7   r�   rX   r;   r;   r9   r<   rˆ   Â   s    ÿÿÿrˆ   c                       sb   e Zd ZdZdeee eeee	 dœ‡ fdd	„Z
deee edd
œdd„Zdedœdd„Z‡  ZS )ÚSpinnerColumna‰  A column with a 'spinner' animation.

    Args:
        spinner_name (str, optional): Name of spinner animation. Defaults to "dots".
        style (StyleType, optional): Style of spinner. Defaults to "progress.spinner".
        speed (float, optional): Speed factor of spinner. Defaults to 1.0.
        finished_text (TextType, optional): Text used when task is finished. Defaults to " ".
    Údotsúprogress.spinnerç      ð?ú N)Úspinner_namerh   ÚspeedÚfinished_textrw   c                    s<   t |||d�| _t|tƒr$t |¡n|| _tƒ j|d� d S )N©rh   r’   rŠ   )	r'   ÚspinnerÚ
isinstanceÚstrr+   Úfrom_markupr“   r6   r7   )r8   r‘   rh   r’   r“   rw   r9   r;   r<   r7   Ý   s    ÿýzSpinnerColumn.__init__)r‘   Úspinner_styler’   r>   c                 C   s   t |||d�| _dS )a-  Set a new spinner.

        Args:
            spinner_name (str): Spinner name, see python -m rich.spinner.
            spinner_style (Optional[StyleType], optional): Spinner style. Defaults to "progress.spinner".
            speed (float, optional): Speed factor of spinner. Defaults to 1.0.
        r”   N)r'   r•   )r8   r‘   r™   r’   r;   r;   r<   Úset_spinnerí   s    zSpinnerColumn.set_spinnerr|   r}   c                 C   s    |j r| jn| j | ¡ ¡}|S rE   )Úfinishedr“   r•   r�   rf   )r8   r~   Útextr;   r;   r<   r�   ü   s
    ÿýzSpinnerColumn.render)r�   rŽ   r�   r�   N)rŽ   r�   )rR   rS   rT   rU   r—   r   r(   rV   r,   r)   r7   rš   r!   r�   rX   r;   r;   r9   r<   rŒ   Ó   s,        úú  üûrŒ   c                	       sL   e Zd ZdZdeeeeee	 ee
 ddœ‡ fdd„Zd	ed
œdd„Z‡  ZS )ro   zA column containing text.ÚnoneÚleftTN)Útext_formatrh   ÚjustifyÚmarkupÚhighlighterrw   r>   c                    s:   || _ || _|| _|| _|| _tƒ j|p0tdd�d� d S )NT©Úno_wraprŠ   )rŸ   r    rh   r¡   r¢   r6   r7   r)   )r8   rŸ   rh   r    r¡   r¢   rw   r9   r;   r<   r7     s    	zTextColumn.__init__r|   r}   c                 C   sR   | j j|d�}| jr*tj|| j| jd�}nt|| j| jd�}| jrN| j |¡ |S )N©r~   )rh   r    )	rŸ   Úformatr¡   r+   r˜   rh   r    r¢   Ú	highlight)r8   r~   Ú_textrœ   r;   r;   r<   r�     s    zTextColumn.render)r�   rž   TNN)rR   rS   rT   rU   r—   r(   r    Úboolr   r#   r)   r7   r+   r�   rX   r;   r;   r9   r<   ro     s         ùøro   c                	       sL   e Zd ZdZdee eeeeee ddœ‡ fd	d
„Zde	dœdd„Z
‡  ZS )rq   a  Renders a visual progress bar.

    Args:
        bar_width (Optional[int], optional): Width of bar or None for full width. Defaults to 40.
        style (StyleType, optional): Style for the bar background. Defaults to "bar.back".
        complete_style (StyleType, optional): Style for the completed bar. Defaults to "bar.complete".
        finished_style (StyleType, optional): Style for a finished bar. Defaults to "bar.done".
        pulse_style (StyleType, optional): Style for pulsing bars. Defaults to "bar.pulse".
    é(   r[   r\   r]   r^   N)Ú	bar_widthrh   ri   rj   rk   rw   r>   c                    s0   || _ || _|| _|| _|| _tƒ j|d� d S r‹   )r«   rh   ri   rj   rk   r6   r7   )r8   r«   rh   ri   rj   rk   rw   r9   r;   r<   r7   .  s    	zBarColumn.__init__r|   r}   c                 C   sP   t td|jƒtd|jƒ| jdu r$dn
td| jƒ|j | ¡ | j| j| j	| j
d�	S )z&Gets a progress bar widget for a task.r   Nr   )	rb   r5   ÚwidthÚpulseÚanimation_timerh   ri   rj   rk   )r&   Úmaxrb   r5   r«   Ústartedrf   rh   ri   rj   rk   r†   r;   r;   r<   r�   >  s    

÷zBarColumn.render)rª   r[   r\   r]   r^   N)rR   rS   rT   rU   r   Úintr(   r)   r7   r&   r�   rX   r;   r;   r9   r<   rq   #  s"         ùørq   c                   @   s    e Zd ZdZdedœdd„ZdS )ÚTimeElapsedColumnzRenders time elapsed.r|   r}   c                 C   sD   |j r|jn|j}|du r&tddd�S tt|ƒd�}tt|ƒdd�S )úShow time remaining.Nú-:--:--zprogress.elapsed©rh   ©Úseconds)r›   Úfinished_timeÚelapsedr+   r   r±   r—   )r8   r~   r¹   Údeltar;   r;   r<   r�   P  s
    zTimeElapsedColumn.renderN©rR   rS   rT   rU   r+   r�   r;   r;   r;   r<   r²   M  s   r²   c                   @   s$   e Zd ZdZdZdedœdd„ZdS )rr   z!Renders estimated time remaining.ç      à?r|   r}   c                 C   s8   |j }|du rtddd�S tt|ƒd�}tt|ƒdd�S )r³   Nr´   zprogress.remainingrµ   r¶   )Útime_remainingr+   r   r±   r—   )r8   r~   Ú	remainingÚremaining_deltar;   r;   r<   r�   _  s
    zTimeRemainingColumn.renderN)rR   rS   rT   rU   rv   r+   r�   r;   r;   r;   r<   rr   Y  s   rr   c                   @   s    e Zd ZdZdedœdd„ZdS )ÚFileSizeColumnzRenders completed filesize.r|   r}   c                 C   s   t  t|jƒ¡}t|dd�S )úShow data completed.zprogress.filesizerµ   )r   Údecimalr±   r5   r+   ©r8   r~   Ú	data_sizer;   r;   r<   r�   k  s    zFileSizeColumn.renderNr»   r;   r;   r;   r<   rÀ   h  s   rÀ   c                   @   s    e Zd ZdZdedœdd„ZdS )ÚTotalFileSizeColumnzRenders total filesize.r|   r}   c                 C   s   t  t|jƒ¡}t|dd�S )rÁ   zprogress.filesize.totalrµ   )r   rÂ   r±   rb   r+   rÃ   r;   r;   r<   r�   t  s    zTotalFileSizeColumn.renderNr»   r;   r;   r;   r<   rÅ   q  s   rÅ   c                       s@   e Zd ZdZdeee ddœ‡ fdd„Zdedœd	d
„Z	‡  Z
S )ÚDownloadColumnz¤Renders file size downloaded and total, e.g. '0.5/2.3 GB'.

    Args:
        binary_units (bool, optional): Use binary units, KiB, MiB etc. Defaults to False.
    FN)Úbinary_unitsrw   r>   c                    s   || _ tƒ j|d� d S r‹   )rÇ   r6   r7   )r8   rÇ   rw   r9   r;   r<   r7   �  s    zDownloadColumn.__init__r|   r}   c                 C   s¬   t |jƒ}t |jƒ}| jr2t |g d¢d¡\}}nt |g d¢d¡\}}|| }|| }|dkrddnd}|d|› d�›}	|d|› d�›}
|	› d	|
› d
|› �}t|dd�}|S )z.Calculate common unit for completed and total.)	ÚbytesÚKiBÚMiBÚGiBÚTiBÚPiBÚEiBÚZiBÚYiBi   )	rÈ   ÚKBÚMBÚGBÚTBÚPBÚEBÚZBÚYBéè  r   r   z,.Úfú/r�   zprogress.downloadrµ   )r±   r5   rb   rÇ   r   Úpick_unit_and_suffixr+   )r8   r~   r5   rb   ÚunitÚsuffixÚcompleted_ratioÚtotal_ratioÚ	precisionÚcompleted_strÚ	total_strÚdownload_statusÚdownload_textr;   r;   r<   r�   ‡  s&    

ý

ÿzDownloadColumn.render)FN)rR   rS   rT   rU   r©   r   r)   r7   r+   r�   rX   r;   r;   r9   r<   rÆ   z  s    ÿþrÆ   c                   @   s    e Zd ZdZdedœdd„ZdS )ÚTransferSpeedColumnz&Renders human readable transfer speed.r|   r}   c                 C   s@   |j p
|j}|du r tddd�S t t|ƒ¡}t|› d�dd�S )zShow data transfer speed.Nú?zprogress.data.speedrµ   z/s)Úfinished_speedr’   r+   r   rÂ   r±   )r8   r~   r’   Ú
data_speedr;   r;   r<   r�   ¢  s
    zTransferSpeedColumn.renderNr»   r;   r;   r;   r<   ræ   Ÿ  s   ræ   c                   @   s"   e Zd ZU dZeed< eed< dS )ÚProgressSamplez$Sample of progress for a given time.rƒ   r5   N)rR   rS   rT   rU   rV   r‡   r;   r;   r;   r<   rê   «  s   
rê   c                   @   sˆ  e Zd ZU dZeed< eed< eed< eed< eed< dZ	e
e ed< d	Zeed
< eed�Zeeef ed< edddd�Ze
e ed< edddd�Ze
e ed< dZe
e ed< eeddd�Zee ed< eded�Zeed< edœdd„Zeedœdd„ƒZeedœdd„ƒZee
e dœdd„ƒZeedœdd „ƒZ eedœd!d"„ƒZ!ee
e dœd#d$„ƒZ"ee
e dœd%d&„ƒZ#ddœd'd(„Z$dS ))r|   z…Information regarding a progress task.

    This object should be considered read-only outside of the :class:`~Progress` class.

    r   ra   rb   r5   Ú	_get_timeNr¸   TÚvisible)Údefault_factoryÚfieldsF)ÚdefaultÚinitÚreprÚ
start_timeÚ	stop_timerè   )rí   rð   rñ   Ú	_progress)rñ   rí   Ú_lockr=   c                 C   s   |   ¡ S )z(float: Get the current time, in seconds.)rë   rH   r;   r;   r<   rf   ä  s    zTask.get_timec                 C   s
   | j duS )z#bool: Check if the task as started.N)rò   rH   r;   r;   r<   r°   è  s    zTask.startedc                 C   s   | j | j S )z)float: Get the number of steps remaining.)rb   r5   rH   r;   r;   r<   r¾   í  s    zTask.remainingc                 C   s2   | j du rdS | jdur$| j| j  S |  ¡ | j  S )z]Optional[float]: Time elapsed since task was started, or ``None`` if the task hasn't started.N)rò   ró   rf   rH   r;   r;   r<   r¹   ò  s
    

zTask.elapsedc                 C   s
   | j duS )zCheck if the task has finished.N)r¸   rH   r;   r;   r<   r›   û  s    zTask.finishedc                 C   s.   | j s
dS | j| j  d }tdtd|ƒƒ}|S )z,float: Get progress of task as a percentage.ç        ç      Y@)rb   r5   Úminr¯   )r8   r5   r;   r;   r<   Ú
percentage   s
    zTask.percentagec                 C   s¬   | j du rdS | j�„ | j}|s0W d  ƒ dS |d j|d j }|dkr\W d  ƒ dS t|ƒ}t|ƒ tdd„ |D ƒƒ}|| }|W  d  ƒ S 1 sž0    Y  dS )z=Optional[float]: Get the estimated speed in steps per second.Néÿÿÿÿr   c                 s   s   | ]}|j V  qd S rE   )r5   )Ú.0Úsampler;   r;   r<   Ú	<genexpr>  ó    zTask.speed.<locals>.<genexpr>)rò   rõ   rô   rƒ   ÚiterÚnextÚsum)r8   r1   Ú
total_timeÚiter_progressÚtotal_completedr’   r;   r;   r<   r’   	  s    
z
Task.speedc                 C   s*   | j r
dS | j}|sdS t| j| ƒ}|S )zJOptional[float]: Get estimated time to completion, or ``None`` if no data.rö   N)r›   r’   r	   r¾   )r8   r’   Úestimater;   r;   r<   r½     s    zTask.time_remainingc                 C   s   | j  ¡  d| _d| _dS )zReset progress.N)rô   Úclearr¸   rè   rH   r;   r;   r<   Ú_reset&  s    
zTask._reset)%rR   rS   rT   rU   r-   r‡   r—   rV   ÚGetTimeCallabler¸   r   rì   r©   r   Údictrî   r   r   rò   ró   rè   r   rô   r   rê   r   rõ   rf   Úpropertyr°   r¾   r¹   r›   rù   r’   r½   r  r;   r;   r;   r<   r|   ´  s@   
ÿ
r|   c                   @   sX  e Zd ZdZdddddddddddœ
eeef ee e	e
e
e	e	e	ee e	e	ddœd	d
„Zeedœdd„ƒZeee dœdd„ƒZeee dœdd„ƒZee	dœdd„ƒZddœdd„Zddœdd„Zd dœdd„Zeee  ee ee ddœdd„ZdEeee ee f ee
 ee ee
ee dœd d!„Zedd"œd#d$„Z edd"œd%d&„Z!ddddddd'œeee
 ee
 ee
 ee ee	 e	e"dd(œ	d)d*„Z#ddd+ddd,œee	ee
 e$ee	 ee e"dd-œd.d/„Z%dFee
dd1œd2d3„Z&ddœd4d5„Z'e(dœd6d7„Z)ee( dœd8d9„Z*ee e+d:œd;d<„Z,e(dœd=d>„Z-dGee	e
e$e	e"ed@œdAdB„Z.edd"œdCdD„Z/dS )Hr0   a�  Renders an auto-updating progress bar(s).

    Args:
        console (Console, optional): Optional Console instance. Default will an internal Console instance writing to stdout.
        auto_refresh (bool, optional): Enable auto refresh. If disabled, you will need to call `refresh()`.
        refresh_per_second (Optional[float], optional): Number of times per second to refresh the progress information or None to use default (10). Defaults to None.
        speed_estimate_period: (float, optional): Period (in seconds) used to calculate the speed estimate. Defaults to 30.
        transient: (bool, optional): Clear the progress on exit. Defaults to False.
        redirect_stdout: (bool, optional): Enable redirection of stdout, so ``print`` may be used. Defaults to True.
        redirect_stderr: (bool, optional): Enable redirection of stderr. Defaults to True.
        get_time: (Callable, optional): A callable that gets the current time, or None to use Console.get_time. Defaults to None.
        disable (bool, optional): Disable progress display. Defaults to False
        expand (bool, optional): Expand tasks table to fit width. Defaults to False.
    NTrZ   g      >@F)
rd   rc   rg   Úspeed_estimate_periodre   Úredirect_stdoutÚredirect_stderrrf   rl   Úexpand)rt   rd   rc   rg   r  re   r  r  rf   rl   r  r>   c       
      	   G   s¦   |d u s|dksJ dƒ‚t ƒ | _|p:tdƒtƒ tdƒtƒ f| _|| _|	| _|
| _i | _	t
dƒ| _t|pjtƒ |||||| jd�| _|pŠ| jj| _| jj| _| jj| _d S )Nr   zrefresh_per_second must be > 0rm   rn   )rd   rc   rg   re   r  r  Úget_renderable)r   rõ   ro   rq   rr   rt   r  rl   r  Ú_tasksr-   Ú_task_indexr%   r   r  Úliverd   rf   ÚprintÚlog)r8   rd   rc   rg   r  re   r  r  rf   rl   r  rt   r;   r;   r<   r7   =  s<    ÿÿþü
ù	
zProgress.__init__r=   c                 C   s   | j jS rE   )r  rd   rH   r;   r;   r<   rd   h  s    zProgress.consolec                 C   s8   | j � t| j ¡ ƒW  d  ƒ S 1 s*0    Y  dS )zGet a list of Task instances.N)rõ   Úlistr  ÚvaluesrH   r;   r;   r<   Útasksl  s    zProgress.tasksc                 C   s8   | j � t| j ¡ ƒW  d  ƒ S 1 s*0    Y  dS )zA list of task IDs.N)rõ   r  r  ÚkeysrH   r;   r;   r<   Útask_idsr  s    zProgress.task_idsc                 C   sX   | j �> | jsW d  ƒ dS tdd„ | j ¡ D ƒƒW  d  ƒ S 1 sJ0    Y  dS )z'Check if all tasks have been completed.NTc                 s   s   | ]}|j V  qd S rE   )r›   )rû   r~   r;   r;   r<   rý   ~  rþ   z$Progress.finished.<locals>.<genexpr>)rõ   r  Úallr  rH   r;   r;   r<   r›   x  s    zProgress.finishedc                 C   s   | j s| jjdd� dS )zStart the progress display.T)r?   N)rl   r  rG   rH   r;   r;   r<   rG   €  s    zProgress.startc                 C   s    | j  ¡  | jjs| j ¡  dS )zStop the progress display.N)r  Ústoprd   Úis_interactiver  rH   r;   r;   r<   r  …  s    
zProgress.stopc                 C   s   |   ¡  | S rE   rF   rH   r;   r;   r<   rI   ‹  s    zProgress.__enter__rJ   c                 C   s   |   ¡  d S rE   )r  rP   r;   r;   r<   rQ   �  s    zProgress.__exit__rY   r_   )r`   rb   r2   ra   r3   r>   c                 c   sà   |du r2t |tƒr tt|ƒƒ}q6td|›d�ƒ‚n|}|du rN| j||d�}n| j||d� | jjr°t	| ||ƒ�.}|D ]}|V  | j
d7  _
qvW d  ƒ qÜ1 s¤0    Y  n,| j}	| j}
|D ]}|V  |	|dƒ |
ƒ  qÀdS )a‡  Track progress by iterating over a sequence.

        Args:
            sequence (Sequence[ProgressType]): A sequence of values you want to iterate over and track progress.
            total: (float, optional): Total number of steps. Default is len(sequence).
            task_id: (TaskID): Task to track. Default is new task.
            description: (str, optional): Description of task, if new task is created.
            update_period (float, optional): Minimum time (in seconds) between calls to update(). Defaults to 0.1.

        Returns:
            Iterable[ProgressType]: An iterable of values taken from the provided sequence.
        Nzunable to get size of z, please specify 'total'©rb   r   )r–   r   rV   ÚlenÚ
ValueErrorÚadd_taskrB   r  rc   r/   r5   r@   r?   )r8   r`   rb   r2   ra   r3   Ú
task_totalÚtrack_threadÚvaluer@   r?   r;   r;   r<   rs   —  s*    

ÿ0
zProgress.track)r2   r>   c                 C   sH   | j �. | j| }|jdu r&|  ¡ |_W d  ƒ n1 s:0    Y  dS )zéStart a task.

        Starts a task (used when calculating elapsed time). You may need to call this manually,
        if you called ``add_task`` with ``start=False``.

        Args:
            task_id (TaskID): ID of task.
        N)rõ   r  rò   rf   )r8   r2   r~   r;   r;   r<   Ú
start_taskÈ  s    	

zProgress.start_taskc                 C   sR   | j �8 | j| }|  ¡ }|jdu r*||_||_W d  ƒ n1 sD0    Y  dS )z†Stop a task.

        This will freeze the elapsed time on the task.

        Args:
            task_id (TaskID): ID of task.
        N)rõ   r  rf   rò   ró   )r8   r2   r~   r‚   r;   r;   r<   Ú	stop_taskÖ  s    

zProgress.stop_task)rb   r5   r@   ra   rì   r?   )	r2   rb   r5   r@   ra   rì   r?   rî   r>   c                K   sH  | j �� | j| }	|	j}
|dur:||	jkr:||	_|	 ¡  |durP|	 j|7  _|dur^||	_|durl||	_|durz||	_|	j |¡ |	j|
 }|  	¡ }|| j
 }|	j}|j}|rÈ|d j|k rÈ|ƒ  q®t|ƒdkrÜ|ƒ  qÈ|dkrô| t||ƒ¡ |	j|	jk�r|	jdu �r|	j|	_W d  ƒ n1 �s,0    Y  |�rD|  ¡  dS )a˜  Update information associated with a task.

        Args:
            task_id (TaskID): Task id (returned by add_task).
            total (float, optional): Updates task.total if not None.
            completed (float, optional): Updates task.completed if not None.
            advance (float, optional): Add a value to task.completed if not None.
            description (str, optional): Change task description if not None.
            visible (bool, optional): Set visible flag if not None.
            refresh (bool): Force a refresh of progress information. Default is False.
            **fields (Any): Additional data fields required for rendering.
        Nr   rÙ   )rõ   r  r5   rb   r  ra   rì   rî   rB   rf   r  rô   Úpopleftrƒ   r  Úappendrê   r¸   r¹   r?   )r8   r2   rb   r5   r@   ra   rì   r?   rî   r~   Úcompleted_startÚupdate_completedr‚   Úold_sample_timerô   r&  r;   r;   r<   rB   å  s<    



(zProgress.updater   )rG   rb   r5   rì   ra   )r2   rG   rb   r5   rì   ra   rî   r>   c          
      K   sš   |   ¡ }| j�p | j| }	|	 ¡  |r*|nd|	_|dur>||	_||	_|durR||	_|r\||	_|durj||	_	d|	_
W d  ƒ n1 s„0    Y  |  ¡  dS )aÔ  Reset a task so completed is 0 and the clock is reset.

        Args:
            task_id (TaskID): ID of task.
            start (bool, optional): Start the task after reset. Defaults to True.
            total (float, optional): New total steps in task, or None to use current total. Defaults to None.
            completed (int, optional): Number of steps completed. Defaults to 0.
            **fields (str): Additional data fields required for rendering.
        N)rf   rõ   r  r  rò   rb   r5   rì   rî   ra   r¸   r?   )
r8   r2   rG   rb   r5   rì   ra   rî   r‚   r~   r;   r;   r<   Úreset   s     
$zProgress.resetr   )r2   r@   r>   c           
      C   sÔ   |   ¡ }| j�² | j| }|j}| j|7  _|j| }|| j }|j}|j}	|rh|d j|k rh|	ƒ  qNt|ƒdkr||	ƒ  qh| 	t
||ƒ¡ |j|jkr²|jdu r²|j|_|j|_W d  ƒ n1 sÆ0    Y  dS )z«Advance task by a number of steps.

        Args:
            task_id (TaskID): ID of task.
            advance (float): Number of steps to advance. Default is 1.
        r   rÙ   N)rf   rõ   r  r5   r  rô   r&  rƒ   r  r'  rê   rb   r¸   r¹   r’   rè   )
r8   r2   r@   r‚   r~   r(  r)  r*  rô   r&  r;   r;   r<   r@   E  s"    


zProgress.advancec                 C   s   | j s| jjr| j ¡  dS )z*Refresh (render) the progress information.N)rl   r  Ú
is_startedr?   rH   r;   r;   r<   r?   _  s    zProgress.refreshc                 C   s   t |  ¡ Ž }|S )z*Get a renderable for the progress display.)r"   Úget_renderables)r8   r„   r;   r;   r<   r  d  s    zProgress.get_renderablec                 c   s   |   | j¡}|V  dS )z5Get a number of renderables for the progress display.N)Úmake_tasks_tabler  )r8   Útabler;   r;   r<   r-  i  s    zProgress.get_renderables)r  r>   c                    sR   dd„ | j D ƒ}tj|d| jdœŽ}|D ]$‰ ˆ jr(|j‡ fdd„| j D ƒŽ  q(|S )z×Get a table to render the Progress display.

        Args:
            tasks (Iterable[Task]): An iterable of Task instances, one per row of the table.

        Returns:
            Table: A table instance.
        c                 s   s.   | ]&}t |tƒrtd d�n
| ¡  ¡ V  qdS )Tr£   N)r–   r—   r)   r{   Úcopy)rû   Ú_columnr;   r;   r<   rý   w  s   ýÿz,Progress.make_tasks_table.<locals>.<genexpr>)r   r   )Úpaddingr  c                 3   s,   | ]$}t |tƒr|jˆ d �n|ˆ ƒV  qdS )r¥   N)r–   r—   r¦   )rû   Úcolumnr¥   r;   r<   rý   „  s   ýÿ)rt   r*   Úgridr  rì   Úadd_row)r8   r  Útable_columnsr/  r;   r¥   r<   r.  n  s    	ú
úÿ
zProgress.make_tasks_tablec                 C   s2   | j � |  ¡ W  d  ƒ S 1 s$0    Y  dS )z+Makes the Progress class itself renderable.N)rõ   r  rH   r;   r;   r<   Ú__rich__�  s    zProgress.__rich__r÷   )ra   rG   rb   r5   rì   rî   r>   c           	      K   s†   | j �d t| j|||||| j| j d�}|| j| j< |rB|  | j¡ | j}tt| jƒd ƒ| _W d  ƒ n1 sp0    Y  |  ¡  |S )aÞ  Add a new 'task' to the Progress display.

        Args:
            description (str): A description of the task.
            start (bool, optional): Start the task immediately (to calculate elapsed time). If set to False,
                you will need to call `start` manually. Defaults to True.
            total (float, optional): Number of total steps in the progress if know. Defaults to 100.
            completed (int, optional): Number of steps completed so far.. Defaults to 0.
            visible (bool, optional): Enable display of the task. Defaults to True.
            **fields (str): Additional data fields required for rendering.

        Returns:
            TaskID: An ID you can use when calling `update`.
        )rì   rî   rë   rõ   r   N)	rõ   r|   r  rf   r  r$  r-   r±   r?   )	r8   ra   rG   rb   r5   rì   rî   r~   Únew_task_indexr;   r;   r<   r   ”  s$    ø
2zProgress.add_taskc                 C   s2   | j � | j|= W d  ƒ n1 s$0    Y  dS )z]Delete a task if it exists.

        Args:
            task_id (TaskID): A task ID.

        N)rõ   r  )r8   r2   r;   r;   r<   Úremove_task¾  s    zProgress.remove_task)NNrY   r_   )r   )Tr÷   r   T)0rR   rS   rT   rU   r   r—   ru   r   r   r©   rV   r  r7   r
  rd   r   r|   r  r-   r  r›   rG   r  rI   r   rW   r   rQ   r   r.   r   rs   r$  r%  r   rB   r±   r+  r@   r?   r!   r  r-  r*   r.  r7  r   r9  r;   r;   r;   r<   r0   -  sÖ   ô
ó+
û    úù1÷õ?øö%!    úø*r0   Ú__main__)ÚPanel)ÚRule)ÚSyntax)r*   a~  def loop_last(values: Iterable[T]) -> Iterable[Tuple[bool, T]]:
    """Iterate and generate a tuple with a flag for last value."""
    iter_values = iter(values)
    try:
        previous_value = next(iter_values)
    except StopIteration:
        return
    for value in iter_values:
        yield False, previous_value
        previous_value = value
    yield True, previous_valueÚpython)Úline_numbersZfooÚbarZbazÚ1Ú2Ú3z:Text may be printed while the progress bars are rendering.z(In fact, [i]any[/i] renderable will workzSuch as [magenta]tables[/]...zPretty printed structures...ZexamplezPretty printed)Útyperœ   z	Syntax...zGive it a try!)Úcycle)Úrecordrm   rn   )rd   re   z[red]DownloadingrÙ   r  z[green]Processingz[yellow]Thinking)rb   rG   r¼   )r@   g333333Ó?g{®Gáz„?éd   )rY   NTNFNrZ   r[   r\   r]   r^   r_   F)mÚabcr   r   Úcollectionsr   Úcollections.abcr   Údataclassesr   r   Údatetimer   Úmathr	   Ú	threadingr
   r   r   Útypesr   Útypingr   r   r   r   r   r   r   r   r   r   r   r   r   r   r‰   r   r   rd   r   r    r!   r"   r¢   r#   Újupyterr$   r  r%   Úprogress_barr&   r•   r'   rh   r(   r/  r)   r*   rœ   r+   r,   r±   r-   r.   rV   r  r/   r—   r©   rs   ru   rˆ   rŒ   ro   rq   r²   rr   rÀ   rÅ   rÆ   ræ   rê   r|   r0   rR   ÚrandomÚtimeÚpanelr;  Úruler<  Úsyntaxr=  r5  Zprogress_renderablesÚ	itertoolsrE  Zexamplesr1   r   Ztask1Ztask2Ztask3r›   rB   ÚsleepÚrandintr  r   r;   r;   r;   r<   Ú<module>   sê   @
*             òñD*2*		%	x   
ó÷
ø	
