Bump

PixelBar and PixelSpinner, based on Braille characters

README.rst

@@ -1,19 +1,30 @@
 Easy progress reporting for Python
 ==================================
 
+|pypi|
+
+|demo|
+
+.. |pypi| image:: https://img.shields.io/pypi/v/progress.svg
+.. |demo| image:: https://raw.github.com/verigak/progress/master/demo.gif
+   :alt: Demo
+
 Bars
 ----
 
-There are 6 progress bars to choose from:
+There are 7 progress bars to choose from:
 
-- Bar
-- ChargingBar
-- FillingSquaresBar
-- FillingCirclesBar
-- IncrementalBar
-- ShadyBar
+- ``Bar``
+- ``ChargingBar``
+- ``FillingSquaresBar``
+- ``FillingCirclesBar``
+- ``IncrementalBar``
+- ``PixelBar``
+- ``ShadyBar``
 
-To use them, just call ``next`` to advance and ``finish`` to finish. ::
+To use them, just call ``next`` to advance and ``finish`` to finish:
+
+.. code-block:: python
 
     from progress.bar import Bar
 
@@ -28,13 +39,17 @@ The result will be a bar like the following: ::
     Processing |#############                   | 42/100
 
 To simplify the common case where the work is done in an iterator, you can
-use the ``iter`` method. ::
+use the ``iter`` method:
+
+.. code-block:: python
 
     for i in Bar('Processing').iter(it):
         # Do some work
 
 Progress bars are very customizable, you can change their width, their fill
-character, their suffix and more. ::
+character, their suffix and more:
+
+.. code-block:: python
 
     bar = Bar('Loading', fill='@', suffix='%(percent)d%%')
 
@@ -44,48 +59,73 @@ This will produce a bar like the following: ::
 
 You can use a number of template arguments in ``message`` and ``suffix``:
 
-=========  =============================
-Name       Value
-=========  =============================
-index      current value
-max        maximum value
-remaining  max - index
-progress   index / max
-percent    progress * 100
-avg        rolling average time per item (in seconds)
-eta        avg * remaining
-=========  =============================
+==========  ================================
+Name        Value
+==========  ================================
+index       current value
+max         maximum value
+remaining   max - index
+progress    index / max
+percent     progress * 100
+avg         simple moving average time per item (in seconds)
+elapsed     elapsed time in seconds
+elapsed_td  elapsed as a timedelta (useful for printing as a string)
+eta         avg * remaining
+eta_td      eta as a timedelta (useful for printing as a string)
+==========  ================================
 
 Instead of passing all configuration options on instatiation, you can create
-your custom subclass. ::
+your custom subclass:
+
+.. code-block:: python
 
     class FancyBar(Bar):
         message = 'Loading'
         fill = '*'
         suffix = '%(percent).1f%% - %(eta)ds'
 
+You can also override any of the arguments or create your own:
+
+.. code-block:: python
+
+    class SlowBar(Bar):
+        suffix = '%(remaining_hours)d hours remaining'
+        @property
+        def remaining_hours(self):
+            return self.eta // 3600
+
 
 Spinners
 ========
 
-For actions with an unknown number of steps you can use a spinner. ::
+For actions with an unknown number of steps you can use a spinner:
+
+.. code-block:: python
 
     from progress.spinner import Spinner
-    
+
     spinner = Spinner('Loading ')
     while state != 'FINISHED':
         # Do some work
         spinner.next()
 
-There are 4 predefined spinners:
+There are 5 predefined spinners:
+
+- ``Spinner``
+- ``PieSpinner``
+- ``MoonSpinner``
+- ``LineSpinner``
+- ``PixelSpinner``
 
-- Spinner
-- PieSpinner
-- MoonSpinner
-- LineSpinner
 
 Other
 =====
 
-Thera are a number of other classes available too, please check the source or
+There are a number of other classes available too, please check the source or
 subclass one of them to create your own.
+
+
+License
+=======
+
+progress is licensed under ISC

