m_librarian/pbar.py

   1
   2 try:
   3     from m_lib.pbar.tty_pbar import ttyProgressBar
   4 except ImportError:
   5     ttyProgressBar = None
   6
   7 if ttyProgressBar:
   8     class ml_ttyProgressBar(object):
   9         def __init__(self, width=20):
  10             self.max = None
  11             self.pbar = None
  12             self.width = width
  13
  14         def set_max(self, max_value):
  15             self.max = max_value
  16             self.pbar = ttyProgressBar(0, max_value, width1=self.width)
  17
  18         def display(self, value):
  19             if self.pbar:
  20                 self.pbar.display(value)
  21
  22         def close(self):
  23             if self.pbar:
  24                 self.pbar.erase()
  25                 self.pbar = None