termstatus.py

from __future__ import division
import sys

class statusTxt:
    def __init__(self,initTxt):
        self.txtlen = len(initTxt)
        self.txt = initTxt
        print(initTxt)
        sys.stdout.flush()

    def update(self, updateTxt):
        print ("\010"+"\010"*self.txtlen+updateTxt)
        sys.stdout.flush()
        self.txtlen = len(updateTxt)
        self.txt = updateTxt

    def end(self,updateTxt=""):
        if updateTxt == "":
            print("")
        else:
            print ("\010"+"\010"*self.txtlen+updateTxt)

        sys.stdout.flush()

class progressBar:
    def __init__(self, width, nTot):
        self.width=width
        self.nTot=nTot
        self.stepped=0
        sys.stdout.write('['+' '*width+']')
        sys.stdout.write('\r'*(width+2))
        sys.stdout.flush() 
        self.done=False

    def update(self, val):
        steps = int(round(val/self.nTot*self.width))
        if steps > self.stepped:
            for ii in range(steps-self.stepped):
                sys.stdout.write('\r'+'*')
                sys.stdout.flush()
            self.stepped=steps

        if val == self.nTot:
            self.end()

    def end(self):
        if self.done is False:
            sys.stdout.write('\b] Done!')
            sys.stdout.flush()
            self.done=True