class ProgressBar(object):
def __init__(self, title='Progress', max_value=100):
try:
from progressive.bar import Bar
self.support_progressive = True
self.bar = Bar(max_value=max_value, title='Receiving')
self.bar.cursor.clear_lines(1)
self.bar.cursor.save()
except ImportError:
self.support_progressive = False
from progress.bar import Bar
self.bar = Bar('Receiving', max=max_value)
def next(self, count=0):
if self.support_progressive:
self.bar.cursor.restore()
self.bar.draw(value=count)
else:
self.bar.next()
def finish(self):
if not self.support_progressive:
self.bar.finish()
