R`i: +
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/ ej` dk\ rd dlm1Z1 nd dl2m1Z1 ddl3m4Z4m5Z5 ddl6m7Z7m8Z8m9Z9m:Z: ddl;mZ> ddl?m@Z@ ddlAmBZB ddlCmDZD ddlEmFZF ddlGmHZHmIZI ddlJmKZKmLZL e(deM ZN e.d ZOe g ePf ZQ ej\ d e+e ZR G d d e ZS dd'e/e*eO e%eO f d(eTd)e)eP d*eUd+e)e7 d,eUd-e)e g ePf d.ePd/eFd0eFd1eFd2eFd3ePd4eUd5eUd6e%eO f d7ZV G d8 d9ee ZW G d: d;e!eR e$eR ZXde d)eMd(eTd*eUd+e)e7 d,eUd-e)e g ePf d.ePd/eFd0eFd1eFd2eFd4eUd6e!e f d?ZYej ddde/eTdAe[f dBe/e1dC e1dD f dEeMdFe)eT dGe)eT dHe)eT d)e)eM d(eTd*eUd+e)e7 d,eUd-e)e g ePf d.ePd/eFd0eFd1eFd2eFd4eUd6e!e+ f&dI Z\ej ddde/eTdAe[f dBe1dJ dEeMdFe)eT dGe)eT dHe)eT d)e)eM d(eTd*eUd+e)e7 d,eUd-e)e g ePf d.ePd/eFd0eFd1eFd2eFd4eUd6e!e f&dK Z\ ddde/eTdAe[f dBe/e1dJ e1dC e1dD f dEeMdFe)eT dGe)eT dHe)eT d)e)eM d(eTd*eUd+e)e7 d,eUd-e)e g ePf d.ePd/eFd0eFd1eFd2eFd4eUd6e/e!e e!e+ f f&dLZ\ G dM dNe Z] G dO dPe] Z^ G dQ dRe] Z_ G dS dTe] Z` G dU dVe] Za G dW dXe] Zb G dY dZe` Zc G d[ d\e] Zd G d] d^e] Ze G d_ d`e] Zf G da dbe] Zg G dc dde] Zh G de dfe] Zi G dg dhe' Zje
G di dj Zk G dk dle> Zlemdmk( rhd dlnZnd dloZoddnlpmqZq ddolrmsZs ddpltmuZu ddqlGmIZI eudrdsd t Zt eIdudvdw ZGeGj dxdydz d{ eqd| d}eGd~ddddet esd g Zwd dlxmyZy eyew Zz e7d Z6 el e_ gelj eb e6d!d5 Z|e|j dd Z~e|j dd Ze|j dd Ze|j sxe|j e~d e|j ed eoj d enj d d dk re|j eez e|j sxddd yy# 1 sw Y yxY w) N)ABCabstractmethoddeque) dataclassfield) timedelta) RawIOBaseUnsupportedOperation)ceil)mmap) length_hint)PathLikestat)EventRLockThread)
TracebackType)AnyBinaryIOCallableContextManagerDequeDictGenericIterableList
NamedTupleNewTypeOptionalSequenceTextIOTupleTypeTypeVarUnion) )Literal )filesize get_console)ConsoleGroup
JustifyMethodRenderableType) Highlighter) JupyterMixin)Live) ProgressBar)Spinner) StyleType)ColumnTable)TextTextTypeTaskID ProgressType_Ic p e Zd ZdZdddddef fd Zdd
Zdd Zd ee e
d
ee
dee dd fdZ xZ
S ) _TrackThreadz)A thread to periodically update progress.progressProgresstask_idr;
update_periodc x || _ || _ || _ t | _ d| _ t | y )Nr )r@ rB rC r done completedsuper__init__)selfr@ rB rC __class__s /builddir/build/BUILDROOT/alt-python312-pip-23.3.1-3.el8.x86_64/opt/alt/python312/lib/python3.12/site-packages/pip/_vendor/rich/progress.pyrH z_TrackThread.__init__B s4
*G
returnNc T | j }| j j }| j }d}| j j
} || s(| j }||k7 r ||||z
|} || s(| j j | j | j d y )Nr T)rF refresh)rB r@ advancerC rE waitrF update)rI rB rP rC last_completedrQ rF s rK runz_TrackThread.runK s ,,--'' **
yy~~}% I *^!; ?@UW NN - - '
* 5 d ;
!-3H
>> E{- "
s0 A B B7B8B< BBB Bc e Zd ZdZ d'dedddededdf
d Zd(d
Zd e e
e d e e d
e e ddfdZ
defdZdefdZedefd ZdefdZdefdZedefd Zedefd ZdefdZdefdZdefdZd)dedefdZ de e e e f fd Z!d)dedefd Z"d)d ede#e fd Z$d*d Z%d+d!ed"edefd#Z&defd$Z'd%e(defd&Z)y),_Readerz9A reader that tracks progress while it's being read from.handler@ rA task close_handlerM Nc J || _ || _ || _ || _ d| _ y NF)r r@ r r _closed)rI r r@ r r s rK rH z_Reader.__init__ s(
( rL c : | j j | S rV )r rZ rY s rK rZ z_Reader.__enter__ s rL r\ r] r^ c $ | j y rV )closerb s rK rc z_Reader.__exit__ s
rL c | S rV rY s rK __iter__z_Reader.__iter__ s rL c t | j }| j j | j t | |S NrP )nextr r@ rP r len)rI lines rK __next__z_Reader.__next__ s4 DKK
diiT ; rL c | j S rV )r rY s rK closedz_Reader.closed s ||rL c 6 | j j S rV )r filenorY s rK r z_Reader.fileno {{!!##rL c 6 | j j S rV )r isattyrY s rK r z_Reader.isatty r rL c . | j j S rV )r moderY s rK r z _Reader.mode {{ rL c . | j j S rV )r namerY s rK r z _Reader.name r rL c 6 | j j S rV )r readablerY s rK r z_Reader.readable {{##%%rL c 6 | j j S rV )r seekablerY s rK r z_Reader.seekable r rL c yr r rY s rK writablez_Reader.writable s rL sizec | j j | }| j j | j t | |S r )r readr@ rP r r )rI r blocks rK r z _Reader.read s: &
diiU < rL bc | j j | }| j j | j | |S r )r readintor@ rP r )rI r ns rK r z_Reader.readinto s6 KK #
dii 3rL c | j j | }| j j | j t | |S r )r readliner@ rP r r )rI r r s rK r z_Reader.readline s: {{##D)
diiT ; rL hintc | j j | }| j j | j t t
t | |S r )r readlinesr@ rP r summapr )rI r liness rK r z_Reader.readlines sA %%d+
diiSe_1E F rL c ^ | j r| j j d| _ y )NT)r r r r rY s rK r z
_Reader.close s" KK rL offsetwhencec | j j || }| j j | j | |S )NrF )r seekr@ rR r )rI r r poss rK r z _Reader.seek s8 kk vv.
TYY# 6
rL c 6 | j j S rV )r tellrY s rK r z _Reader.tell
s {{ !!rL sc t d )Nwrite)r )rI r s rK r z
_Reader.write
s
"7++rL )T)rM r )rd )r )*re rf rg rh r r; boolrH rZ r r$ rj r rc r bytesr propertyr intr r strr r r r r r r& bytearray
memoryviewr
r r r r r r r r r r rL rK r r s C "
4
./ -( '
( %
$ $$ $ c c &$ &&$ &$ e
% :t ;<
S %
c 4;
3 C
"c ",s ,s ,rL r c ` e Zd ZdZdddeddfdZdefdZd eee d
ee d ee
ddfd Z y)
_ReadContextzEA utility class to handle a context for both a reader and a progress.r@ rA readerrM Nc || _ || _ y rV )r@ r )rI r@ r s rK rH z_ReadContext.__init__ s
rL c j | j j | j j S rV )r@ rX r rZ rY s rK rZ z_ReadContext.__enter__ s%
{{$$&&rL r\ r] r^ c r | j j | j j ||| y rV )r@ stopr rc rb s rK rc z_ReadContext.__exit__ s*
Xw7rL ) re rf rg rh r= rH rZ r r$ rj r rc r rL rK r r sj O! !R !D !'2 '84
./8 -(8 ' 8
8rL r
Reading...) rs ru rv rw rx ry rz r{ r| r} r~ filec
|r t d gng }
|
j t || |
| t t f t |
|||||xs d| d}|j
| || }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.
r r rm r )rt rs )r r r DownloadColumnr rA wrap_filer )r rt rs ru rv rw rx ry rz r{ r| r} r~ r r@ r s rK r r & s J EP> ?@UW NN - - '
!
!-3H
E{
KF & ))rL ) rt rs ru rv rw rx ry rz r{ r| r} r~
PathLike[str]r rtr bufferingencodingerrorsnewlinec y rV r r r r r r r rt rs ru rv rw rx ry rz r{ r| r} r~ s rK openr g , rL rbc y rV r r s rK r r r rL c
|r t d gng }|j t |
||| t t f t ||| |
| | xs d|d}|j
| ||||||| }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.
r r rm r )r r r r r rt rs )r r r r r rA r r )r r r r r r rt rs ru rv rw rx ry rz r{ r| r} r~ r r@ r s rK r r s b EP> ?@UW NN - - '
!
!-3H ]]
F & ))rL c t e Zd ZU dZdZee ed< d dee ddfdZ defdZ
dd de fd
Z e
dd de fd Zy)
ProgressColumnz3Base class for a widget to use in progress display.N max_refresh table_columnrM c . || _ i | _ d | _ y rV )
_table_column_renderable_cache _update_time)rI r s rK rH zProgressColumn.__init__ s )MO -1rL c 2 | j xs
t S )z.Get a table column, used to build tasks table.)r r7 rY s rK get_table_columnz ProgressColumn.get_table_column s !!-VX-rL r Taskc . |j }| j =|j s1 | j |j \ }}|| j z |kD r|S | j
| }||f| j |j < |S # t
$ r Y 9w xY w)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).
)rx r rF r idKeyErrorrender)rI r current_time timestamp
renderables rK __call__zProgressColumn.__call__ s }} '
&(,(>(>tww(G% : t / //,>%%[[&
+7*D tww'
s B BBc y)z"Should return a renderable object.Nr rI r s rK r zProgressColumn.render s rL rV )re rf rg rh r r ri __annotations__r7 rH r r0 r r r r rL rK r r sl =#'K% '2Xf%5 2 2
.& .V . 16 1n 1 1rL r c L e Zd ZdZ d dddedee f fdZddd efd
Z xZ S ) RenderableColumnzA column to insert an arbitrary column.
Args:
renderable (RenderableType, optional): Any renderable. Defaults to empty string.
Nr r r c 4 || _ t | | y Nr
)r rG rH )rI r r rJ s rK rH zRenderableColumn.__init__ s %
l3rL r r rM c | j S rV )r r s rK r zRenderableColumn.render& s rL ) )
re rf rg rh r0 r r7 rH r rk rl s @rK r r sB ,.4SW4(4@H@P4 6 n rL r c e Zd ZdZ ddedee dededee f
fd
Z
dded ee ded
dfd Z d d
d
e fdZ
xZS )
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 " ".
N spinner_namerz speed
finished_textr c t ||| | _ t |t rt j
| n|| _ t | ! | y )Nrz r r
) r5 spinner
isinstancer r9 from_markupr rG rH )rI r rz r r r rJ s rK rH zSpinnerColumn.__init__4 sN |5F --
] +
l3rL
spinner_stylerM c * t ||| | _ y)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)r5 r )rI r r r s rK set_spinnerzSpinnerColumn.set_spinnerD s |=N rL r r c |j r| j }|S | j j |j }|S rV )finishedr r r rx )rI r texts rK r zSpinnerColumn.renderS sH }}
$$T]]_5
rL )dotsprogress.spinner ? N)r r! )re rf rg rh r r r6 ri r: r7 rH r r0 r rk rl s @rK r r * s #%7"%)-
44 "4 4
4 v&
4& .@
O
O *
O
O
O 6 n rL r c h e Zd ZdZ ddededededee dee
d df fd
Z d d d e fd
Z
xZS )r zA column containing text.N text_formatrz justifymarkup highlighterr rM c || _ || _ || _ || _ || _ t
| |xs t d y )NTno_wrapr
)r$ r% rz r& r' rG rH r7 )rI r$ rz r% r& r' r rJ s rK rH zTextColumn.__init___ sD '&-
&
l&JfT6JKrL r r c B | j j | }| j r-t j || j
| j }n"t || j
| j }| j r| j j | |S Nr )rz r% ) r$ formatr& r9 r rz r% r' highlight)rI r _textr s rK r zTextColumn.rendero sv ''T'2 ;;##ET\\RDTZZ FD & &t , rL )noneleftTNN)re rf rg rh r r6 r/ r r r1 r7 rH r9 r rk rl s @rK r r \ s #
"!'-1)-LL L L
L k*
L v&L
L 6 d rL r c j e Zd ZdZ ddee dededededee d df fd
Zd d d e fd
Z
xZ S )r 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".
N bar_widthrz r{ r| r} r rM c l || _ || _ || _ || _ || _ t
| | y r )r4 rz r{ r| r} rG rH )rI r4 rz r{ r| r} r rJ s rK rH zBarColumn.__init__ s; #
, ,&
l3rL r r c \ t |j t d|j ndt d|j | j dnt d| j |j
|j
| j | j | j | j S )z&Gets a progress bar widget for a task.Nr r* ) rt rF widthpulseanimation_timerz r{ r| r} ) r4 rt maxrF r4 startedrx rz r{ r| r} r s rK r zBarColumn.render s (,
(>#a$D!T^^, ..0$c!T^^6Lll" ==?** .. .. ((
rL )( rn ro rp rq N) re rf rg rh r r r6 r7 rH r4 r rk rl s @rK r r z s $&%$2$2!,)-4C=4 4 " 4
" 4
4 v&4
4
6
k
rL r c e Zd ZdZdddefdZy)TimeElapsedColumnzRenders time elapsed.r r rM c |j r |j n |j }|
t dd S t t | }t t
| d S )zShow time elapsed.-:--:--progress.elapsedrz )seconds)r
finished_timeelapsedr9 r r r )rI r rE deltas rK r zTimeElapsedColumn.render sL (,
$$$4<< ? );< <#g,/CJ&899rL Nre rf rg rh r9 r r rL rK r> r> s :6 :d :rL r> c e Zd ZdZ ddedededededee d ee
d
ed df fd
Z e d
ee
d efd Zddd efdZ xZS )r 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.
Nr$ text_format_no_percentagerz r% r& r' r r rM c L || _ || _ t |
|||||| y )N)r$ rz r% r& r' r )rI r rG rH )
rI r$ rI rz r% r& r' r r rJ s
rK rH zTaskProgressColumn.__init__ s9 *C&$
##%
rL r c |
t dd S t j t | g dd \ }}||z }t |d| dd S )zRender the speed in iterations per second.
Args:
task (Task): A Task object.
Returns:
Text: Text object containing the task speed.
r zprogress.percentagerB )r u ×10³u ×10⁶u ×10⁹u ×10¹² z.1fz it/s)r9 r+ pick_unit_and_suffixr )clsr unitsuffix
data_speeds rK render_speedz TaskProgressColumn.render_speed s_ ="78 8 44 J <
f
T\
z#&vhe43.0f}%r r1 r2 TNNF)re rf rg rh r r6 r/ r r r1 r7 rH classmethodri r9 rR r rk rl s @rK r r s K)+!!'-1)-
$'
k*
v&
. S% ST S S&
6
d
rL r c R e Zd ZdZdZ d dededee f fd
Zddd e fd
Z
xZ S ) r 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.
?compactr r c B || _ || _ t |
| y r )rX r rG rH )rI rX r r rJ s rK rH z TimeRemainingColumn.__init__ s$ %:"
l3rL r r rM c | j r|j r|j }d}n|j }d}|j
t d| S |!t | j rd| S d| S t t | d \ }}t |d \ }}| j r |s
|dd |d}n
|d
d |dd |d}t || S ) zShow time remaining.rA zprogress.remainingr rB z--:--r@ < 02d:d) r r rD time_remainingrt r9 rX divmodr )rI r task_timerz minutesrC hours formatteds rK r zTimeRemainingColumn.render s % %$-- **I&E ++I(E :: %( ( 4<<eL LYeL L "#i."5 ,w <<"3-q
6I )1WSM73-@IIU++rL )FFN) re rf rg rh r r r r7 rH r9 r rk rl s @rK r r sS K &+)- 44 $4 v& 4,6 ,d ,rL r c e Zd ZdZdddefdZy)FileSizeColumnzRenders completed filesize.r r rM c l t j t |j }t |d S )Show data completed.zprogress.filesizerB )r+ decimalr rF r9 rI r data_sizes rK r zFileSizeColumn.render, s) $$S %89 I%899rL NrG r rL rK rf rf ) s %:6 :d :rL rf c e Zd ZdZdddefdZy)TotalFileSizeColumnzRenders total filesize.r r rM c |j (t j t |j nd}t |d S )rh r zprogress.filesize.totalrB )rt r+ ri r r9 rj s rK r zTotalFileSizeColumn.render5 s4 9=9OH$$S_5UW I%>??rL NrG r rL rK rm rm 2 s !@6 @d @rL rm c D e Zd ZdZd dedee f fd
ZdddefdZ xZ
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 "/".
separatorr c 4 || _ t | | y r )rq rG rH )rI rq r rJ s rK rH zMofNCompleteColumn.__init__G s "
l3rL r r rM c t |j }|j t |j nd}t t | }t || d| j | d S )zShow completed/total.?r^ progress.downloadrB )r rF rt r r r9 rq )rI r rF rt total_widths rK r zMofNCompleteColumn.renderK sc ' #'::#9DJJs#e*o + a( (8 @%
rL )/N) re rf rg rh r r r7 rH r9 r rk rl s @rK rp rp ; s3 4# 48F;K 4
6
d
rL rp c J e Zd ZdZ d
dedee ddf fd
Zdddefd Z xZ
S ) r zRenders 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.
N binary_unitsr rM c 4 || _ t | | y r )ry rG rH )rI ry r rJ s rK rH zDownloadColumn.__init__] s )
l3rL r r c t |j }|j t |j n|}| j r t j
|g dd \ }}n t j
|g dd \ }}|dk( rdnd}||z }|d| d}|j #t |j } | |z }
|
d| d} nd } | d
| d | } t
| d
}
|
S )z.Calculate common unit for completed and total.) r KiBMiBGiBTiBPiBEiBZiBYiBi ) r kBMBGBTBPBEBZBYBrL r* r z,.frt rw r" ru rB )r rF rt ry r+ rM r9 )rI r rF unit_and_suffix_calculation_baserO rP precisioncompleted_ratio
completed_strrt total_ratio total_strdownload_status
download_texts rK r zDownloadColumn.renderc s ' $zz 5C
O9 ) #880Q LD& $880I LD&
A #d**2i[/:
:: !
OE $,K&r)Ao6II *O1YKqA _4GH
rL )FN) re rf rg rh r r r7 rH r9 r rk rl s @rK r r V sD LP4 48@8H4
4 ! 6 ! d ! rL r c e Zd ZdZdddefdZy)TransferSpeedColumnz&Renders human readable transfer speed.r r rM c |j xs |j }|
t dd S t j t | }t | dd S )zShow data transfer speed.rt zprogress.data.speedrB z/s)rT r r9 r+ ri r )rI r r rQ s rK r zTransferSpeedColumn.render sR ##1tzz =#89 9 %%c%j1
zl"%-BCCrL NrG r rL rK r r s 0D6 Dd DrL r c ( e Zd ZU dZeed< eed< y)ProgressSamplez$Sample of progress for a given time.r rF N)re rf rg rh ri r r rL rK r r s . $rL r c e Zd ZU dZeed< eed< ee ed< eed< e ed< dZ
ee ed< d Z e ed
< e
e Zeeef ed < e
dd
d
Zee ed< e
dd
d
Zee ed< dZee ed< e
d d
d
Zee ed< e
d
e Zeed< defdZede fd Z edee fd Z edee fd Z ede fd Z edefd Z edee fd Z!edee fd Z"d!d Z#y)"r zInformation regarding a progress task.
This object should be considered read-only outside of the :class:`~Progress` class.
r rs rt rF _get_timeNrD Tvisible)default_factoryfieldsF)defaultinitrepr
start_time stop_timerT c t d S )NrL )maxlenr r rL rK z
Task. s T 2rL )r r r _progress)r r _lockrM c " | j S )z(float: Get the current time, in seconds.)r rY s rK rx z
Task.get_time s ~~ rL c | j duS )z#bool: Check if the task as started.N)r rY s rK r; z Task.started s d**rL c N | j y| j | j z
S )zPOptional[float]: Get the number of steps remaining, if a non-None total was set.N)rt rF rY s rK remainingzTask.remaining s$ :: zzDNN**rL c | j y| j | j | j z
S | j | j z
S )z]Optional[float]: Time elapsed since task was started, or ``None`` if the task hasn't started.N)r r rx rY s rK rE z Task.elapsed sC ?? " >> %>>DOO3 3}} 00rL c | j duS )z Check if the task has finished.N)rD rY s rK r z
Task.finished s !!--rL c | j sy| j | j z dz }t dt d| }|S )zOfloat: Get progress of task as a percentage. If a None total was set, returns 0 Y@)rt rF minr: )rI rF s rK
percentagezTask.percentage s> zz^^djj0E9 s3 23 rL c J | j y| j 5 | j }|s
ddd y|d j |d j z
}|dk( r
ddd yt | }t | t
d |D }||z }|cddd S # 1 sw Y yxY w)z=Optional[float]: Get the estimated speed in steps per second.Nr r c 3 4 K | ] }|j y wrV r ).0samples rK z Task.speed.. s !O v&"2"2 )r r r r iterr r )rI r@
total_time
iter_progresstotal_completedr s rK r z
Task.speed s ?? "
ZZ~~H Z "" //(1+2G2GGJQ
Z !NM !!O !O OO#j0E ZZs B%B!.BB"c x | j ry| j }|sy| j }|yt ||z }|S )zJOptional[float]: Get estimated time to completion, or ``None`` if no data.r N)r r r r )rI r r estimates rK r_ zTask.time_remaining sC ==
NN E)*rL c T | j j d| _ d| _ y)zReset progress.N)r clearrD rT rY s rK _resetz Task._reset s" ! "rL rd )$re rf rg rh r; r r r ri GetTimeCallablerD r r r dictr r r r r rT r r r r r rx r r; r rE r r r r_ r r rL rK r r s JG' E?>* +%)M8E?)(GTJ "4 8FDcN8?"'5u"MJMN!&t%e!LIxLN&*NHUO*>',2U(Iu^$ eU;E5; % + + + +8E? + + 1% 1 1 .$ . . E x " #rL r c e Zd ZdZddddddddddd
deeef d ee d
e d e
d e
d
e de de dee de de ddfdZ e
deedf fd Zedefd Zedee fd Zedee fd Zede fd ZdOdZdOd ZdPd Zd eee d ee d ee ddfd!Z dQd"ee e e!e f d#ee
d$ee d%ed&e
de e f d'Z" dRdd(d)d*e#d#ee$ d$ee d%ede#f
d+Z%e&jN dSddd(d,d*eed-e(f d.e)d/ d0e$d1ee d2ee d3ee d#ee$ d$ee d%ede#fd4 Z*e&jN dSddd(d,d*eed-e(f d.ee)d5 e)d6 f d0e$d1ee d2ee d3ee d#ee$ d$ee d%ede+fd7 Z* dTddd(d,d*eed-e(f d.ee)d/ e)d6 e)d5 f d0e$d1ee d2ee d3ee d#ee$ d$ee d%edee#e+f fd8Z*d$eddfd9Z,d$eddfd:Z-ddddddd;d$ed#ee
dee d?e d@e.ddfdAZ/dddBdddCd$edDe d#ee
dee d%ee d@e.ddfdEZ0dUd$ed=e
ddfdFZ1dOdGZ2de3fdHZ4de e3 fdIZ5dJe e de6fdKZ7de3fdLZ8 dVd%edDe d#ee
de d@e.defdMZ9d$eddfdNZ:y)WrA 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.
NTrm g >@F)
rv ru ry speed_estimate_periodrw redirect_stdoutredirect_stderrrx r~ expandr rv ru ry r rw r r rx r~ r rM c
|dkD sJ d t | _ | xs | j | _ || _ | | _ |
| _ i | _ t d | _ t |xs
t |||||| j | _
|xs | j j | _ | j j | _ | j j" | _ y )Nr z refresh_per_second must be > 0)rv ru ry rw r r get_renderable)r r get_default_columnsr r r~ r _tasksr; _task_indexr3 r, r liverv rx printlog) rI rv ru ry r rw r r rx r~ r r s rK rH zProgress.__init__* s "A%G'GG%W
<$":":"< %:" *, #)!9,{}% 1 + + ..
!9DLL$9$9
\\''
<<##rL .c P 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.
r )r r r r )rN s rK r z Progress.get_default_columnsN s( 2
A B K !
rL c . | j j S rV )r rv rY s rK rv zProgress.consolem s yy rL c | j 5 t | j j cddd S # 1 sw Y yxY w)z Get a list of Task instances.N)r listr valuesrY s rK taskszProgress.tasksq s* ZZ **,- ZZ #:Ac | j 5 t | j j cddd S # 1 sw Y yxY w)zA list of task IDs.N)r r r keysrY s rK task_idszProgress.task_idsw s* ZZ ((*+ ZZr c | j 5 | j s
ddd yt d | j j D cddd S # 1 sw Y yxY w)z'Check if all tasks have been completed.NTc 3 4 K | ] }|j y wrV )r )r r s rK r z$Progress.finished.. s F1Et}}1Er )r r allr rY s rK r zProgress.finished} sC ZZ;; Z F1C1C1EFF ZZs A)AA c V | j s | j j d yy)zStart the progress display.T)rO N)r~ r rX rY s rK rX zProgress.start s || IIOODO ) rL c | j j | j j s| j j yy)zStop the progress display.N)r r rv is_interactiver rY s rK r z
Progress.stop s2 ||** LL +rL c & | j | S rV rW rY s rK rZ zProgress.__enter__ r[ rL r\ r] r^ c $ | j y rV )r rb s rK rc zProgress.__exit__ s
rL rr rt rB rs rC c # K |t t | xs d}|| j || }n| j || | j j
r8t
| || 5 }|D ] }| |xj dz
c_ ddd y| j }| j } |D ] }| ||d | y# 1 sw Y yxY ww)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.
Nrt r* )
ri r add_taskrR r ru r? rF rP rO )
rI rr rt rB rs rC track_threadvaluerP rO s
rK r zProgress.track s ( =+h/08DE ?mmKum=G KKuK - 99 ! ! dG];| %E K **a/* & <;
llGllG! # " <;s A'C)!C =CC
Cr )rB rs r c d}||}n0|.| j 5 | j | j }ddd | t d || j || }n| j || t
|| |d S # 1 sw Y NxY w)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'r Fr )r r rt
ValueErrorr rR r )rI r rt rB rs total_bytess rK r zProgress.wrap_file s . (, K
"kk' 2 8 8 Q
?mmK{mCG KK{K 3tT7?? s A==B)rt rB rs r r r r r r r c y rV r
rI r r r r r r rt rB rs s
rK r z
Progress.open
rL r r c y rV r r s
rK r z
Progress.open r rL c dj t |d }
|
dvrt dj | |dk( } |
dk( r"|dk( r t j
dt d }n|
d
v r|d k( r t d |dk( rd }|t | j }|| j | |
}n| j ||
t j |d| } t | | |d }
|dv rt j |
|||| 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/Or r )r Tr )r r )r r r line_buffering)ra sortedr r. warningswarnRuntimeWarningr st_sizer rR ior r
TextIOWrapper)rI r r r r r r rt rB rs _moder r r s rK r z
Progress.open s( B tU34 ) ) 0 7 7 => > #a D=Y!^ MMs
I
k
!A~ !ABBa =J&&E ?mmKum=G KKuK - ty9wTB ; ##! -
rL c | j 5 | j | }|j | j |_ ddd y# 1 sw Y yxY w)zStart 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 rx )rI rB r s rK
start_taskzProgress.start_task\ s: ZZ;;w'D&"&--/ ZZs 1AAc | j 5 | j | }| j }|j ||_ ||_ ddd y# 1 sw Y yxY w)zStop a task.
This will freeze the elapsed time on the task.
Args:
task_id (TaskID): ID of task.
N)r r rx r r )rI rB r r s rK stop_taskzProgress.stop_taskj sF ZZ;;w'D ==?L&". )DN ZZs :AA)rt rF rP rs r rO rF rP r rO r c < | j 5 | j | } | j }
|&|| j k7 r|| _ | j || xj |z
c_ ||| _ ||| _ ||| _ | j j | | j |
z
} | j } | | j z
}
| j }|j }|r.|d j |
k r | |r|d j |
k r | dkD r|j t | | | j 6| j | j k\ r | j | j" | _ ddd |r| j% yy# 1 sw Y xY w)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 rF rt r rs r r rR rx r r popleftr appendr rD rE rO )rI rB rt rF rP rs r rO r r completed_startupdate_completedr old_sample_timer r s rK rR zProgress.updatey sq 0 ZZ;;w'D "nnO Udjj%8 "
"')$!*&#. " & KK v & #~~ ? ==?L *T-G-G GO I ''G ! 6 6 H ! 6 6 H !# >N!OP
&NNdjj0&&.%)\\"A D LLN E Zs DFA"FFr )rX rt rF r rs rX c > | j }| j 5 | j | } | j |r|nd| _ ||| _ || _ ||| _ |r|| _ ||| _ d| _
ddd | j y# 1 sw Y xY w)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) rx r r r r rt rF r r rs rD rO )
rI rB rX rt rF r rs r r r s
rK resetzProgress.reset s , }}
ZZ;;w'D KKM.3lDO "
&DN" & $ &#. !%D
Zs A BB c | j }| j 5 | j | }|j }|xj |z
c_ |j |z
}|| j z
}|j
}|j } |r.|d j |k r | |r|d j |k r t | dkD r | t | dkD r|j t || |j G|j |j k\ r.|j "|j |_ |j |_ ddd y# 1 sw Y yxY w)zAdvance task by a number of steps.
Args:
task_id (TaskID): ID of task.
advance (float): Number of steps to advance. Default is 1.
r rL N)rx r r rF r r r r r r r rt rD rE r rT )
rI rB rP r r r r r r r s
rK rP zProgress.advance s }}
ZZ;;w'D "nnO NNg %N #~~ ? *T-G-G GO I ''G ! 6 6 H ! 6 6 Hi.4' i.4' ^L:J K L
&NNdjj0&&.%)\\"&*jj#) ZZs BE4#EA.EEc | j s2| j j r| j j yyy)z*Refresh (render) the progress information.N)r~ r
is_startedrO rY s rK rO zProgress.refresh s, || 4 4 II !5|rL c 2 t | j }|S )z*Get a renderable for the progress display.)r. get_renderables)rI r s rK r zProgress.get_renderable s D 0 0 23
rL c # J K | j | j }| yw)z5Get a number of renderables for the progress display.N)make_tasks_tabler )rI tables rK r
zProgress.get_renderables s %%djj1 s !#r c d | j D }t j |d| j d}|D ]1 j s |j
fd| j D 3 |S )zGet 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 3 K | ]> }t |t r t d n |j j @ yw)Tr) N)r r r7 r copy)r _columns rK r z,Progress.make_tasks_table.. sG
( gs+ t$ --/4467 (s AA)r r* )paddingr c 3 r K | ]. }t |t r|j n | 0 yw)r- N)r r r. )r columnr s rK r z,Progress.make_tasks_table.. s@ '3F *&# 6 #MMtM 4!' . '3s 47)r r8 gridr r add_row)rI r
table_columnsr r s @rK r zProgress.make_tasks_table
si
<<
M6$++ND||
'+ll
rL c f | j 5 | j cddd S # 1 sw Y yxY w)z+Makes the Progress class itself renderable.N)r r rY s rK __rich__zProgress.__rich__+ s
ZZ&&( ZZs '0c | j 5 t | j |||||| j | j }|| j | j < |r| j | j | j }t
t | j dz | _ ddd | j S # 1 sw Y xY w)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 rx r r r; r rO ) rI rs rX rt rF r r r new_task_indexs rK r zProgress.add_task0 s 0 ZZ --jj D -1DKK(( ) 0 01 ! - -N %c$*:*:&;a&? @D
# Zs BCC
c ` | j 5 | j |= ddd y# 1 sw Y yxY w)z]Delete a task if it exists.
Args:
task_id (TaskID): A task ID.
N)r r )rI rB s rK remove_taskzProgress.remove_task[ s! ZZ G$ ZZs $-rd )rM rA )NN
Working...皙?rV r NNNr r NNN)r* )Tr r T);re rf rg rh r& r r r r- r ri r rH rU r# r r rv r r r r; r r rX r rZ r$ rj r rc r r< r! r r r r typingoverloadr r) r r" r r r rR r rP rO r0 r r
r8 r r r r r rL rK rA rA s
$ &* !$&'+ $ $.2 "$^+,"$ '""$ "$
" "$ %
"$ "$ "$ "$ ?+"$ "$ "$
"$H
E.#*=$>
< ! ! ! .tDz . .
,$v, , ,
G$ G G*
! 4
./ -( '
"&$(' "
' . 0FFG' ' &! '
'
'
, 'X $(@
%)'
(@(@ }(@
&! (@
(@
(@T __
"& $!%
$$('
C %/0
dm
3-
#
}
&!
__
"& $!%
$$('
C %/0
GCL'$-/0
3-
#
}
&!
$ CF"& $!%K $$('KC %/0K GDM74='#,>?K K
3- K
K # K }K &!K K
x KZ 2& 2T 2
*
*D
*& "&%)#'%)"& ;; ;
E? ; %
; c]; $ ; ; ;
;B !%"&%)%% %
%
% $ % c]% %
%N 1v 1 1d 1<
.!9
htn B). ) !&
) ) ) )
)
) )
) V%6 %d %rL rA __main__)Panel)Rule)Syntax)r8 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_valuepython) line_numbersfoobarbaz123z: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...examplezPretty printed)typer z Syntax...zGive it a try!)cycle)record)rv rw z[red]DownloadingrL r z[green]Processingz[yellow]ThinkingrW r g333333?g{Gz?d )r! NTNFNrm rn ro rp rq r" FTr# r$ )r sysr% r abcr r collectionsr 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) pip._vendor.typing_extensionsr r+ r, rv r- r. r/ r0 r' r1 jupyterr2 r r3 progress_barr4 r r5 rz r6 r r7 r8 r r9 r: r r; r< ri r r= r? r r r r r r r& r r r r r r r r> r r rf rm rp r r r r rA re randomtimepanelr( ruler) syntaxr* r progress_renderables itertoolsr5 examplesr r@ r task1task2task3r rR sleeprandintr r r rL rK rS sT
# ( . * * * v 5 # B B $ ! % 3 ~& 2u9% V^^D&(+%6 %T $ ! !%.2 "! . . ( C
H\*H\,BBCC
C
E?C
C
g
C
C
xE *+C
C
C
C
C
C
C
C
C
l!C
La,i a,H8>"%wr{ 82 $ !%.2 "! . . ( >*
>* >* >*
>* g
>* >* xE *+>* >* >* >* >* >* >* H >*B " !
# !%.2 "! . . ('
_e+
,
ws|+
, sm
SM c]
C= g
xE *+ ! " # $ % & ' ( F) 0 " !
# !%.2 "! . . ('
_e+
,
$- sm
SM c]
C= g
xE *+ ! " # $ % & ' ( H ) 4 ?B " !
S* # !%.2 "! . . ('S*
_e+
,S*
wt}gcl:
;S* S* sm S*
SM S* c]
S* C=S* S* S* g
S* S* xE *+S* S* S* !S*" #S*$ %S*& 'S*( >( #^F%; ;<)S*l'1S '1T ~ "/N /d <'
'
T : :F FR-,. -,`:^ :@. @
6. ^ . b D. D%Z % z# z# z#zI %| I %X z
" F
% &E MM#sC E
89'
&$45
)*HT"G
% % '
!!"4D!A!!"5T!B!!"4D!A## OOE3O / OOE3O / DJJt v~~a%) T(^, ##
_ ^
s
?B>UU