progress/__init__.py

@@ -14,97 +14,104 @@
 
 from __future__ import division
 
+from collections import deque
+from datetime import timedelta
 from math import ceil
 from sys import stderr
 from time import time
 
 
-__version__ = '1.0.1'
+__version__ = '1.3'
 
 
 class Infinite(object):
     file = stderr
-    avg_window = 10
+    sma_window = 10         # Simple Moving Average window
 
     def __init__(self, *args, **kwargs):
-        self.ctx = {}
-        for key, val in kwargs.items():
-            if hasattr(self, key):
-                setattr(self, key, val)
-            else:
-                self.ctx[key] = val
-
         self.index = 0
-        self.avg = None
-        self._ts = time()
+        self.start_ts = time()
+        self.avg = 0
+        self._ts = self.start_ts
+        self._xput = deque(maxlen=self.sma_window)
+        for key, val in kwargs.items():
+            setattr(self, key, val)
 
-    def update_stats(self):
-        # Calculate moving average
-        now = time()
-        dt = now - self._ts
-        self._ts = now
-        w = self.avg_window
-        self.avg = dt if self.avg is None else (dt + w * self.avg) / (w + 1)
+    def __getitem__(self, key):
+        if key.startswith('_'):
+            return None
+        return getattr(self, key, None)
+
+    @property
+    def elapsed(self):
+        return int(time() - self.start_ts)
+
+    @property
+    def elapsed_td(self):
+        return timedelta(seconds=self.elapsed)
+
+    def update_avg(self, n, dt):
+        if n > 0:
+            self._xput.append(dt / n)
+            self.avg = sum(self._xput) / len(self._xput)
 
     def update(self):
-        self.update_stats()
-        kv = [(key, val) for key, val in self.__dict__.items()
-                         if not key.startswith('_')]
-        self.ctx.update(kv)
+        pass
+
+    def start(self):
+        pass
 
     def finish(self):
         pass
 
-    def next(self):
-        self.index = self.index + 1
+    def next(self, n=1):
+        now = time()
+        dt = now - self._ts
+        self.update_avg(n, dt)
+        self._ts = now
+        self.index = self.index + n
         self.update()
 
     def iter(self, it):
-        for x in it:
-            yield x
-            self.next()
-        self.finish()
+        try:
+            for x in it:
+                yield x
+                self.next()
+        finally:
+            self.finish()
 
 
 class Progress(Infinite):
-    backtrack = False
-
     def __init__(self, *args, **kwargs):
         super(Progress, self).__init__(*args, **kwargs)
         self.max = kwargs.get('max', 100)
-        self.eta = None
 
-    def update_stats(self):
-        if self.delta <= 0:
-            return
+    @property
+    def eta(self):
+        return int(ceil(self.avg * self.remaining))
 
-        # Calculate moving average
-        now = time()
-        dt = (now - self._ts) / self.delta
-        self._ts = now
-        w = self.avg_window
-        self.avg = dt if self.avg is None else (dt + w * self.avg) / (w + 1)
+    @property
+    def eta_td(self):
+        return timedelta(seconds=self.eta)
 
-        self.progress = min(1, self.index / self.max)
-        self.percent = self.progress * 100
-        self.remaining = self.max - self.index
-        self.eta = int(ceil(self.avg * self.remaining))
+    @property
+    def percent(self):
+        return self.progress * 100
 
-    def next(self):
-        prev = self.index
-        self.index = min(self.index + 1, self.max)
-        self.delta = self.index - prev
+    @property
+    def progress(self):
+        return min(1, self.index / self.max)
+
+    @property
+    def remaining(self):
+        return max(self.max - self.index, 0)
+
+    def start(self):
         self.update()
 
     def goto(self, index):
-        index = min(index, self.max)
-        delta = index - self.index
-        if delta <= 0 and not self.backtrack:
-            return
-
-        self.index = index
-        self.delta = delta
-        self.update()
+        incr = index - self.index
+        self.next(incr)
 
     def iter(self, it):
         try:
