Fix a bug in Output.progress_generator

[snf-image-creator] / image_creator / output.py

diff --git a/image_creator/output.py b/image_creator/output.py
index 0f33f6d..fc3c422 100644 (file)
--- a/image_creator/output.py
+++ b/image_creator/output.py
@@ -86,12 +86,10 @@ class Output(object):
         def success(self, result):
             sucess(result)
 
-    def progress_gen(self, message):
-        
-        progress = getattr(self, 'Progress')
-
+    def progress_generator(self, message):
         def generator(n):
-            progressbar = progress(message, 'default')
+            progressbar = self.Progress(message, 'default')
+            progressbar.max = n
 
             for _ in range(n):
                 yield