File: //usr/local/lib/python3.7/site-packages/pip/_vendor/rich/__pycache__/progress.cpython-37.pyc
B
L��g:� � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ d dlmZ d dl m
Z
m Z d dl m
Z
d dl mZmZ d dlmZ d dlmZ d d lmZ d d
lmZmZ d d lmZmZmZ d d l m Z d d
lm 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+m,Z,m-Z-m.Z.m/Z/ ej0dk�r&d dlm1Z1 n d dl2m1Z1 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8m9Z9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@ ddlAmBZB ddlCmDZD ddlEmFZF ddlGmHZHmIZI ddlJmKZKmLZL e(deM�ZNe.d �ZOe g ePf ZQe�.d e+e �ZRG d d � d e�ZSd�e/e*eO e%eO f eTe)eP eUe)e7 eUe)e g ePf ePeFeFeFeFePeUeUe%eO d)�d*d+�ZVG d,d-� d-ee �ZWG d.d/� d/e!eR e$eR �ZXd0d!dd"dd#d$d%d&d'd"d1� e eMeTeUe)e7 eUe)e g ePf ePeFeFeFeFeUe!e d2�d3d4�ZYejZd�dd0d!dd"dd#d$d%d&d'd"d6� e/eTd7e[f e/e1d8 e1d9 f eMe)eT e)eT e)eT e)eM eTeUe)e7 eUe)e g ePf ePeFeFeFeFeUe!e+ d:�d;d<��Z\ejZd�dd0d!dd"dd#d$d%d&d'd"d6� e/eTd7e[f e1d= eMe)eT e)eT e)eT e)eM eTeUe)e7 eUe)e g ePf ePeFeFeFeFeUe!e d:�d>d<��Z\d�dd0d!dd"dd#d$d%d&d'd"d6� e/eTd7e[f e/e1d= e1d8 e1d9 f eMe)eT e)eT e)eT e)eM eTeUe)e7 eUe)e g ePf ePeFeFeFeFeUe/e!e e!e+ f d:�d?d<�Z\G d@dA� dAe�Z]G dBdC� dCe]�Z^G dDdE� dEe]�Z_G dFdG� dGe]�Z`G dHdI� dIe]�ZaG dJdK� dKe]�ZbG dLdM� dMe`�ZcG dNdO� dOe]�ZdG dPdQ� dQe]�ZeG dRdS� dSe]�ZfG dTdU� dUe]�ZgG dVdW� dWe]�ZhG dXdY� dYe]�ZiG dZd[� d[e'�Zje
G d\d]� d]��ZkG d^d_� d_e>�Zlemd`k�r�d dlnZnd dloZoddalpmqZq ddblrmsZs ddcltmuZu dddlGmIZI eudedfd!dg�ZteIdhdidj�ZGeG�vdkdldm� dneqdo�dpeGdqdrdsdt�duetesdv�g Zwd dwlxmyZy eyew�Zze7d!dx�Z6ele_� fel�{� eb� f�e6d"dy����Z|e|j}dzd{d|�Z~e|j}d}d{d|�Ze|j}d~dd|�Z�xTe|j��s�e|j�e~dd�� e|j�ed�d�� eo��d�� en��d d��dk �r�e|��e�ez�� �q�W W dQ R X dS )�� N)�ABC�abstractmethod)�deque)� dataclass�field)� timedelta)� RawIOBase�UnsupportedOperation)�ceil)�mmap)� length_hint)�PathLike�stat)�Event�RLock�Thread)�
TracebackType)�Any�BinaryIO�Callable�ContextManager�Deque�Dict�Generic�Iterable�List�
NamedTuple�NewType�Optional�Sequence�TextIO�Tuple�Type�TypeVar�Union)� � )�Literal� )�filesize� get_console)�Console�Group�
JustifyMethod�RenderableType)� Highlighter)� JupyterMixin)�Live)� ProgressBar)�Spinner)� StyleType)�Column�Table)�Text�TextType�TaskID� ProgressType�_Ic sj e Zd ZdZdded�� fdd� Zdd�d d
�Zd d�d d �Zee e
ee
ee dd
�dd�Z � Z
S )� _TrackThreadz)A thread to periodically update progress.�Progressr9 )�progress�task_id�
update_periodc s. || _ || _|| _t� | _d| _t� �� d S )Nr )r> r? r@ r �done� completed�super�__init__)�selfr> r? r@ )� __class__� �=/tmp/pip-unpacked-wheel-hv55ucu3/pip/_vendor/rich/progress.pyrD B s z_TrackThread.__init__N)�returnc C sh | j }| jj}| j}d}| jj}x,||�sL| j}||kr"|||| � |}q"W | jj| j | jdd� d S )Nr T)rB �refresh)r? r> �advancer@ rA �waitrB �update)rE r? rK r@ Zlast_completedrL rB rG rG rH �runK s
z_TrackThread.runc C s | � � | S )N)�start)rE rG rG rH � __enter__Y s z_TrackThread.__enter__)�exc_type�exc_val�exc_tbrI c C s | j �� | �� d S )N)rA �set�join)rE rQ rR rS rG rG rH �__exit__] s
z_TrackThread.__exit__)�__name__�
__module__� __qualname__�__doc__�floatrD rN rP r r" �
BaseExceptionr rV �
__classcell__rG rG )rF rH r< ? s
r< �
Working...TF�
�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_styler@ �disable�
show_speedrI c c s� |rt d�gng }|�t|| |
| d�t|d�tdd�f� t||||||pLd|
d��}|� |j| ||| d�E d H W d Q R X 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.finished".
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.
show_speed (bool, optional): Show speed if total isn't known. Defaults to True.
Returns:
Iterable[ProgressType]: An iterable of the values in the sequence.
z([progress.description]{task.description})rm rn ro rp )rr T)�elapsed_when_finishedr_ )rh ri rj rk rl rq )rg rf r@ N)�
TextColumn�extend� BarColumn�TaskProgressColumn�TimeRemainingColumnr= �track)re rf rg rh ri rj rk rl rm rn ro rp r@ rq rr �columnsr> rG rG rH ry g s( (
ry c @ s� e Zd ZdZd9edeedd�dd�Zd d�d d
�Ze e
e e e e e dd �d d
�Z
ed�dd�Zed�dd�Zeed�dd��Zed�dd�Zed�dd�Zeed�dd��Zeed�dd��Zed�d d �Zed�d d �Zed�d d!�Zd:eed#�d$d%�Z e e e e f d&�d'd(�Z!d;eed#�d)d*�Z"d<ee#e d+�d,d-�Z$dd�d.d/�Z%d=eeed1�d2d3�Z&ed�d4d5�Z'e(ed6�d7d8�Z)dS )>�_Readerz9A reader that tracks progress while it's being read from.Tr= N)�handler> �task� close_handlerI c C s" || _ || _|| _|| _d| _d S )NF)r| r> r} r~ �_closed)rE r| r> r} r~ rG rG rH rD � s
z_Reader.__init__)rI c C s | j �� | S )N)r| rP )rE rG rG rH rP � s
z_Reader.__enter__)rQ rR rS rI c C s | � � d S )N)�close)rE rQ rR rS rG rG rH rV � s z_Reader.__exit__c C s | S )NrG )rE rG rG rH �__iter__� s z_Reader.__iter__c C s$ t | j�}| jj| jt|�d� |S )N)rK )�nextr| r> rK r} �len)rE �linerG rG rH �__next__� s
z_Reader.__next__c C s | j S )N)r )rE rG rG rH �closed� s z_Reader.closedc C s
| j �� S )N)r| �fileno)rE rG rG rH r� � s z_Reader.filenoc C s
| j �� S )N)r| �isatty)rE rG rG rH r� � s z_Reader.isattyc C s | j jS )N)r| �mode)rE rG rG rH r� � s z _Reader.modec C s | j jS )N)r| �name)rE rG rG rH r� � s z _Reader.namec C s
| j �� S )N)r| �readable)rE rG rG rH r� � s z_Reader.readablec C s
| j �� S )N)r| �seekable)rE rG rG rH r� � s z_Reader.seekablec C s dS )NFrG )rE rG rG rH �writable� s z_Reader.writable���)�sizerI c C s& | j �|�}| jj| jt|�d� |S )N)rK )r| �readr> rK r} r� )rE r� �blockrG rG rH r� � s z _Reader.read)�bc C s" | j �|�}| jj| j|d� |S )N)rK )r| �readintor> rK r} )rE r� �nrG rG rH r� � s z_Reader.readintoc C s& | j �|�}| jj| jt|�d� |S )N)rK )r| �readliner> rK r} r� )rE r� r� rG rG rH r� � s z_Reader.readline)�hintrI c C s, | j �|�}| jj| jttt|��d� |S )N)rK )r| � readlinesr> rK r} �sum�mapr� )rE r� �linesrG rG rH r� � s z_Reader.readlinesc C s | j r| j�� d| _d S )NT)r~ r| r� r )rE rG rG rH r� s
z
_Reader.closer )�offset�whencerI c C s$ | j �||�}| jj| j|d� |S )N)rB )r| �seekr> rM r} )rE r� r� �posrG rG rH r� s z _Reader.seekc C s
| j �� S )N)r| �tell)rE rG rG rH r�
s z _Reader.tell)�srI c C s t d��d S )N�write)r )rE r� rG rG rH r�
s z
_Reader.write)T)r� )r� )r� )r )*rW rX rY rZ r r9 �boolrD rP r r" r\ r rV r� �bytesr� �propertyr� �intr� r� �strr� r� r� r� r� r� r$ � bytearray�
memoryviewr r� r� r r� r� r� r� r r� rG rG rG rH r{ � s:
r{ c @ sT e Zd ZdZdedd�dd�Zed�dd �Zeee ee ee
dd
�d d �Z dS )
� _ReadContextzEA utility class to handle a context for both a reader and a progress.r= N)r> �readerrI c C s || _ || _d S )N)r> r� )rE r> r� rG rG rH rD s z_ReadContext.__init__)rI c C s | j �� | j�� S )N)r> rO r� rP )rE rG rG rH rP s
z_ReadContext.__enter__)rQ rR rS rI c C s | j �� | j�|||� d S )N)r> �stopr� rV )rE rQ rR rS rG rG rH rV s
z_ReadContext.__exit__) rW rX rY rZ r; rD rP r r" r\ r rV rG rG rG rH r� s
r� z
Reading...) rf rh ri rj rk rl rm rn ro rp rq )�filerg rf rh ri rj rk rl rm rn ro rp rq rI c C sh |rt d�gng }
|
�t|| |
| d�t� t� f� t|
|||||pDd| d��}|j| ||d�}t||�S )a Read bytes from a file while tracking progress.
Args:
file (Union[str, PathLike[str], BinaryIO]): The path to the file to read, or a file-like object in binary mode.
total (int): Total number of bytes to read.
description (str, optional): Description of task show next to progress bar. Defaults to "Reading".
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.finished".
pulse_style (StyleType, optional): Style for pulsing bars. Defaults to "bar.pulse".
disable (bool, optional): Disable display of progress.
Returns:
ContextManager[BinaryIO]: A context manager yielding a progress reader.
z([progress.description]{task.description})rm rn ro rp r_ )rh ri rj rk rl rq )rg rf )rt ru rv �DownloadColumnrx r= � wrap_filer� )r� rg rf rh ri rj rk rl rm rn ro rp rq rz r> r� rG rG rH r� & s& %
r� r� ) rg rf rh ri rj rk rl rm rn ro rp rq z
PathLike[str]�rt�r)r� r� � buffering�encoding�errors�newlinerg rf rh ri rj rk rl rm rn ro rp rq rI c C s d S )NrG )r� r� r� r� r� r� rg rf rh ri rj rk rl rm rn ro rp rq rG rG rH �openg s r� �rbc C s d S )NrG )r� r� r� r� r� r� rg rf rh ri rj rk rl rm rn ro rp rq rG rG rH r� � s c
C sr |rt d�gng }|�t|
|||d�t� t� f� t||| |
| | pDd|d��}|j| |||||||d�}t||�S )a� Read bytes from a file while tracking progress.
Args:
path (Union[str, PathLike[str], BinaryIO]): The path to the file to read, or a file-like object in binary mode.
mode (str): The mode to use to open the file. Only supports "r", "rb" or "rt".
buffering (int): The buffering strategy to use, see :func:`io.open`.
encoding (str, optional): The encoding to use when reading in text mode, see :func:`io.open`.
errors (str, optional): The error handling strategy for decoding errors, see :func:`io.open`.
newline (str, optional): The strategy for handling newlines in text mode, see :func:`io.open`
total: (int, optional): Total number of bytes to read. Must be provided if reading from a file handle. Default for a path is os.stat(file).st_size.
description (str, optional): Description of task show next to progress bar. Defaults to "Reading".
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.finished".
pulse_style (StyleType, optional): Style for pulsing bars. Defaults to "bar.pulse".
disable (bool, optional): Disable display of progress.
encoding (str, optional): The encoding to use when reading in text mode.
Returns:
ContextManager[BinaryIO]: A context manager yielding a progress reader.
z([progress.description]{task.description})rm rn ro rp r_ )rh ri rj rk rl rq )r� r� r� r� r� rg rf )rt ru rv r� rx r= r� r� )r� r� r� r� r� r� rg rf rh ri rj rk rl rm rn ro rp rq rz r> r� rG rG rH r� � s6 1
c @ sj e Zd ZU dZdZee ed< dee dd�dd�Z ed�dd �Z
d
e d �d d
�Z e
d
e d �dd��ZdS )�ProgressColumnz3Base class for a widget to use in progress display.N� max_refresh)� table_columnrI c C s || _ i | _d | _d S )N)�
_table_column�_renderable_cacheZ _update_time)rE r� rG rG rH rD � s zProgressColumn.__init__)rI c C s | j p
t� S )z.Get a table column, used to build tasks table.)r� r5 )rE rG rG rH �get_table_column� s z ProgressColumn.get_table_column�Task)r} rI c C sr |� � }| jdk rT|jsTy| j|j \}}W n tk
r@ Y nX || j |krT|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)rk r� rB r� �id�KeyError�render)rE r} � current_time� timestamp�
renderablerG rG rH �__call__� s
zProgressColumn.__call__c C s dS )z"Should return a renderable object.NrG )rE r} rG rG rH r� s zProgressColumn.render)N)rW rX rY rZ r� r r[ �__annotations__r5 rD r� r. r� r r� rG rG rG rH r� � s
r� c sD e Zd ZdZd dd�eee d�� fdd�Zded �d
d �Z� Z S )
�RenderableColumnz�A column to insert an arbitrary column.
Args:
renderable (RenderableType, optional): Any renderable. Defaults to empty string.
� N)r� )r� r� c s || _ t� j|d� d S )N)r� )r� rC rD )rE r� r� )rF rG rH rD s zRenderableColumn.__init__r� )r} rI c C s | j S )N)r� )rE r} rG rG rH r� & s zRenderableColumn.render)r� )
rW rX rY rZ r. r r5 rD r� r] rG rG )rF rH r� s r� c sb e Zd ZdZdeee eeee d�� fdd �
Z
deee edd
�d d �Z d
e d�dd�Z
� ZS )�
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_namerm �speed�
finished_textr� c s< t |||d�| _t|t�r$t�|�n|| _t� j|d� d S )N)rm r� )r� ) r3 �spinner�
isinstancer� r7 � from_markupr� rC rD )rE r� rm r� r� r� )rF rG rH rD 4 s zSpinnerColumn.__init__)r� �
spinner_styler� rI c C s t |||d�| _dS )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.
)rm r� N)r3 r� )rE r� r� r� rG rG rH � set_spinnerD s
zSpinnerColumn.set_spinnerr� )r} rI c C s |j r | jn| j�|�� �}|S )N)�finishedr� r� r� rk )rE r} �textrG rG rH r� S s zSpinnerColumn.render)r� r� r� r� N)r� r� )rW rX rY rZ r� r r4 r[ r8 r5 rD r� r. r� r] rG rG )rF rH r� * s $
r� c sL e Zd ZdZd
eeeeee ee
dd�� fdd�
Z d e d
�d d �Z
� ZS )rt zA column containing text.�none�leftTN)� text_formatrm �justify�markup� highlighterr� rI c s: || _ || _|| _|| _|| _t� j|p0tdd�d� d S )NT)�no_wrap)r� )r� r� rm r� r� rC rD r5 )rE r� rm r� r� r� r� )rF rG rH rD _ s zTextColumn.__init__r� )r} rI c C sR | j j|d�}| jr*tj|| j| jd�}nt|| j| jd�}| jrN| j�|� |S )N)r} )rm r� ) r� �formatr� r7 r� rm r� r� � highlight)rE r} �_textr� rG rG rH r� o s zTextColumn.render)r� r� TNN)rW rX rY rZ r� r4 r- r� r r/ r5 rD r7 r� r] rG rG )rF rH rt \ s rt c sL e Zd ZdZdee eeeeee dd�� fd d
�
Zd e d �d
d�Z
� Z S )rv 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.finished".
pulse_style (StyleType, optional): Style for pulsing bars. Defaults to "bar.pulse".
�( �bar.back� bar.complete� bar.finished� bar.pulseN)� bar_widthrm rn ro rp r� rI c s0 || _ || _|| _|| _|| _t� j|d� d S )N)r� )r� rm rn ro rp rC rD )rE r� rm rn ro rp r� )rF rG rH rD � s zBarColumn.__init__r� )r} rI c C s^ t |jdk rtd|j�ndtd|j�| jdkr2dn
td| j�|j |�� | j| j| j | j
d� S )z&Gets a progress bar widget for a task.Nr r( ) rg rB �widthZpulseZanimation_timerm rn ro rp ) r2 rg �maxrB r� �startedrk rm rn ro rp )rE r} rG rG rH r� � s
zBarColumn.render)r� r� r� r� r� N) rW rX rY rZ r r� r4 r5 rD r2 r� r] rG rG )rF rH rv z s rv c @ s e Zd ZdZded�dd�ZdS )�TimeElapsedColumnzRenders time elapsed.r� )r} rI c C sD |j r |jn|j}|dkr&tddd�S tt|�d�}tt|�dd�S )zShow time elapsed.Nz-:--:--zprogress.elapsed)rm )�seconds)r� �
finished_time�elapsedr7 r r� r� )rE r} r� �deltarG rG rH r� � s
zTimeElapsedColumn.renderN)rW rX rY rZ r7 r� rG rG rG rH r� � s r� c sh e Zd ZdZdeeeeeee ee
edd � � fd
d �
Z e ee
ed �d
d��Zded�dd�Z� ZS )rw a Show task progress as a percentage.
Args:
text_format (str, optional): Format for percentage display. Defaults to "[progress.percentage]{task.percentage:>3.0f}%".
text_format_no_percentage (str, optional): Format if percentage is unknown. Defaults to "".
style (StyleType, optional): Style of output. Defaults to "none".
justify (JustifyMethod, optional): Text justification. Defaults to "left".
markup (bool, optional): Enable markup. Defaults to True.
highlighter (Optional[Highlighter], optional): Highlighter to apply to output. Defaults to None.
table_column (Optional[Column], optional): Table Column to use. Defaults to None.
show_speed (bool, optional): Show speed if total is unknown. Defaults to False.
�-[progress.percentage]{task.percentage:>3.0f}%r� r� r� TNF) r� �text_format_no_percentagerm r� r� r� r� rr rI c s( || _ || _t� j||||||d� d S )N)r� rm r� r� r� r� )r� rr rC rD ) rE r� r� rm r� r� r� r� rr )rF rG rH rD � s zTaskProgressColumn.__init__)r� rI c C sT |dkrt ddd�S t�t|�dddddgd �\}}|| }t |d
�|� d �dd�S ) z�Render the speed in iterations per second.
Args:
task (Task): A Task object.
Returns:
Text: Text object containing the task speed.
Nr� zprogress.percentage)rm u ×10³u ×10⁶u ×10⁹u ×10¹²i� z.1fz it/s)r7 r) �pick_unit_and_suffixr� )�clsr� �unit�suffix�
data_speedrG rG rH � render_speed� s
z TaskProgressColumn.render_speedr� )r} rI c C s� |j d kr"| jr"| �|jp |j�S |j d kr2| jn| j}|j|d�}| jr`t j
|| j | j d�}nt || j | j d�}| j
r�| j
�|� |S )N)r} )rm r� )rg rr r� �finished_speedr� r� r� r� r� r7 r� rm r� r� r� )rE r} r� r� r� rG rG rH r� � s zTaskProgressColumn.render)r� r� r� r� TNNF)rW rX rY rZ r� r4 r- r� r r/ r5 rD � classmethodr[ r7 r� r� r] rG rG )rF rH rw � s
rw c sD e Zd ZdZdZd eeee d�� fdd�
Zde d �d
d �Z
� Z S )
rx a Renders estimated time remaining.
Args:
compact (bool, optional): Render MM:SS when time remaining is less than an hour. Defaults to False.
elapsed_when_finished (bool, optional): Render time elapsed when the task is finished. Defaults to False.
g �?FN)�compactrs r� c s || _ || _t� j|d� d S )N)r� )r� rs rC rD )rE r� rs r� )rF rG rH rD s z TimeRemainingColumn.__init__r� )r} rI c C s� | j r|jr|j}d}n
|j}d}|jdkr8td|d�S |dkrVt| jrLdnd|d�S tt|�d�\}}t|d�\}}| jr�|s�|d �d
|d ��}n|d �d
|d �d
|d ��}t||d�S ) zShow time remaining.zprogress.elapsedzprogress.remainingNr� )rm z--:--z-:--:--�< Z02d�:�d) rs r� r� �time_remainingrg r7 r� �divmodr� )rE r} Z task_timerm �minutesr� �hours� formattedrG rG rH r� s
zTimeRemainingColumn.render)FFN) rW rX rY rZ r� r� r r5 rD r7 r� r] rG rG )rF rH rx � s rx c @ s e Zd ZdZded�dd�ZdS )�FileSizeColumnzRenders completed filesize.r� )r} rI c C s t �t|j��}t|dd�S )zShow data completed.zprogress.filesize)rm )r) �decimalr� rB r7 )rE r} � data_sizerG rG rH r� , s zFileSizeColumn.renderN)rW rX rY rZ r7 r� rG rG rG rH r ) s r c @ s e Zd ZdZded�dd�ZdS )�TotalFileSizeColumnzRenders total filesize.r� )r} rI c C s* |j dk rt�t|j ��nd}t|dd�S )zShow data completed.Nr� zprogress.filesize.total)rm )rg r) r r� r7 )rE r} r rG rG rH r� 5 s zTotalFileSizeColumn.renderN)rW rX rY rZ r7 r� rG rG rG rH r 2 s r c s> e Zd ZdZd eee d�� fdd�
Zded�d d
�Z � Z
S ) �MofNCompleteColumnaH Renders completed count/total, e.g. ' 10/1000'.
Best for bounded tasks with int quantities.
Space pads the completed count so that progress length does not change as task progresses
past powers of 10.
Args:
separator (str, optional): Text to separate completed and total values. Defaults to "/".
�/N)� separatorr� c s || _ t� j|d� d S )N)r� )r rC rD )rE r r� )rF rG rH rD G s zMofNCompleteColumn.__init__r� )r} rI c C sP t |j�}|jdk r t |j�nd}tt|��}t||� d��| j� |� �dd�S )zShow completed/total.N�?r� zprogress.download)rm )r� rB rg r� r� r7 r )rE r} rB rg Z total_widthrG rG rH r� K s
zMofNCompleteColumn.render)r N) rW rX rY rZ r� r r5 rD r7 r� r] rG rG )rF rH r ; s
r c s@ e Zd ZdZd eee dd�� fdd�
Zded�d d
�Z � Z
S ) r� z�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_unitsr� rI c s || _ t� j|d� d S )N)r� )r
rC rD )rE r
r� )rF rG rH rD ] s zDownloadColumn.__init__r� )r} rI c C s� t |j�}|jdk r t |j�n|}| jrNt�|dddddddd d
g d �\}}n$t�|dd d
ddddddg d�\}}|dkr~dnd}|| }|d|� d��}|jdk r�t |j�} | | }
|
d|� d��} nd} |� d| � d|� �} t| d d �}
|
S ) z.Calculate common unit for completed and total.Nr� ZKiBZMiBZGiBZTiBZPiBZEiBZZiBZYiBi ZkBZMBZGBZTBZPBZEBZZBZYBi� r( r z,.�fr r r� zprogress.download)rm )r� rB rg r
r) r� r7 )rE r} rB Z unit_and_suffix_calculation_baser� r� Z precisionZcompleted_ratioZ
completed_strrg Z total_ratioZ total_strZdownload_statusZ
download_textrG rG rH r� c s,
zDownloadColumn.render)FN) rW rX rY rZ r� r r5 rD r7 r� r] rG rG )rF rH r� V s
r� c @ s e Zd ZdZded�dd�ZdS )�TransferSpeedColumnz&Renders human readable transfer speed.r� )r} rI c C s@ |j p
|j}|dkr tddd�S t�t|��}t|� d�dd�S )zShow data transfer speed.Nr zprogress.data.speed)rm z/s)r� r� r7 r) r r� )rE r} r� r� rG rG rH r� � s
zTransferSpeedColumn.renderN)rW rX rY rZ r7 r� rG rG rG rH r � s r c @ s" e Zd ZU dZeed< eed< dS )�ProgressSamplez$Sample of progress for a given time.r� rB N)rW rX rY rZ r[ r� rG rG rG rH r
� s
r
c @ s� e Zd ZU dZeed< eed< ee ed< eed< e ed< dZ
ee ed< d Z e ed
< e
ed �Zeeef ed < e
dd
d
d�Zee ed< e
dd
d
d�Zee ed< dZee ed< e
dd� d
d
d�Zee ed< e
d
ed�Zeed< ed�dd�Zee d�dd ��Z eee d�d d ��Z eee d�d d ��Z ee d�d!d"��Z eed�d#d$��Z eee d�d%d&��Z!eee d�d'd(��Z"dd�d)d*�Z#dS )+r� z�Information regarding a progress task.
This object should be considered read-only outside of the :class:`~Progress` class.
r� rf rg rB � _get_timeNr� T�visible)�default_factory�fieldsF)�default�init�repr�
start_time� stop_timer� c C s
t dd�S )Ni� )�maxlen)r rG rG rG rH �<lambda>� � z
Task.<lambda>)r r r � _progress)r r �_lock)rI c C s | � � S )z(float: Get the current time, in seconds.)r )rE rG rG rH rk � s z
Task.get_timec C s
| j dk S )z#bool: Check if the task as started.N)r )rE rG rG rH r� � s z Task.startedc C s | j dkrdS | j | j S )zPOptional[float]: Get the number of steps remaining, if a non-None total was set.N)rg rB )rE rG rG rH � remaining� s
zTask.remainingc C s2 | j dkrdS | jdk 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 rk )rE rG rG rH r� � s
z Task.elapsedc C s
| j dk S )z Check if the task has finished.N)r� )rE rG rG rH r� � s z
Task.finishedc C s. | j s
dS | j| j d }tdtd|��}|S )zOfloat: Get progress of task as a percentage. If a None total was set, returns 0g g Y@)rg rB �minr� )rE rB rG rG rH �
percentage� s
zTask.percentagec C s| | j dkrdS | j�^ | j}|s$dS |d j|d j }|dkrDdS t|�}t|� tdd� |D ��}|| }|S Q R X dS )z=Optional[float]: Get the estimated speed in steps per second.Nr� r c s s | ] }|j V qd S )N)rB )�.0�samplerG rG rH � <genexpr> s z Task.speed.<locals>.<genexpr>)r r r r� �iterr� r� )rE r> Z
total_timeZ
iter_progressZtotal_completedr� rG rG rH r� � s
z
Task.speedc C s: | j r
dS | j}|sdS | j}|dkr*dS t|| �}|S )zJOptional[float]: Get estimated time to completion, or ``None`` if no data.g N)r� r� r r
)rE r� r ZestimaterG rG rH r� s zTask.time_remainingc C s | j �� d| _d| _dS )zReset progress.N)r �clearr� r� )rE rG rG rH �_reset s
z Task._reset)$rW rX rY rZ r9 r� r� r r[ �GetTimeCallabler� r r� r �dictr r r r r r� r r r
r r rk r� r� r r� r� r r� r� r$ rG rG rG rH r� � s>
r� c @ s� e Zd ZdZddddddddddd�
eeef ee e e
e
e e e ee e e dd� d d
�Z e
eed f d �d
d��Zeed �dd��Zeee d �dd��Zeee d �dd��Zee d �dd��Zdd �dd�Zdd �dd�Zd d �dd �Zeee ee ee dd �d d �Z dXee e e!e f ee
ee ee
e e d"�d#d$�Z"dYdd%d&�e#ee$ ee ee#d'�d(d)�Z%e&j'dZddd%d+�eed,e(f e)d- e$ee ee ee ee$ ee ee#d.�
d/d0��Z*e&j'd[ddd%d+�eed,e(f ee)d1 e)d2 f e$ee ee ee ee$ ee ee+d.�
d3d0��Z*d\ddd%d+�eed,e(f ee)d- e)d2 e)d1 f e$ee ee ee ee$ ee eee#e+f d.�
d4d0�Z*edd5�d6d7�Z,edd5�d8d9�Z-ddddddd:�eee
ee
ee
ee ee e e.dd;� d<d=�Z/ddd>ddd?�ee ee
e$ee ee e.dd@�dAdB�Z0d]ee
ddD�dEdF�Z1dd �dGdH�Z2e3d �dIdJ�Z4e e3 d �dKdL�Z5e e e6dM�dNdO�Z7e3d �dPdQ�Z8d^ee ee
e$e e.edS�dTdU�Z9edd5�dVdW�Z:dS )_r= 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.
NTr_ g >@F)
ri rh rl �speed_estimate_periodrj �redirect_stdout�redirect_stderrrk rq �expand) rz ri rh rl r' rj r( r) rk rq r* rI c
G s� |dkst d��t� | _| p"| �� | _|| _| | _|
| _i | _t d�| _
t |pRt � |||||| j
d�| _|pr| jj| _| jj| _| jj| _d S )Nr z refresh_per_second must be > 0)ri rh rl rj r( r) �get_renderable)�AssertionErrorr r �get_default_columnsrz r' rq r* �_tasksr9 � _task_indexr1 r* r+ �liveri rk �print�log) rE ri rh rl r' rj r( r) rk rq r* rz rG rG rH rD * s&
zProgress.__init__.)rI c C s t d�t� t� t� fS )a� Get the default columns used for a new Progress instance:
- a text column for the description (TextColumn)
- the bar itself (BarColumn)
- a text column showing completion percentage (TextColumn)
- an estimated-time-remaining column (TimeRemainingColumn)
If the Progress instance is created without passing a columns argument,
the default columns defined here will be used.
You can also create a Progress instance using custom columns before
and/or after the defaults, as in this example:
progress = Progress(
SpinnerColumn(),
*Progress.default_columns(),
"Elapsed:",
TimeElapsedColumn(),
)
This code shows the creation of a Progress display, containing
a spinner to the left, the default columns, and a labeled elapsed
time column.
z([progress.description]{task.description})rt rv rw rx )r� rG rG rH r- N s z Progress.get_default_columnsc C s | j jS )N)r0 ri )rE rG rG rH ri m s zProgress.consolec C s | j � t| j�� �S Q R X dS )z Get a list of Task instances.N)r �listr. �values)rE rG rG rH �tasksq s zProgress.tasksc C s | j � t| j�� �S Q R X dS )zA list of task IDs.N)r r3 r. �keys)rE rG rG rH �task_idsw s zProgress.task_idsc C s4 | j �$ | jsdS tdd� | j�� D ��S Q R X dS )z'Check if all tasks have been completed.Tc s s | ] }|j V qd S )N)r� )r r} rG rG rH r! � s z$Progress.finished.<locals>.<genexpr>N)r r. �allr4 )rE rG rG rH r� } s zProgress.finishedc C s | j s| jjdd� dS )zStart the progress display.T)rJ N)rq r0 rO )rE rG rG rH rO � s zProgress.startc C s | j �� | jjs | j�� dS )zStop the progress display.N)r0 r� ri Zis_interactiver1 )rE rG rG rH r� � s
z
Progress.stopc C s | � � | S )N)rO )rE rG rG rH rP � s zProgress.__enter__)rQ rR rS rI c C s | � � d S )N)r� )rE rQ rR rS rG rG rH rV � s zProgress.__exit__�
Working...皙�����?)re rg r? rf r@ rI c
c s� |dkrt t|��pd}|dkr0| j||d�}n| j||d� | jjr�t| ||��(}x |D ]}|V | jd7 _qZW W dQ R X n0| j}| j } x"|D ]}|V ||d� | � q�W dS )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.
N)rg r( )
r[ r �add_taskrM r0 rh r< rB rK rJ )
rE re rg r? rf r@ Z track_thread�valuerK rJ rG rG rH ry � s
zProgress.trackz
Reading...)r? rf )r� rg r? rf rI c C s~ d}|dk r|}n&|dk r8| j � | j| j}W dQ R X |dkrHtd��|dkr`| j||d�}n| j||d� t|| |dd�S )aj Track progress file reading from a binary file.
Args:
file (BinaryIO): A file-like object opened in binary mode.
total (int, optional): Total number of bytes to read. This must be provided unless a task with a total is also given.
task_id (TaskID): Task to track. Default is new task.
description (str, optional): Description of task, if new task is created.
Returns:
BinaryIO: A readable file-like object in binary mode.
Raises:
ValueError: When no total value can be extracted from the arguments or the task.
Nz?unable to get the total number of bytes, please specify 'total')rg F)r~ )r r. rg �
ValueErrorr; rM r{ )rE r� rg r? rf � total_bytesrG rG rH r� � s zProgress.wrap_filer� )rg r? rf z
PathLike[str]r� )
r� r� r� r� r� r� rg r? rf rI c
C s d S )NrG )
rE r� r� r� r� r� r� rg r? rf rG rG rH r� � s z
Progress.openr� r� c
C s d S )NrG )
rE r� r� r� r� r� r� rg r? rf rG rG rH r� � s c C s� d� t|dd��}
|
dkr(td�|���|dk} |
dkrR|dkrRt�dt� d }n&|
d
krx|d krltd ��n |dkrxd }|d
kr�t|�j}|d
kr�| j | |d�}n| j
||d� t j |d|d�} t
| | |dd�}
|dkr�t j|
|||| d�S |
S )a# Track progress while reading from a binary file.
Args:
path (Union[str, PathLike[str]]): The path to the file to read.
mode (str): The mode to use to open the file. Only supports "r", "rb" or "rt".
buffering (int): The buffering strategy to use, see :func:`io.open`.
encoding (str, optional): The encoding to use when reading in text mode, see :func:`io.open`.
errors (str, optional): The error handling strategy for decoding errors, see :func:`io.open`.
newline (str, optional): The strategy for handling newlines in text mode, see :func:`io.open`.
total (int, optional): Total number of bytes to read. If none given, os.stat(path).st_size is used.
task_id (TaskID): Task to track. Default is new task.
description (str, optional): Description of task, if new task is created.
Returns:
BinaryIO: A readable file-like object in binary mode.
Raises:
ValueError: When an invalid mode is given.
r� F)�reverse)�brr� r� zinvalid mode {!r}r( r@ zaline buffering (buffering=1) isn't supported in binary mode, the default buffer size will be usedr� )r� r� r z can't have unbuffered text I/ON)rg r� )r� T)r~ )r� r� )r� r� r� �line_buffering)rU �sortedr= r� �warnings�warn�RuntimeWarningr �st_sizer; rM �ior� r{ �
TextIOWrapper)rE r� r� r� r� r� r� rg r? rf �_moderA r| r� rG rG rH r� s: !
)r? rI c C s4 | j �$ | j| }|jdkr&| �� |_W dQ R X dS )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 rk )rE r? r} rG rG rH �
start_task\ s
zProgress.start_taskc C s> | j �. | j| }| �� }|jdkr*||_||_W dQ R X dS )z�Stop a task.
This will freeze the elapsed time on the task.
Args:
task_id (TaskID): ID of task.
N)r r. rk r r )rE r? r} r� rG rG rH � stop_taskj s
zProgress.stop_task)rg rB rK rf r rJ ) r? rg rB rK rf r rJ r rI c K s. | j �� | j| } | j}
|dk r:|| jkr:|| _| �� |dk rP| j|7 _|dk r^|| _|dk rl|| _|dk rz|| _| j�|� | j|
} | � � } | | j
}
| j }|j }x |r�|d j
|
k r�|� q�W | dkr�|�t| | �� | jdk �r| j| jk�r| jdk�r| j| _W dQ R X |�r*| �� dS )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. rB rg r$ rf r r rM rk r' r �popleftr� �appendr
r� r� rJ )rE r? rg rB rK rf r rJ r r} �completed_start�update_completedr� �old_sample_timer rL rG rG rH rM y s<
zProgress.updater )rO rg rB r rf )r? rO rg rB r rf r rI c
K s� | � � }| j�f | j| } | �� |r*|nd| _|dk r>|| _|| _|dk rR|| _|r\|| _|dk rj|| _ d| _
W dQ R X | � � dS )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.
visible (bool, optional): Enable display of the task. Defaults to True.
description (str, optional): Change task description if not None. Defaults to None.
**fields (str): Additional data fields required for rendering.
N) rk r r. r$ r rg rB r r rf r� rJ )
rE r? rO rg rB r rf r r� r} rG rG rH �reset� s
zProgress.resetr( )r? rK rI c
C s� | � � }| j�� | j| }|j}| j|7 _|j| }|| j }|j}|j} x |rj|d j|k rj| � qPW xt|�dkr�| � qnW |� t
||�� |j dk r�|j|j kr�|j dkr�|j
|_ |j|_W dQ R X dS )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 i� N)rk r r. rB r' r rL r� r� rM r
rg r� r� r� r� )
rE r? rK r� r} rN rO rP r rL rG rG rH rK � s&
zProgress.advancec C s | j s| jjr| j�� dS )z*Refresh (render) the progress information.N)rq r0 Z
is_startedrJ )rE rG rG rH rJ � s zProgress.refreshc C s t | �� � }|S )z*Get a renderable for the progress display.)r, �get_renderables)rE r� rG rG rH r+ s zProgress.get_renderablec c s | � | j�}|V dS )z5Get a number of renderables for the progress display.N)�make_tasks_tabler5 )rE �tablerG rG rH rR s zProgress.get_renderables)r5 rI c sV dd� | j D �}tj|d| jd��}x,|D ]$� � jr*|j� fdd�| j D �� q*W |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�rtd d�n
|�� �� V qdS )T)r� N)r� r� r5 r� �copy)r Z_columnrG rG rH r! s z,Progress.make_tasks_table.<locals>.<genexpr>)r r( )�paddingr* c 3 s, | ]$}t |t�r |j� d �n|� �V qdS ))r} N)r� r� r� )r �column)r} rG rH r! ! s )rz r6 Zgridr* r �add_row)rE r5 Z
table_columnsrT rG )r} rH rS
s
zProgress.make_tasks_tablec C s | j �
| �� S Q R X dS )z+Makes the Progress class itself renderable.N)r r+ )rE rG rG rH �__rich__+ s zProgress.__rich__� Y@)rf rO rg rB r r rI c K sr | j �Z t| j|||||| j| j d�}|| j| j< |rB| �| j� | j}tt| j�d �| _W dQ R X | �� |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 known.
Set to None to render a pulsing animation. 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/ rk r. rJ r9 r� rJ ) rE rf rO rg rB r r r} Znew_task_indexrG rG rH r; 0 s"
zProgress.add_taskc C s | j � | j|= W dQ R X dS )z]Delete a task if it exists.
Args:
task_id (TaskID): A task ID.
N)r r. )rE r? rG rG rH � remove_task[ s zProgress.remove_task)NNr9 r: )N)r� NNN)r� NNN)r� r� NNN)r( )TrZ r T);rW rX rY rZ r$ r� r� r r+ r� r[ r% rD r� r! r- r� ri r r� r5 r9 r7 r� rO r� rP r" r\ r rV r r: r ry r r� r� �typing�overloadr� r' r� r rJ rK r rM rQ rK rJ r. r+ rR r6 rS rY r; r[ rG rG rG rH r= s� ,
$%# : F LA(6 ! #r= �__main__)�Panel)�Rule)�Syntax)r6 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)Z line_numbersZfoo�barZbaz�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...ZexamplezPretty printed)�typer� z Syntax...zGive it a try!)�cycle)�record)ri rj z[red]Downloadingi� )rg z[green]Processingz[yellow]Thinkingg �?)rK g333333�?g{�G�z�?�d )r^ NTNFNr_ r` ra rb rc rd FT)r� NNN)r� NNN)r� r� NNN)�rG �sysr\ rC �abcr r � collectionsr Z dataclassesr r �datetimer r r �mathr
r �operatorr �osr
r � threadingr r r �typesr r r r r r r r r r r r r r r r! r" r# r$ � version_infor' Z pip._vendor.typing_extensionsr� r) r* ri r+ r, r- r. r� r/ Zjupyterr0 r0 r1 � progress_barr2 r� r3 rm r4 rT r5 r6 r� r7 r8 r� r9 r: r[ r% r; r< r� r� ry r{ r� r� r] r� r� r� r� r� rt rv r� rw rx r r r r� r r
r� r= rW �random�timeZpanelr_ Zruler` Zsyntaxra rX Zprogress_renderables� itertoolsrh Zexamplesr- r> r; Ztask1Ztask2Ztask3r� rM �sleep�randintr2 r� rG rG rG rH �<module> sb P
* D6d02 d X j B*2 * I0 1 ~ P