124 lines
2.9 KiB
Python
124 lines
2.9 KiB
Python
|
#!/usr/bin/python
|
||
|
|
# coding: utf8
|
||
|
|
|
||
|
|
""" Progress bar """
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import progressbar
|
||
|
|
|
||
|
|
class Pbar(object): # pylint: disable=useless-object-inheritance
|
||
|
|
"""
|
||
|
|
Progress bar
|
||
|
|
|
||
|
|
This class abstract a progress bar that could be enable/disable by
|
||
|
|
configuration/script parameters.
|
||
|
|
"""
|
||
|
|
|
||
|
|
__pbar = None
|
||
|
|
__count = None
|
||
|
|
|
||
|
|
def __init__(self, name, maxval, enabled=True):
|
||
|
|
if enabled and maxval:
|
||
|
|
self.__count = 0
|
||
|
|
self.__pbar = progressbar.ProgressBar(
|
||
|
|
widgets=[
|
||
|
|
name + ': ',
|
||
|
|
progressbar.Percentage(),
|
||
|
|
' ',
|
||
|
|
progressbar.Bar(),
|
||
|
|
' ',
|
||
|
|
progressbar.SimpleProgress(),
|
||
|
|
progressbar.ETA()
|
||
|
|
],
|
||
|
|
maxval=maxval
|
||
|
|
).start()
|
||
|
|
else:
|
||
|
|
logging.info(name)
|
||
|
|
|
||
|
|
def increment(self, step=None):
|
||
|
|
"""
|
||
|
|
Increment the progress bar
|
||
|
|
|
||
|
|
:param step: The step (optional, default: 1)
|
||
|
|
"""
|
||
|
|
if self.__pbar:
|
||
|
|
self.__count += step if step else 1
|
||
|
|
self.__pbar.update(self.__count)
|
||
|
|
|
||
|
|
def finish(self):
|
||
|
|
""" Finish the progress bar """
|
||
|
|
if self.__pbar:
|
||
|
|
self.__pbar.finish()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
# Run tests
|
||
|
|
import time
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
default_max_val = 10
|
||
|
|
|
||
|
|
# Options parser
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
'-v', '--verbose',
|
||
|
|
action="store_true",
|
||
|
|
dest="verbose",
|
||
|
|
help="Enable verbose mode"
|
||
|
|
)
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
'-d', '--debug',
|
||
|
|
action="store_true",
|
||
|
|
dest="debug",
|
||
|
|
help="Enable debug mode"
|
||
|
|
)
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
'-l', '--log-file',
|
||
|
|
action="store",
|
||
|
|
type=str,
|
||
|
|
dest="logfile",
|
||
|
|
help="Log file path"
|
||
|
|
)
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
'-p', '--progress',
|
||
|
|
action="store_true",
|
||
|
|
dest="progress",
|
||
|
|
help="Enable progress bar"
|
||
|
|
)
|
||
|
|
|
||
|
|
parser.add_argument(
|
||
|
|
'-C', '--count',
|
||
|
|
action="store",
|
||
|
|
type=int,
|
||
|
|
dest="count",
|
||
|
|
help="Progress bar max value (default: %s)" % default_max_val,
|
||
|
|
default=default_max_val
|
||
|
|
)
|
||
|
|
|
||
|
|
options = parser.parse_args()
|
||
|
|
|
||
|
|
# Initialize logs
|
||
|
|
logformat = '%(asctime)s - Test Pbar - %(levelname)s - %(message)s'
|
||
|
|
if options.debug:
|
||
|
|
loglevel = logging.DEBUG
|
||
|
|
elif options.verbose:
|
||
|
|
loglevel = logging.INFO
|
||
|
|
else:
|
||
|
|
loglevel = logging.WARNING
|
||
|
|
|
||
|
|
if options.logfile:
|
||
|
|
logging.basicConfig(filename=options.logfile, level=loglevel, format=logformat)
|
||
|
|
else:
|
||
|
|
logging.basicConfig(level=loglevel, format=logformat)
|
||
|
|
|
||
|
|
pbar = Pbar('Test', options.count, enabled=options.progress)
|
||
|
|
|
||
|
|
for idx in range(0, options.count): # pylint: disable=unused-variable
|
||
|
|
pbar.increment()
|
||
|
|
time.sleep(0.3)
|
||
|
|
pbar.finish()
|