@@ -112,7 +119,9 @@ class Progress(Infinite):
         except TypeError:
             pass
 
-        for x in it:
-            yield x
-            self.next()
-        self.finish()
+        try:
+            for x in it:
+                yield x
+                self.next()
+        finally:
+            self.finish()

progress/bar.py

@@ -14,6 +14,7 @@
 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
+from __future__ import unicode_literals
 from . import Progress
 from .helpers import WritelnMixin
 
@@ -26,16 +27,16 @@ class Bar(WritelnMixin, Progress):
     bar_suffix = '| '
     empty_fill = ' '
     fill = '#'
+    hide_cursor = True
 
     def update(self):
-        super(Bar, self).update()
         filled_length = int(self.width * self.progress)
         empty_length = self.width - filled_length
 
-        message = self.message % self.ctx
+        message = self.message % self
         bar = self.fill * filled_length
         empty = self.empty_fill * empty_length
-        suffix = self.suffix % self.ctx
+        suffix = self.suffix % self
         line = ''.join([message, self.bar_prefix, bar, empty, self.bar_suffix,
                         suffix])
         self.writeln(line)
@@ -45,41 +46,43 @@ class ChargingBar(Bar):
     suffix = '%(percent)d%%'
     bar_prefix = ' '
     bar_suffix = ' '
-    empty_fill = u'∙'
-    fill = u'█'
+    empty_fill = '∙'
+    fill = '█'
 
 
 class FillingSquaresBar(ChargingBar):
-    empty_fill = u'▢'
-    fill = u'▣'
+    empty_fill = '▢'
+    fill = '▣'
 
 
 class FillingCirclesBar(ChargingBar):
-    empty_fill = u'◯'
-    fill = u'◉'
+    empty_fill = '◯'
+    fill = '◉'
 
 
 class IncrementalBar(Bar):
-    phases = (u' ', u'▏', u'▎', u'▍', u'▌', u'▋', u'▊', u'▉', u'█')
+    phases = (' ', '▏', '▎', '▍', '▌', '▋', '▊', '▉', '█')
 
     def update(self):
-        super(IncrementalBar, self).update()
-
         nphases = len(self.phases)
-        expanded_length = int(nphases * self.width * self.progress)
-        filled_length = int(self.width * self.progress)
-        empty_length = self.width - filled_length
-        phase = expanded_length - (filled_length * nphases)
+        filled_len = self.width * self.progress
+        nfull = int(filled_len)                      # Number of full chars
+        phase = int((filled_len - nfull) * nphases)  # Phase of last char
+        nempty = self.width - nfull                  # Number of empty chars
 
-        message = self.message % self.ctx
-        bar = self.phases[-1] * filled_length
+        message = self.message % self
+        bar = self.phases[-1] * nfull
         current = self.phases[phase] if phase > 0 else ''
-        empty = self.empty_fill * max(0, empty_length - len(current))
-        suffix = self.suffix % self.ctx
+        empty = self.empty_fill * max(0, nempty - len(current))
+        suffix = self.suffix % self
         line = ''.join([message, self.bar_prefix, bar, current, empty,
                         self.bar_suffix, suffix])
         self.writeln(line)
 
 
+class PixelBar(IncrementalBar):
+    phases = ('⡀', '⡄', '⡆', '⡇', '⣇', '⣧', '⣷', '⣿')
+
+
 class ShadyBar(IncrementalBar):
-    phases = (u' ', u'░', u'▒', u'▓', u'█')
+    phases = (' ', '░', '▒', '▓', '█')

progress/counter.py

@@ -14,33 +14,35 @@
 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
+from __future__ import unicode_literals
 from . import Infinite, Progress
 from .helpers import WriteMixin
 
 
 class Counter(WriteMixin, Infinite):
     message = ''
+    hide_cursor = True
 
     def update(self):
-        super(Counter, self).update()
         self.write(str(self.index))
 
 
 class Countdown(WriteMixin, Progress):
