Fix missing imports introduced in 023e1217b32385ba
[snf-image-creator] / image_creator / dialog_util.py
index 3039473..00f8683 100644 (file)
@@ -34,6 +34,8 @@
 # or implied, of GRNET S.A.
 
 import os
 # 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
 
 SMALL_WIDTH = 60
 WIDTH = 70
@@ -68,6 +70,15 @@ class Reset(Exception):
     pass
 
 
     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()
 def extract_image(session):
     d = session['dialog']
     dir = os.getcwd()