Fixed zero division error if max_count_value is zero.

This commit is contained in:
Paul Müller 2015-01-13 18:50:50 +01:00
parent fc7993d589
commit 4e711d4336
2 changed files with 6 additions and 2 deletions

View file

@ -117,7 +117,7 @@ class ProgressBar(object):
kwargs[self.cm[0]] = progress.UnsignedIntValue(0)
if not kwargs.has_key(self.cm[1]) or kwargs[self.cm[1]] is None:
# max_count
kwargs[self.cm[1]] = progress.UnsignedIntValue(0)
kwargs[self.cm[1]] = progress.UnsignedIntValue(1)
with progress.ProgressBar(kwargs[self.cm[0]], kwargs[self.cm[1]],
*self.args, **self.kwargs) as pb:
pb.start()

View file

@ -929,7 +929,11 @@ class ProgressBar(Progress):
l = len(s1) + len(s3)
l2 = width - l - 1
a = int(l2 * count_value / max_count_value)
if max_count_value != 0:
a = int(l2 * count_value / max_count_value)
else:
a = 0
b = l2 - a
s2 = "="*a + ">" + " "*b
print(s1+s2+s3)