Skip to content

Instantly share code, notes, and snippets.

@nonamenix
Created May 14, 2013 09:58
Show Gist options
  • Save nonamenix/5574909 to your computer and use it in GitHub Desktop.
Save nonamenix/5574909 to your computer and use it in GitHub Desktop.
# https://github.com/ipython/ipython/issues/1527/
import sys, time
try:
import IPython
clear_output = IPython.core.display.clear_output
except:
IPython = None
class ProgressBar:
def __init__(self, iterations):
self.iterations = iterations
self.prog_bar = '[]'
self.fill_char = '*'
self.width = 40
self.__update_amount(0)
if IPython:
self.animate = self.animate_ipython
else:
self.animate = self.animate_noipython
def animate_ipython(self, iter):
if sys.platform.lower().startswith('win'):
print self, '\r',
else:
print self, chr(27),
self.update_iteration(iter + 1)
# time.sleep(0.5)
clear_output()
def update_iteration(self, elapsed_iter):
self.__update_amount((elapsed_iter / float(self.iterations)) * 100.0)
self.prog_bar += ' %d of %s complete' % (elapsed_iter, self.iterations)
def __update_amount(self, new_amount):
percent_done = int(round((new_amount / 100.0) * 100.0))
all_full = self.width - 2
num_hashes = int(round((percent_done / 100.0) * all_full))
self.prog_bar = '[' + self.fill_char * num_hashes + ' ' * (all_full - num_hashes) + ']'
pct_place = (len(self.prog_bar) / 2) - len(str(percent_done))
pct_string = '%d%%' % percent_done
self.prog_bar = self.prog_bar[0:pct_place] + \
(pct_string + self.prog_bar[pct_place + len(pct_string):])
def __str__(self):
return str(self.prog_bar)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment