Source code for skchem.utils.progress

#! /usr/bin/env python
#
# Copyright (C) 2016 Rich Lewis <rl403@cam.ac.uk>
# License: 3-clause BSD


"""
# skchem.utils.progress

Module implementing progress bars.
"""

import progressbar


[docs]class NamedProgressBar(progressbar.ProgressBar):

    def __init__(self, name=None, **kwargs):
        kwargs.setdefault('redirect_stdout', True)
        kwargs.setdefault('redirect_stderr', True)
        super(NamedProgressBar, self).__init__(**kwargs)
        self.name = name

[docs]    def default_widgets(self):
        return [self.name, ': '] + super(NamedProgressBar, self).default_widgets()