Implement a WizardExit exception in dialog_wizard

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

diff --git a/image_creator/main.py b/image_creator/main.py
index 49b76e7..85c9272 100644 (file)
--- a/image_creator/main.py
+++ b/image_creator/main.py
@@ -36,7 +36,7 @@
 from image_creator import __version__ as version
 from image_creator import util
 from image_creator.disk import Disk
-from image_creator.util import get_command, FatalError, MD5
+from image_creator.util import FatalError, MD5
 from image_creator.output.cli import SilentOutput, SimpleOutput, \
     OutputWthProgress
 from image_creator.os_type import os_cls
@@ -219,7 +219,7 @@ def image_creator():
         metadata = image_os.meta
         dev.umount()
 
-        size = options.shrink and dev.shrink() or dev.meta['SIZE']
+        size = options.shrink and dev.shrink() or dev.size
         metadata.update(dev.meta)
 
         # Add command line metadata to the collected ones...