Fix a pep8 error

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

diff --git a/image_creator/dialog_main.py b/image_creator/dialog_main.py
index d9cea55..52d63cb 100644 (file)
--- a/image_creator/dialog_main.py
+++ b/image_creator/dialog_main.py
@@ -67,6 +67,7 @@ def image_creator(d, media, out):
         disk.cleanup()
 
     signal.signal(signal.SIGINT, signal_handler)
         disk.cleanup()
 
     signal.signal(signal.SIGINT, signal_handler)

+    signal.signal(signal.SIGTERM, signal_handler)

     try:
         snapshot = disk.snapshot()
         dev = disk.get_device(snapshot)
     try:
         snapshot = disk.snapshot()
         dev = disk.get_device(snapshot)

@@ -140,14 +141,17 @@ def select_file(d, media):
             else:
                 break
 
             else:
                 break
 

-        (code, media) = d.fselect(root, 10, 50,
-                                  title="Please select input media")
+        (code, media) = d.fselect(root, 10, 60, extra_button=1,
+            title="Please select an input media.",
+            extra_label="Running System")

         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
             if confirm_exit(d, "You canceled the media selection dialog box."):
                 sys.exit(0)
             else:
                 media = None
                 continue
         if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
             if confirm_exit(d, "You canceled the media selection dialog box."):
                 sys.exit(0)
             else:
                 media = None
                 continue

+        elif code == d.DIALOG_EXTRA:
+            media = '/'

 
     return media
 
 
     return media