o
3g @ s d Z ddlmZ ddlZddlZddlZddlmZmZ ddl m
Z
G dd deZ G dd d e Z G d
d d e Z
G d d
d
e Zee
ee dd ZedkrYe dS dS )z
Progress bar
------------
) annotationsN)ABCabstractmethod)get_terminal_sizec @ s e Zd ZdZ d"ddZdd Zd d
Zed d Zd
d Z dd Z
e dd Z e j
dd Z edd Zdd Zdd Zdd Zed d Zed d Zed#d d!ZdS )$ ProgressBasea Base class for progress bars. Customize for types of progress bars.
:param iterator: The iterator to wrap with a progress bar
:param length: The length of the iterator (will use ``__len__`` if None)
:param timer: Try to time the completion status of the iterator
:param body: True if the slow portion occurs outside the iterator (in a loop, for example)
:param has_output: True if the iteration body produces output to the screen (forces rewrite off)
:param clear: Clear the progress bar afterwards, if applicable.
NTFc C sd |d u r t |}n|d u rt|}n |d u r |d u r td|| _|| _|| _|| _|| _|| _d S )Nz'Expected either an iterator or a length) lenrange TypeErrorlengthiteratortimerbody
has_outputclear)selfr r
r r
r r r L/home/garg/my-data/venv/lib/python3.10/site-packages/plumbum/cli/progress.py__init__ s
zProgressBase.__init__c C | j S N)r
r r r r __len__2 s zProgressBase.__len__c C s | | S r )startr r r r __iter__5 s zProgressBase.__iter__c C s, t | j| _ | jr dnd| _tj | _dS )z8This should initialize the progress bar and the iteratorr N)iterr r
valuedatetimenow _start_timer r r r r 9 s zProgressBase.startc C s2 z t | j}| W |S ty | w r )nextr increment
StopIterationdone)r rvalr r r __next__@ s
zProgressBase.__next__c C s t | S r )r r r r r r I s zProgressBase.nextc C r )zHThis is the current value, as a property so setting it can be customized_valuer r r r r L zProgressBase.valuec C s
|| _ d S r r& r valr r r r Q s
c C dS )z!Called to update the progress barNr r r r r displayU zProgressBase.displayc C s | j d7 _ | dS )z$Sets next value and displays the bar N)r r, r r r r r! Y s zProgressBase.incrementc C sh | j dk rdS tj | j }|jd d d |j |jd | j }|| j| j }|td|dfS )z4Get the time remaining for the progress bar, guessesr. )NN < g .Ar ) r r r r daysseconds microsecondsr
timedelta)r elapsed_time time_eachtime_remainingr r r r7 ^ s
zProgressBase.time_remainingc C sP | j dk rdS ttt| \}}|dd }|dd }| d| dS )z*Returns a string version of time remainingr. z$Starting... .r z completed, z
remaining)r listmapstrr7 split)r r5 r7 completed remainingr r r str_time_remainingk s
z ProgressBase.str_time_remainingc C r+ )z$Is called when the iterator is done.Nr r r r r r# u r- zProgressBase.donec O s | t | fddi|S )zMFast shortcut to create a range based progress bar, assumes work done in bodyr
T)r )clsr kargsr r r r y s zProgressBase.rangec K s | ||fddi|S )zEShortcut to wrap an iterator that does not do all the work internallyr
Tr )r@ r r
rA r r r wrap~ s zProgressBase.wrap)NNTFFTr )__name__
__module__ __qualname____doc__r r r r r r% r propertyr setterr, r! r7 r? r# classmethodr rB r r r r r s<
r c s4 e Zd Z fddZdd Zdd Zdd Z ZS ) Progressc s t | d S r )superr r, r __class__r r r s
zProgress.startc C sX | j | _| | jr | js tjdtt | d d ntjd tj
d S )N
) r
r r, r r sysstdoutwriter r; flushr r r r r# s " z
Progress.donec C s( t ddd }| jdkrd| _dS t| jd| j }d| jr"| n | j d| j d }|t| dk s8| jrNd| _| jrH|d d
| S |d dS |t| d d | _t || j }d
d| d| j| d | }d|d d}|d | jd d | || jd t| d d S )N)r r )defaultr z 0/0 completerO z of z complete
z.0%z complete: r. [*])
r r
widthmaxr r r? r r int)r r[ percentendingnstarspbar str_percentr r r __str__ s2
" zProgress.__str__c C sP t | }| jdks | jrtj|d n tjd tj| tj d S )Nr rP rN )r; r[ r rQ rR rS rT )r disptxtr r r r, s zProgress.display)rC rD rE r r# rc r,
__classcell__r r rL r rJ s
rJ c sV e Zd ZdZ fddZ fddZedd Zejdd Zd d
Z d d Z
Z S )
ProgressIPyzic s t - t d z ddlm}m}m} W n ty) ddlm}m}m} Y nw W d n1 s4w Y t j
|i | || j d| _ | | _
|| j | j
f| _d S )Nignorer )HTMLHBox IntProgress)r\ )warningscatch_warnings simplefilter
ipywidgetsrh ri rj ImportErrorIPython.html.widgetsrK r r
prog_label_box)r argsrA rh ri rj rL r r r s
zProgressIPy.__init__c s$ ddl m} || j t d S )Nr )r, )IPython.displayr, rs rK r )r r, rL r r r s
zProgressIPy.startc C r )zGThis is the current value, -1 allowed (automatically fixed for display)r& r r r r r r( zProgressIPy.valuec C sV || _ t|d| j_| j| j d| j_| jr'|dkr)| j| | j
_d S d S d S )Nr z.2%) r' r\ rq r r
descriptionr HTMLBOXformatr? rr r) r r r r s c C s d S r r r r r r r, s zProgressIPy.displayc C s | j r
| j d S d S r )r rs closer r r r r# s zProgressIPy.done) rC rD rE rw r r rG r rH r, r# re r r rL r rf s
rf c @ s e Zd ZdZdd ZdS ) ProgressAutoa< Automatically selects the best progress bar (IPython HTML or text). Does not work with qtconsole
(as that is correctly identified as identical to notebook, since the kernel is the same); it will still
iterate, but no graphical indication will be displayed.
:param iterator: The iterator to wrap with a progress bar
:param length: The length of the iterator (will use ``__len__`` if None)
:param timer: Try to time the completion status of the iterator
:param body: True if the slow portion occurs outside the iterator (in a loop, for example)
c O s z/t zddlm} W n ty ddlm} Y nw z t|i |W W S |y/ t dw ttfyA t|i | Y S w )z_Uses the generator trick that if a cls instance is returned, the __init__ method is not called.r )
TraitErrorN) __IPYTHON__ traitletsr{ ro IPython.utils.traitletsrf NameErrorrJ )r@ rt rA r{ r r r __new__ s zProgressAuto.__new__N)rC rD rE rF r r r r r rz s
rz c C s* dd l } td}|D ]}| d q d S )Nr r. )timerJ r sleep)r tst_r r r main s
r __main__)rF
__future__r r rQ rk abcr r plumbum.cli.termsizer r rJ rf rz registerr rC r r r r s" t8,