385 lines
14 KiB
Python
385 lines
14 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: iso-8859-1 -*-
|
|
#
|
|
# progressbar - Text progressbar library for python.
|
|
# Copyright (c) 2005 Nilton Volpato
|
|
#
|
|
# This library is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation; either
|
|
# version 2.1 of the License, or (at your option) any later version.
|
|
#
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this library; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
|
|
"""Text progressbar library for python.
|
|
|
|
This library provides a text mode progressbar. This is typically used
|
|
to display the progress of a long running operation, providing a
|
|
visual clue that processing is underway.
|
|
|
|
The ProgressBar class manages the progress, and the format of the line
|
|
is given by a number of widgets. A widget is an object that may
|
|
display diferently depending on the state of the progress. There are
|
|
three types of widget:
|
|
- a string, which always shows itself;
|
|
- a ProgressBarWidget, which may return a diferent value every time
|
|
it's update method is called; and
|
|
- a ProgressBarWidgetHFill, which is like ProgressBarWidget, except it
|
|
expands to fill the remaining width of the line.
|
|
|
|
The progressbar module is very easy to use, yet very powerful. And
|
|
automatically supports features like auto-resizing when available.
|
|
"""
|
|
|
|
from __future__ import division
|
|
|
|
__author__ = "Nilton Volpato"
|
|
__author_email__ = "first-name dot last-name @ gmail.com"
|
|
__date__ = "2006-05-07"
|
|
__version__ = "2.3-dev"
|
|
|
|
import sys, time, os
|
|
from array import array
|
|
try:
|
|
from fcntl import ioctl
|
|
import termios
|
|
except ImportError:
|
|
pass
|
|
import signal
|
|
try:
|
|
basestring
|
|
except NameError:
|
|
basestring = (str,)
|
|
|
|
class ProgressBarWidget(object):
|
|
"""This is an element of ProgressBar formatting.
|
|
|
|
The ProgressBar object will call it's update value when an update
|
|
is needed. It's size may change between call, but the results will
|
|
not be good if the size changes drastically and repeatedly.
|
|
"""
|
|
def update(self, pbar):
|
|
"""Returns the string representing the widget.
|
|
|
|
The parameter pbar is a reference to the calling ProgressBar,
|
|
where one can access attributes of the class for knowing how
|
|
the update must be made.
|
|
|
|
At least this function must be overriden."""
|
|
pass
|
|
|
|
class ProgressBarWidgetHFill(object):
|
|
"""This is a variable width element of ProgressBar formatting.
|
|
|
|
The ProgressBar object will call it's update value, informing the
|
|
width this object must the made. This is like TeX \\hfill, it will
|
|
expand to fill the line. You can use more than one in the same
|
|
line, and they will all have the same width, and together will
|
|
fill the line.
|
|
"""
|
|
def update(self, pbar, width):
|
|
"""Returns the string representing the widget.
|
|
|
|
The parameter pbar is a reference to the calling ProgressBar,
|
|
where one can access attributes of the class for knowing how
|
|
the update must be made. The parameter width is the total
|
|
horizontal width the widget must have.
|
|
|
|
At least this function must be overriden."""
|
|
pass
|
|
|
|
|
|
class ETA(ProgressBarWidget):
|
|
"Widget for the Estimated Time of Arrival"
|
|
def format_time(self, seconds):
|
|
return time.strftime('%H:%M:%S', time.gmtime(seconds))
|
|
def update(self, pbar):
|
|
if pbar.currval == 0:
|
|
return 'ETA: --:--:--'
|
|
elif pbar.finished:
|
|
return 'Time: %s' % self.format_time(pbar.seconds_elapsed)
|
|
else:
|
|
elapsed = pbar.seconds_elapsed
|
|
eta = elapsed * pbar.maxval / pbar.currval - elapsed
|
|
return 'ETA: %s' % self.format_time(eta)
|
|
|
|
class FileTransferSpeed(ProgressBarWidget):
|
|
"Widget for showing the transfer speed (useful for file transfers)."
|
|
def __init__(self, unit='B'):
|
|
self.unit = unit
|
|
self.fmt = '%6.2f %s'
|
|
self.prefixes = ['', 'K', 'M', 'G', 'T', 'P']
|
|
def update(self, pbar):
|
|
if pbar.seconds_elapsed < 2e-6:#== 0:
|
|
bps = 0.0
|
|
else:
|
|
bps = pbar.currval / pbar.seconds_elapsed
|
|
spd = bps
|
|
for u in self.prefixes:
|
|
if spd < 1000:
|
|
break
|
|
spd /= 1000
|
|
return self.fmt % (spd, u + self.unit + '/s')
|
|
|
|
class RotatingMarker(ProgressBarWidget):
|
|
"A rotating marker for filling the bar of progress."
|
|
def __init__(self, markers='|/-\\'):
|
|
self.markers = markers
|
|
self.curmark = -1
|
|
def update(self, pbar):
|
|
if pbar.finished:
|
|
return self.markers[0]
|
|
self.curmark = (self.curmark + 1) % len(self.markers)
|
|
return self.markers[self.curmark]
|
|
|
|
class Percentage(ProgressBarWidget):
|
|
"Just the percentage done."
|
|
def update(self, pbar):
|
|
return '%3d%%' % pbar.percentage()
|
|
|
|
class SimpleProgress(ProgressBarWidget):
|
|
"Returns what is already done and the total, e.g.: '5 of 47'"
|
|
def __init__(self, sep=' of '):
|
|
self.sep = sep
|
|
def update(self, pbar):
|
|
return '%d%s%d' % (pbar.currval, self.sep, pbar.maxval)
|
|
|
|
class Bar(ProgressBarWidgetHFill):
|
|
"The bar of progress. It will stretch to fill the line."
|
|
def __init__(self, marker='#', left='|', right='|'):
|
|
self.marker = marker
|
|
self.left = left
|
|
self.right = right
|
|
def _format_marker(self, pbar):
|
|
if isinstance(self.marker, basestring):
|
|
return self.marker
|
|
else:
|
|
return self.marker.update(pbar)
|
|
def update(self, pbar, width):
|
|
percent = pbar.percentage()
|
|
cwidth = width - len(self.left) - len(self.right)
|
|
marked_width = int(percent * cwidth // 100)
|
|
m = self._format_marker(pbar)
|
|
bar = (self.left + (m * marked_width).ljust(cwidth) + self.right)
|
|
return bar
|
|
|
|
class ReverseBar(Bar):
|
|
"The reverse bar of progress, or bar of regress. :)"
|
|
def update(self, pbar, width):
|
|
percent = pbar.percentage()
|
|
cwidth = width - len(self.left) - len(self.right)
|
|
marked_width = int(percent * cwidth // 100)
|
|
m = self._format_marker(pbar)
|
|
bar = (self.left + (m*marked_width).rjust(cwidth) + self.right)
|
|
return bar
|
|
|
|
default_widgets = [Percentage(), ' ', Bar()]
|
|
class ProgressBar(object):
|
|
"""This is the ProgressBar class, it updates and prints the bar.
|
|
|
|
A common way of using it is like:
|
|
>>> pbar = ProgressBar().start()
|
|
>>> for i in xrange(100):
|
|
... # do something
|
|
... pbar.update(i+1)
|
|
...
|
|
>>> pbar.finish()
|
|
|
|
You can also use a progressbar as an iterator:
|
|
>>> progress = ProgressBar()
|
|
>>> for i in progress(some_iterable):
|
|
... # do something
|
|
...
|
|
|
|
But anything you want to do is possible (well, almost anything).
|
|
You can supply different widgets of any type in any order. And you
|
|
can even write your own widgets! There are many widgets already
|
|
shipped and you should experiment with them.
|
|
|
|
The term_width parameter must be an integer or None. In the latter case
|
|
it will try to guess it, if it fails it will default to 80 columns.
|
|
|
|
When implementing a widget update method you may access any
|
|
attribute or function of the ProgressBar object calling the
|
|
widget's update method. The most important attributes you would
|
|
like to access are:
|
|
- currval: current value of the progress, 0 <= currval <= maxval
|
|
- maxval: maximum (and final) value of the progress
|
|
- finished: True if the bar has finished (reached 100%), False o/w
|
|
- start_time: the time when start() method of ProgressBar was called
|
|
- seconds_elapsed: seconds elapsed since start_time
|
|
- percentage(): percentage of the progress [0..100]. This is a method.
|
|
|
|
The attributes above are unlikely to change between different versions,
|
|
the other ones may change or cease to exist without notice, so try to rely
|
|
only on the ones documented above if you are extending the progress bar.
|
|
"""
|
|
|
|
__slots__ = ('currval', 'fd', 'finished', 'last_update_time', 'maxval',
|
|
'next_update', 'num_intervals', 'seconds_elapsed',
|
|
'signal_set', 'start_time', 'term_width', 'update_interval',
|
|
'widgets', '_iterable')
|
|
|
|
_DEFAULT_MAXVAL = 100
|
|
|
|
def __init__(self, maxval=None, widgets=default_widgets, term_width=None,
|
|
fd=sys.stderr):
|
|
self.maxval = maxval
|
|
self.widgets = widgets
|
|
self.fd = fd
|
|
self.signal_set = False
|
|
if term_width is not None:
|
|
self.term_width = term_width
|
|
else:
|
|
try:
|
|
self._handle_resize(None, None)
|
|
signal.signal(signal.SIGWINCH, self._handle_resize)
|
|
self.signal_set = True
|
|
except (SystemExit, KeyboardInterrupt):
|
|
raise
|
|
except:
|
|
self.term_width = int(os.environ.get('COLUMNS', 80)) - 1
|
|
|
|
self.currval = 0
|
|
self.finished = False
|
|
self.start_time = None
|
|
self.last_update_time = None
|
|
self.seconds_elapsed = 0
|
|
self._iterable = None
|
|
|
|
def __call__(self, iterable):
|
|
try:
|
|
self.maxval = len(iterable)
|
|
except TypeError:
|
|
# If the iterable has no length, then rely on the value provided
|
|
# by the user, otherwise fail.
|
|
if not (isinstance(self.maxval, (int, long)) and self.maxval > 0):
|
|
raise RuntimeError('Could not determine maxval from iterable. '
|
|
'You must explicitly provide a maxval.')
|
|
self._iterable = iter(iterable)
|
|
self.start()
|
|
return self
|
|
|
|
def __iter__(self):
|
|
return self
|
|
|
|
def next(self):
|
|
try:
|
|
next = self._iterable.next()
|
|
self.update(self.currval + 1)
|
|
return next
|
|
except StopIteration:
|
|
self.finish()
|
|
raise
|
|
|
|
def _handle_resize(self, signum, frame):
|
|
h, w = array('h', ioctl(self.fd, termios.TIOCGWINSZ, '\0' * 8))[:2]
|
|
self.term_width = w
|
|
|
|
def percentage(self):
|
|
"Returns the percentage of the progress."
|
|
return self.currval * 100.0 / self.maxval
|
|
|
|
def _format_widgets(self):
|
|
r = []
|
|
hfill_inds = []
|
|
num_hfill = 0
|
|
currwidth = 0
|
|
for i, w in enumerate(self.widgets):
|
|
if isinstance(w, ProgressBarWidgetHFill):
|
|
r.append(w)
|
|
hfill_inds.append(i)
|
|
num_hfill += 1
|
|
elif isinstance(w, basestring):
|
|
r.append(w)
|
|
currwidth += len(w)
|
|
else:
|
|
weval = w.update(self)
|
|
currwidth += len(weval)
|
|
r.append(weval)
|
|
for iw in hfill_inds:
|
|
widget_width = int((self.term_width - currwidth) // num_hfill)
|
|
r[iw] = r[iw].update(self, widget_width)
|
|
return r
|
|
|
|
def _format_line(self):
|
|
return ''.join(self._format_widgets()).ljust(self.term_width)
|
|
|
|
def _next_update(self):
|
|
return int((int(self.num_intervals *
|
|
(self.currval / self.maxval)) + 1) *
|
|
self.update_interval)
|
|
|
|
def _need_update(self):
|
|
"""Returns true when the progressbar should print an updated line.
|
|
|
|
You can override this method if you want finer grained control over
|
|
updates.
|
|
|
|
The current implementation is optimized to be as fast as possible and
|
|
as economical as possible in the number of updates. However, depending
|
|
on your usage you may want to do more updates. For instance, if your
|
|
progressbar stays in the same percentage for a long time, and you want
|
|
to update other widgets, like ETA, then you could return True after
|
|
some time has passed with no updates.
|
|
|
|
Ideally you could call self._format_line() and see if it's different
|
|
from the previous _format_line() call, but calling _format_line() takes
|
|
around 20 times more time than calling this implementation of
|
|
_need_update().
|
|
"""
|
|
return self.currval >= self.next_update
|
|
|
|
def update(self, value):
|
|
"Updates the progress bar to a new value."
|
|
assert 0 <= value <= self.maxval, '0 <= %d <= %d' % (value, self.maxval)
|
|
self.currval = value
|
|
if not self._need_update():
|
|
return
|
|
if self.start_time is None:
|
|
raise RuntimeError('You must call start() before calling update()')
|
|
now = time.time()
|
|
self.seconds_elapsed = now - self.start_time
|
|
self.next_update = self._next_update()
|
|
self.fd.write(self._format_line() + '\r')
|
|
self.last_update_time = now
|
|
|
|
def start(self):
|
|
"""Starts measuring time, and prints the bar at 0%.
|
|
|
|
It returns self so you can use it like this:
|
|
>>> pbar = ProgressBar().start()
|
|
>>> for i in xrange(100):
|
|
... # do something
|
|
... pbar.update(i+1)
|
|
...
|
|
>>> pbar.finish()
|
|
"""
|
|
if self.maxval is None:
|
|
self.maxval = self._DEFAULT_MAXVAL
|
|
assert self.maxval > 0
|
|
|
|
self.num_intervals = max(100, self.term_width)
|
|
self.update_interval = self.maxval / self.num_intervals
|
|
self.next_update = 0
|
|
|
|
self.start_time = self.last_update_time = time.time()
|
|
self.update(0)
|
|
return self
|
|
|
|
def finish(self):
|
|
"""Used to tell the progress is finished."""
|
|
self.finished = True
|
|
self.update(self.maxval)
|
|
self.fd.write('\n')
|
|
if self.signal_set:
|
|
signal.signal(signal.SIGWINCH, signal.SIG_DFL)
|