forked from brl/citadel
392 lines
12 KiB
Python
392 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# progressbar - Text progress bar 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
|
|
|
|
"""Default ProgressBar widgets."""
|
|
|
|
from __future__ import division
|
|
|
|
import datetime
|
|
import math
|
|
|
|
try:
|
|
from abc import ABCMeta, abstractmethod
|
|
except ImportError:
|
|
AbstractWidget = object
|
|
abstractmethod = lambda fn: fn
|
|
else:
|
|
AbstractWidget = ABCMeta('AbstractWidget', (object,), {})
|
|
|
|
|
|
def format_updatable(updatable, pbar):
|
|
if hasattr(updatable, 'update'): return updatable.update(pbar)
|
|
else: return updatable
|
|
|
|
|
|
class Widget(AbstractWidget):
|
|
"""The base class for all widgets.
|
|
|
|
The ProgressBar will call the widget's update value when the widget should
|
|
be updated. The widget's size may change between calls, but the widget may
|
|
display incorrectly if the size changes drastically and repeatedly.
|
|
|
|
The boolean TIME_SENSITIVE informs the ProgressBar that it should be
|
|
updated more often because it is time sensitive.
|
|
"""
|
|
|
|
TIME_SENSITIVE = False
|
|
__slots__ = ()
|
|
|
|
@abstractmethod
|
|
def update(self, pbar):
|
|
"""Updates the widget.
|
|
|
|
pbar - a reference to the calling ProgressBar
|
|
"""
|
|
|
|
|
|
class WidgetHFill(Widget):
|
|
"""The base class for all variable width widgets.
|
|
|
|
This widget is much like the \\hfill command in TeX, 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.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def update(self, pbar, width):
|
|
"""Updates the widget providing the total width the widget must fill.
|
|
|
|
pbar - a reference to the calling ProgressBar
|
|
width - The total width the widget must fill
|
|
"""
|
|
|
|
|
|
class Timer(Widget):
|
|
"""Widget which displays the elapsed seconds."""
|
|
|
|
__slots__ = ('format_string',)
|
|
TIME_SENSITIVE = True
|
|
|
|
def __init__(self, format='Elapsed Time: %s'):
|
|
self.format_string = format
|
|
|
|
@staticmethod
|
|
def format_time(seconds):
|
|
"""Formats time as the string "HH:MM:SS"."""
|
|
|
|
return str(datetime.timedelta(seconds=int(seconds)))
|
|
|
|
|
|
def update(self, pbar):
|
|
"""Updates the widget to show the elapsed time."""
|
|
|
|
return self.format_string % self.format_time(pbar.seconds_elapsed)
|
|
|
|
|
|
class ETA(Timer):
|
|
"""Widget which attempts to estimate the time of arrival."""
|
|
|
|
TIME_SENSITIVE = True
|
|
|
|
def update(self, pbar):
|
|
"""Updates the widget to show the ETA or total time when finished."""
|
|
|
|
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 AdaptiveETA(Timer):
|
|
"""Widget which attempts to estimate the time of arrival.
|
|
|
|
Uses a weighted average of two estimates:
|
|
1) ETA based on the total progress and time elapsed so far
|
|
2) ETA based on the progress as per the last 10 update reports
|
|
|
|
The weight depends on the current progress so that to begin with the
|
|
total progress is used and at the end only the most recent progress is
|
|
used.
|
|
"""
|
|
|
|
TIME_SENSITIVE = True
|
|
NUM_SAMPLES = 10
|
|
|
|
def _update_samples(self, currval, elapsed):
|
|
sample = (currval, elapsed)
|
|
if not hasattr(self, 'samples'):
|
|
self.samples = [sample] * (self.NUM_SAMPLES + 1)
|
|
else:
|
|
self.samples.append(sample)
|
|
return self.samples.pop(0)
|
|
|
|
def _eta(self, maxval, currval, elapsed):
|
|
return elapsed * maxval / float(currval) - elapsed
|
|
|
|
def update(self, pbar):
|
|
"""Updates the widget to show the ETA or total time when finished."""
|
|
if pbar.currval == 0:
|
|
return 'ETA: --:--:--'
|
|
elif pbar.finished:
|
|
return 'Time: %s' % self.format_time(pbar.seconds_elapsed)
|
|
else:
|
|
elapsed = pbar.seconds_elapsed
|
|
currval1, elapsed1 = self._update_samples(pbar.currval, elapsed)
|
|
eta = self._eta(pbar.maxval, pbar.currval, elapsed)
|
|
if pbar.currval > currval1:
|
|
etasamp = self._eta(pbar.maxval - currval1,
|
|
pbar.currval - currval1,
|
|
elapsed - elapsed1)
|
|
weight = (pbar.currval / float(pbar.maxval)) ** 0.5
|
|
eta = (1 - weight) * eta + weight * etasamp
|
|
return 'ETA: %s' % self.format_time(eta)
|
|
|
|
|
|
class FileTransferSpeed(Widget):
|
|
"""Widget for showing the transfer speed (useful for file transfers)."""
|
|
|
|
FORMAT = '%6.2f %s%s/s'
|
|
PREFIXES = ' kMGTPEZY'
|
|
__slots__ = ('unit',)
|
|
|
|
def __init__(self, unit='B'):
|
|
self.unit = unit
|
|
|
|
def update(self, pbar):
|
|
"""Updates the widget with the current SI prefixed speed."""
|
|
|
|
if pbar.seconds_elapsed < 2e-6 or pbar.currval < 2e-6: # =~ 0
|
|
scaled = power = 0
|
|
else:
|
|
speed = pbar.currval / pbar.seconds_elapsed
|
|
power = int(math.log(speed, 1000))
|
|
scaled = speed / 1000.**power
|
|
|
|
return self.FORMAT % (scaled, self.PREFIXES[power], self.unit)
|
|
|
|
|
|
class AnimatedMarker(Widget):
|
|
"""An animated marker for the progress bar which defaults to appear as if
|
|
it were rotating.
|
|
"""
|
|
|
|
__slots__ = ('markers', 'curmark')
|
|
|
|
def __init__(self, markers='|/-\\'):
|
|
self.markers = markers
|
|
self.curmark = -1
|
|
|
|
def update(self, pbar):
|
|
"""Updates the widget to show the next marker or the first marker when
|
|
finished"""
|
|
|
|
if pbar.finished: return self.markers[0]
|
|
|
|
self.curmark = (self.curmark + 1) % len(self.markers)
|
|
return self.markers[self.curmark]
|
|
|
|
# Alias for backwards compatibility
|
|
RotatingMarker = AnimatedMarker
|
|
|
|
|
|
class Counter(Widget):
|
|
"""Displays the current count."""
|
|
|
|
__slots__ = ('format_string',)
|
|
|
|
def __init__(self, format='%d'):
|
|
self.format_string = format
|
|
|
|
def update(self, pbar):
|
|
return self.format_string % pbar.currval
|
|
|
|
|
|
class Percentage(Widget):
|
|
"""Displays the current percentage as a number with a percent sign."""
|
|
|
|
def update(self, pbar):
|
|
return '%3d%%' % pbar.percentage()
|
|
|
|
|
|
class FormatLabel(Timer):
|
|
"""Displays a formatted label."""
|
|
|
|
mapping = {
|
|
'elapsed': ('seconds_elapsed', Timer.format_time),
|
|
'finished': ('finished', None),
|
|
'last_update': ('last_update_time', None),
|
|
'max': ('maxval', None),
|
|
'seconds': ('seconds_elapsed', None),
|
|
'start': ('start_time', None),
|
|
'value': ('currval', None)
|
|
}
|
|
|
|
__slots__ = ('format_string',)
|
|
def __init__(self, format):
|
|
self.format_string = format
|
|
|
|
def update(self, pbar):
|
|
context = {}
|
|
for name, (key, transform) in self.mapping.items():
|
|
try:
|
|
value = getattr(pbar, key)
|
|
|
|
if transform is None:
|
|
context[name] = value
|
|
else:
|
|
context[name] = transform(value)
|
|
except: pass
|
|
|
|
return self.format_string % context
|
|
|
|
|
|
class SimpleProgress(Widget):
|
|
"""Returns progress as a count of the total (e.g.: "5 of 47")."""
|
|
|
|
__slots__ = ('sep',)
|
|
|
|
def __init__(self, sep=' of '):
|
|
self.sep = sep
|
|
|
|
def update(self, pbar):
|
|
return '%d%s%d' % (pbar.currval, self.sep, pbar.maxval)
|
|
|
|
|
|
class Bar(WidgetHFill):
|
|
"""A progress bar which stretches to fill the line."""
|
|
|
|
__slots__ = ('marker', 'left', 'right', 'fill', 'fill_left')
|
|
|
|
def __init__(self, marker='#', left='|', right='|', fill=' ',
|
|
fill_left=True):
|
|
"""Creates a customizable progress bar.
|
|
|
|
marker - string or updatable object to use as a marker
|
|
left - string or updatable object to use as a left border
|
|
right - string or updatable object to use as a right border
|
|
fill - character to use for the empty part of the progress bar
|
|
fill_left - whether to fill from the left or the right
|
|
"""
|
|
self.marker = marker
|
|
self.left = left
|
|
self.right = right
|
|
self.fill = fill
|
|
self.fill_left = fill_left
|
|
|
|
|
|
def update(self, pbar, width):
|
|
"""Updates the progress bar and its subcomponents."""
|
|
|
|
left, marked, right = (format_updatable(i, pbar) for i in
|
|
(self.left, self.marker, self.right))
|
|
|
|
width -= len(left) + len(right)
|
|
# Marked must *always* have length of 1
|
|
if pbar.maxval:
|
|
marked *= int(pbar.currval / pbar.maxval * width)
|
|
else:
|
|
marked = ''
|
|
|
|
if self.fill_left:
|
|
return '%s%s%s' % (left, marked.ljust(width, self.fill), right)
|
|
else:
|
|
return '%s%s%s' % (left, marked.rjust(width, self.fill), right)
|
|
|
|
|
|
class ReverseBar(Bar):
|
|
"""A bar which has a marker which bounces from side to side."""
|
|
|
|
def __init__(self, marker='#', left='|', right='|', fill=' ',
|
|
fill_left=False):
|
|
"""Creates a customizable progress bar.
|
|
|
|
marker - string or updatable object to use as a marker
|
|
left - string or updatable object to use as a left border
|
|
right - string or updatable object to use as a right border
|
|
fill - character to use for the empty part of the progress bar
|
|
fill_left - whether to fill from the left or the right
|
|
"""
|
|
self.marker = marker
|
|
self.left = left
|
|
self.right = right
|
|
self.fill = fill
|
|
self.fill_left = fill_left
|
|
|
|
|
|
class BouncingBar(Bar):
|
|
def update(self, pbar, width):
|
|
"""Updates the progress bar and its subcomponents."""
|
|
|
|
left, marker, right = (format_updatable(i, pbar) for i in
|
|
(self.left, self.marker, self.right))
|
|
|
|
width -= len(left) + len(right)
|
|
|
|
if pbar.finished: return '%s%s%s' % (left, width * marker, right)
|
|
|
|
position = int(pbar.currval % (width * 2 - 1))
|
|
if position > width: position = width * 2 - position
|
|
lpad = self.fill * (position - 1)
|
|
rpad = self.fill * (width - len(marker) - len(lpad))
|
|
|
|
# Swap if we want to bounce the other way
|
|
if not self.fill_left: rpad, lpad = lpad, rpad
|
|
|
|
return '%s%s%s%s%s' % (left, lpad, marker, rpad, right)
|
|
|
|
|
|
class BouncingSlider(Bar):
|
|
"""
|
|
A slider that bounces back and forth in response to update() calls
|
|
without reference to the actual value. Based on a combination of
|
|
BouncingBar from a newer version of this module and RotatingMarker.
|
|
"""
|
|
def __init__(self, marker='<=>'):
|
|
self.curmark = -1
|
|
self.forward = True
|
|
Bar.__init__(self, marker=marker)
|
|
def update(self, pbar, width):
|
|
left, marker, right = (format_updatable(i, pbar) for i in
|
|
(self.left, self.marker, self.right))
|
|
|
|
width -= len(left) + len(right)
|
|
if width < 0:
|
|
return ''
|
|
|
|
if pbar.finished: return '%s%s%s' % (left, width * '=', right)
|
|
|
|
self.curmark = self.curmark + 1
|
|
position = int(self.curmark % (width * 2 - 1))
|
|
if position + len(marker) > width:
|
|
self.forward = not self.forward
|
|
self.curmark = 1
|
|
position = 1
|
|
lpad = ' ' * (position - 1)
|
|
rpad = ' ' * (width - len(marker) - len(lpad))
|
|
|
|
if not self.forward:
|
|
temp = lpad
|
|
lpad = rpad
|
|
rpad = temp
|
|
return '%s%s%s%s%s' % (left, lpad, marker, rpad, right)
|