Save bar_type parameter in Output base class

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

diff --git a/image_creator/output/__init__.py b/image_creator/output/__init__.py
index 7143785..99653d3 100644 (file)
--- a/image_creator/output/__init__.py
+++ b/image_creator/output/__init__.py
@@ -45,6 +45,9 @@ class Output(object):
     def output(self, msg='', new_line=True):
         pass
 
+    def cleanup(self):
+        pass
+
     def _get_progress(self):
         progress = self._Progress
         progress.output = self
@@ -55,6 +58,7 @@ class Output(object):
     class _Progress(object):
         def __init__(self, size, title, bar_type='default'):
             self.size = size
+            self.bar_type = bar_type
             self.output.output("%s..." % title, False)
 
         def goto(self, dest):