Fix a pep8 error
[snf-image-creator] / image_creator / dialog_util.py
index 3039473..384019a 100644 (file)
@@ -34,6 +34,8 @@
 # or implied, of GRNET S.A.
 
 import os
+from image_creator.output.dialog import GaugeOutput
+from image_creator.util import MD5
 
 SMALL_WIDTH = 60
 WIDTH = 70
@@ -42,6 +44,7 @@ WIDTH = 70
 def update_background_title(session):
     d = session['dialog']
     dev = session['device']
+    disk = session['disk']
 
     MB = 2 ** 20
 
@@ -49,8 +52,9 @@ def update_background_title(session):
     shrinked = 'shrinked' in session and session['shrinked']
     postfix = " (shrinked)" if shrinked else ''
 
-    title = "OS: %s, Distro: %s, Size: %dMB%s" % \
-            (dev.ostype, dev.distro, size, postfix)
+    title = "OS: %s, Distro: %s, Size: %dMB%s, Source: %s" % \
+            (dev.ostype, dev.distro, size, postfix,
+             os.path.abspath(disk.source))
 
     d.setBackgroundTitle(title)
 
@@ -68,6 +72,15 @@ class Reset(Exception):
     pass
 
 
+def extract_metadata_string(session):
+    metadata = ['%s=%s' % (k, v) for (k, v) in session['metadata'].items()]
+
+    if 'task_metadata' in session:
+        metadata.extend("%s=yes" % m for m in session['task_metadata'])
+
+    return '\n'.join(metadata) + '\n'
+
+
 def extract_image(session):
     d = session['dialog']
     dir = os.getcwd()