+    hide_cursor = True
+
     def update(self):
-        super(Countdown, self).update()
         self.write(str(self.remaining))
 
 
 class Stack(WriteMixin, Progress):
-    phases = (u' ', u'▁', u'▂', u'▃', u'▄', u'▅', u'▆', u'▇', u'█')
+    phases = (' ', '▁', '▂', '▃', '▄', '▅', '▆', '▇', '█')
+    hide_cursor = True
 
     def update(self):
-        super(Stack, self).update()
         nphases = len(self.phases)
         i = min(nphases - 1, int(self.progress * nphases))
         self.write(self.phases[i])
 
 
 class Pie(Stack):
-    phases = (u'○', u'◔', u'◑', u'◕', u'●')
+    phases = ('○', '◔', '◑', '◕', '●')

progress/helpers.py

@@ -15,7 +15,13 @@
 from __future__ import print_function
 
 
+HIDE_CURSOR = '\x1b[?25l'
+SHOW_CURSOR = '\x1b[?25h'
+
+
 class WriteMixin(object):
+    hide_cursor = False
+
     def __init__(self, message=None, **kwargs):
         super(WriteMixin, self).__init__(**kwargs)
         self._width = 0
@@ -23,35 +29,63 @@ class WriteMixin(object):
             self.message = message
 
         if self.file.isatty():
+            if self.hide_cursor:
+                print(HIDE_CURSOR, end='', file=self.file)
             print(self.message, end='', file=self.file)
             self.file.flush()
 
     def write(self, s):
         if self.file.isatty():
             b = '\b' * self._width
-            print(b + s.ljust(self._width), end='', file=self.file)
+            c = s.ljust(self._width)
+            print(b + c, end='', file=self.file)
             self._width = max(self._width, len(s))
             self.file.flush()
 
+    def finish(self):
+        if self.file.isatty() and self.hide_cursor:
+            print(SHOW_CURSOR, end='', file=self.file)
+
 
 class WritelnMixin(object):
+    hide_cursor = False
+
     def __init__(self, message=None, **kwargs):
         super(WritelnMixin, self).__init__(**kwargs)
-        self.max_line_width = 0
         if message:
             self.message = message
 
-    def writeln(self, line):
-        if not self.file.isatty():
-            return
-        if len(line) > self.max_line_width:
-            self.max_line_width = len(line)
-        else:
-            line += ' ' * (self.max_line_width - len(line))     # Add padding
+        if self.file.isatty() and self.hide_cursor:
+            print(HIDE_CURSOR, end='', file=self.file)
 
-        print('\r' + line, end='', file=self.file)
-        self.file.flush()
+    def clearln(self):
+        if self.file.isatty():
+            print('\r\x1b[K', end='', file=self.file)
+
+    def writeln(self, line):
+        if self.file.isatty():
+            self.clearln()
+            print(line, end='', file=self.file)
+            self.file.flush()
 
     def finish(self):
         if self.file.isatty():
             print(file=self.file)
+            if self.hide_cursor:
+                print(SHOW_CURSOR, end='', file=self.file)
+
+
+from signal import signal, SIGINT
+from sys import exit
+
+
+class SigIntMixin(object):
+    """Registers a signal handler that calls finish on SIGINT"""
+
+    def __init__(self, *args, **kwargs):
+        super(SigIntMixin, self).__init__(*args, **kwargs)
+        signal(SIGINT, self._sigint_handler)
+
+    def _sigint_handler(self, signum, frame):
+        self.finish()
+        exit(0)

progress/spinner.py

@@ -14,6 +14,7 @@
 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 # OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 
+from __future__ import unicode_literals
 from . import Infinite
 from .helpers import WriteMixin
 
@@ -21,20 +22,23 @@ from .helpers import WriteMixin
 class Spinner(WriteMixin, Infinite):
     message = ''
     phases = ('-', '\\', '|', '/')
+    hide_cursor = True
 
     def update(self):
