added the prograssbar instance it self a argument passed to the decorated function

This commit is contained in:
Richard Hartmann 2015-08-27 17:02:04 +02:00
parent e9cc57b3e6
commit 047ecf0607

View file

@ -152,6 +152,7 @@ class ProgressBar(object):
prepend = "{} ".format(self.__name__),
**self.kwargs) as pb:
pb.start()
callargs['progress_bar'] = pb
return self.func(**callargs)
@ -208,8 +209,7 @@ class ProgressBarExtended(ProgressBar):
if progress_bar_off:
return self.func(**callargs)
count = callargs[self.cm[0]]
max_count = callargs[self.cm[1]]
@ -268,6 +268,7 @@ class ProgressBarOverrideCount(ProgressBar):
prepend = "{} ".format(self.__name__),
**self.kwargs) as pb:
pb.start()
callargs['progress_bar'] = pb
return self.func(**callargs)