# Class to encapsulate a progress indicator
import sys, os
import unittest
class progressException(Exception): 'Error to raise for any recursive problem.'
class progress:
def __init__ (self):
self._twissler = ["|","/","-","\\","|"]
self._state = 0
def getStart(self):
sys.stdout.write('\t[ ') # include 2 spaces for the twissler
sys.stdout.flush()
def getStart(self, text):
sys.stdout.write('\t %s [ ' % (text)) # include 2 spaces for the twissler
sys.stdout.flush()
def moveOn(self):
try:
self._state = self._state + 1
if self._state >= 5:
self._state = 1
sys.stdout.write( chr(8)+chr(8)+self._twissler[self._state]+']' )
sys.stdout.flush()
except:
raise progressException('failed to progress')
class TestCase(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def testProgress(self):
p = progress()
p.getStart()
for a in range(0,10,1):
p.moveOn()
time.sleep(0.2)
if __name__ == '__main__':
widgetTestSuite = unittest.TestSuite()
widgetTestSuite.addTest(TestCase("testProgress"))
runner = unittest.TextTestRunner()
runner.run(widgetTestSuite)