U
ʗRe @ s, d dl Z d dlmZ d dlmZ d dlmZmZmZ ddl m
Z
m Z ddl m
Z
ddlmZmZmZ dd lmZ dd
lmZ dd lmZ dd lmZmZ d
ZG dd deZ e dkr(e Ze dddZ d dlZe d e d ddD ].Z!e "e! e#e ej$%d e&d qe d e# dS ) N) lru_cache) monotonic)IterableListOptional )Color blend_rgb) ColorTriplet)ConsoleConsoleOptions RenderResult) JupyterMixin) Measurement)Segment)Style StyleType c @ s e Zd ZdZd"ee eee eeeeeee d
d d Z e
d
ddZ e ee d
ddZ
eddd#eee
eeee dddZd$eee ddddZd%eeeee ddd Zeeed d d Zeeed d d!ZdS )& ProgressBara Renders a (progress) bar. Used by rich.progress.
Args:
total (float, optional): Number of steps in the bar. Defaults to 100. Set to None to render a pulsing animation.
completed (float, optional): Number of steps completed. Defaults to 0.
width (int, optional): Width of the bar, or ``None`` for maximum width. Defaults to None.
pulse (bool, optional): Enable pulse effect. Defaults to False. Will pulse if a None total was passed.
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".
animation_time (Optional[float], optional): Time in seconds to use for animation, or None to use system time.
Y@r NFbar.back bar.complete bar.finished bar.pulse) total completedwidthpulsestylecomplete_stylefinished_style pulse_styleanimation_timec
C s@ || _ || _|| _|| _|| _|| _|| _|| _| | _d | _ d S )N)
r r r r r r r r! r" _pulse_segments)
selfr r r r r r r r! r" r% /builddir/build/BUILDROOT/alt-python38-pip-22.2.1-2.el8.x86_64/opt/alt/python38/lib/python3.8/site-packages/pip/_vendor/rich/progress_bar.py__init__! s zProgressBar.__init__)returnc C s d| j d| jdS )Nzr r )r$ r% r% r& __repr__9 s zProgressBar.__repr__c C s2 | j dkrdS | j| j d }tdtd|}|S )z Calculate percentage complete.Nr d g )r r minmax)r$ r r% r% r& percentage_completed< s
z ProgressBar.percentage_completed )maxsize)
fore_style
back_style color_systemno_colorasciir( c C s |rdnd}g }|dks |r\|t ||gtd 7 }|t |r@dn||gttd 7 }|S |j}|jrr|j n
tddd} |jr|j n
tddd}
tj} tj} t }
t }t
j }t tD ]J}|t }d| || d d }t
| |
|d
}||
||||d q|S ) zGet a list of segments to render a pulse animation.
Returns:
List[Segment]: A list of segments, one segment per character.
- ━)standard eight_bit truecolor r g ?g @)
cross_fade)color)r
PULSE_SIZEappendr@
get_truecolorr
mathcospir r from_tripletranger )r$ r2 r3 r4 r5 r6 barsegmentsrB
fore_color
back_colorrE rF _Segment_StylerG indexpositionfader@ r% r% r& _get_pulse_segmentsE s8
z ProgressBar._get_pulse_segments)r r r( c C s || _ |dk r|n| j| _dS )zUpdate progress with new values.
Args:
completed (float): Number of steps completed.
total (float, optional): Total number of steps, or ``None`` to not change. Defaults to None.
Nr* )r$ r r r% r% r& updatet s zProgressBar.update)consoler r6 r( c c s |j | jdd}|j | jdd}| j|||j|j|d}t|}| jdkrPt n| j}|t || d } t | d | }
| |
|
| } | E dH dS )a) Renders the pulse animation.
Args:
console (Console): Console instance.
width (int): Width in characters of pulse animation.
Returns:
RenderResult: [description]
Yields:
Iterator[Segment]: Segments to render pulse
white)defaultblackr6 Nr< )
get_styler! r rR r4 r5 lenr" r int) r$ rT r r6 r2 r3 pulse_segments
segment_count current_timerJ offsetr% r% r&
_render_pulse~ s zProgressBar._render_pulse)rT optionsr( c c s t | jp |j|j}|jp |j}| jp.| jd k}|rN| j|||dE d H d S | jd k rlt | jtd| j nd }|rxdnd}|rdnd}|rdnd} | jr|d k rt
|d | | j n|d }
|
d } |
d } | | j }
| jd kp| j | jk}| |r| j
n| j}t}| r$||| |V | r:||| |V |js|| | }|r|jd k r| s| r|| |
V |d 8 }|r||| |
V d S )
NrX r r7 r8 r= u ╸u ╺r< r )r- r max_widthlegacy_windows
ascii_onlyr r ra r. r r\ rZ r r r r r5 r4 )r$ rT rb r r6 should_pulser rI half_bar_right
half_bar_leftcomplete_halves bar_counthalf_bar_countr is_finishedr rM remaining_barsr% r% r& __rich_console__ sJ z ProgressBar.__rich_console__c C s$ | j d k rt| j | j S td|jS )N )r r rc )r$ rT rb r% r% r& __rich_measure__ s
z ProgressBar.__rich_measure__) r r NFr r r r N)F)N)F) __name__
__module__ __qualname____doc__r floatr\ boolr r' strr+ propertyr/ r r r r rR rS r r ra r r
rn r rp r% r% r% r& r sd . . r __main__2 r, )r r Fe
g?T)'rD functoolsr timer typingr r r r@ r r
color_tripletr
rT r r r
jupyterr measurer segmentr r r r rA r rq rI show_cursorrH nrS printfilewritesleepr% r% r% r& s2 B