-        super(Spinner, self).update()
         i = self.index % len(self.phases)
         self.write(self.phases[i])
 
 
 class PieSpinner(Spinner):
-    phases = [u'◷', u'◶', u'◵', u'◴']
+    phases = ['◷', '◶', '◵', '◴']
 
 
 class MoonSpinner(Spinner):
-    phases = [u'◑', u'◒', u'◐', u'◓']
+    phases = ['◑', '◒', '◐', '◓']
 
 
 class LineSpinner(Spinner):
-    phases = [u'⎺', u'⎻', u'⎼', u'⎽', u'⎼', u'⎻']
+    phases = ['⎺', '⎻', '⎼', '⎽', '⎼', '⎻']
+
+class PixelSpinner(Spinner):
+    phases = ['⣾','⣷', '⣯', '⣟', '⡿', '⢿', '⣻', '⣽']

setup.py

@@ -22,5 +22,8 @@ setup(
         'Programming Language :: Python :: 2.6',
         'Programming Language :: Python :: 2.7',
         'Programming Language :: Python :: 3.3'
+        'Programming Language :: Python :: 3.4'
+        'Programming Language :: Python :: 3.5'
+        'Programming Language :: Python :: 3.6'
     ]
 )

test_progress.py

@@ -2,35 +2,47 @@
 
 from __future__ import print_function
 
-from random import randint
-from time import sleep
+import random
+import time
 
 from progress.bar import (Bar, ChargingBar, FillingSquaresBar,
-                          FillingCirclesBar, IncrementalBar, ShadyBar)
-from progress.spinner import Spinner, PieSpinner, MoonSpinner, LineSpinner
+                          FillingCirclesBar, IncrementalBar, PixelBar,
+                          ShadyBar)
+from progress.spinner import (Spinner, PieSpinner, MoonSpinner, LineSpinner,
+                              PixelSpinner)
 from progress.counter import Counter, Countdown, Stack, Pie
 
 
-for bar in (Bar, ChargingBar, FillingSquaresBar, FillingCirclesBar):
-    for i in bar(bar.__name__).iter(range(100)):
-        sleep(0.04)
+def sleep():
+    t = 0.01
+    t += t * random.uniform(-0.1, 0.1)  # Add some variance
+    time.sleep(t)
 
-for bar in (IncrementalBar, ShadyBar):
-    for i in bar(bar.__name__).iter(range(200)):
-        sleep(0.02)
 
-for spin in (Spinner, PieSpinner, MoonSpinner, LineSpinner):
-    for i in spin(spin.__name__ + ' ').iter(range(30)):
-        sleep(0.1)
+for bar_cls in (Bar, ChargingBar, FillingSquaresBar, FillingCirclesBar):
+    suffix = '%(index)d/%(max)d [%(elapsed)d / %(eta)d / %(eta_td)s]'
+    bar = bar_cls(bar_cls.__name__, suffix=suffix)
+    for i in bar.iter(range(200)):
+        sleep()
+
+for bar_cls in (IncrementalBar, PixelBar, ShadyBar):
+    suffix = '%(percent)d%% [%(elapsed_td)s / %(eta)d / %(eta_td)s]'
+    bar = bar_cls(bar_cls.__name__, suffix=suffix)
+    for i in bar.iter(range(200)):
+        sleep()
+
+for spin in (Spinner, PieSpinner, MoonSpinner, LineSpinner, PixelSpinner):
+    for i in spin(spin.__name__ + ' ').iter(range(100)):
+        sleep()
     print()
 
 for singleton in (Counter, Countdown, Stack, Pie):
     for i in singleton(singleton.__name__ + ' ').iter(range(100)):
-        sleep(0.04)
+        sleep()
     print()
 
-bar = IncrementalBar('Random', backtrack=True, suffix='')
+bar = IncrementalBar('Random', suffix='%(index)d')
 for i in range(100):
-    bar.goto(randint(0, 100))
-    sleep(0.1)
+    bar.goto(random.randint(0, 100))
+    sleep()
 bar.finish()