X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/672cabb9e76fe9b06c10ba96d7c2b0048ac5b37c..9e4b4de20ef2c2c23a96fb1ac5d474fa2370695e:/image_creator/dialog_main.py diff --git a/image_creator/dialog_main.py b/image_creator/dialog_main.py index 9b0c56b..4071018 100644 --- a/image_creator/dialog_main.py +++ b/image_creator/dialog_main.py @@ -38,276 +38,188 @@ import sys import os import textwrap import signal +import optparse from image_creator import __version__ as version -from image_creator.util import FatalError, MD5 -from image_creator.output.dialog import InitializationOutput +from image_creator.util import FatalError +from image_creator.output import Output +from image_creator.output.cli import SimpleOutput +from image_creator.output.dialog import GaugeOutput +from image_creator.output.composite import CompositeOutput from image_creator.disk import Disk from image_creator.os_type import os_cls +from image_creator.dialog_wizard import wizard +from image_creator.dialog_menu import main_menu +from image_creator.dialog_util import SMALL_WIDTH, WIDTH, confirm_exit, \ + Reset, update_background_title -MSGBOX_WIDTH = 60 -YESNO_WIDTH = 50 -MENU_WIDTH = 70 +def image_creator(d, media, out): -class Reset(Exception): - pass - - -def confirm_exit(d, msg=''): - return not d.yesno("%s Do you want to exit?" % msg, width=YESNO_WIDTH) - - -def confirm_reset(d): - return not d.yesno( - "Are you sure you want to reset everything?", - width=YESNO_WIDTH) + d.setBackgroundTitle('snf-image-creator') + gauge = GaugeOutput(d, "Initialization", "Initializing...") + out.add(gauge) + disk = Disk(media, out) -def upload_image(session): - d = session["dialog"] + def signal_handler(signum, frame): + gauge.cleanup() + disk.cleanup() - if "account" not in session: - d.msgbox("You need to provide your ~okeanos login username before you " - "can upload images to pithos+", width=MSGBOX_WIDTH) - return False + signal.signal(signal.SIGINT, signal_handler) + signal.signal(signal.SIGTERM, signal_handler) + try: + snapshot = disk.snapshot() + dev = disk.get_device(snapshot) - if "token" not in session: - d.msgbox("You need to provide your ~okeanos account authentication " - "token before you can upload images to pithos+", - width=MSGBOX_WIDTH) - return False + metadata = {} + for (key, value) in dev.meta.items(): + metadata[str(key)] = str(value) - while 1: - (code, answer) = d.inputbox("Please provide a filename:", - init=session["upload"] if "upload" in session else '') - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - return False + dev.mount(readonly=True) + out.output("Collecting image metadata...") + cls = os_cls(dev.distro, dev.ostype) + image_os = cls(dev.root, dev.g, out) + dev.umount() - answer = answer.strip() - if len(answer) == 0: - d.msgbox("Filename cannot be empty", width=MSGBOX_WIDTH) - continue + for (key, value) in image_os.meta.items(): + metadata[str(key)] = str(value) - session["upload"] = answer - return True + out.success("done") + gauge.cleanup() + out.remove(gauge) + # Make sure the signal handler does not call gauge.cleanup again + def dummy(self): + pass + gauge.cleanup = type(GaugeOutput.cleanup)(dummy, gauge, GaugeOutput) -def register_image(session): - d = session["dialog"] + session = {"dialog": d, + "disk": disk, + "snapshot": snapshot, + "device": dev, + "image_os": image_os, + "metadata": metadata} - if "account" not in session: - d.msgbox("You need to provide your ~okeanos login username before you " - "can register an images to cyclades", - width=MSGBOX_WIDTH) - return False + msg = "snf-image-creator detected a %s system on the input media. " \ + "Would you like to run a wizard to assist you through the " \ + "image creation process?\n\nChoose to run the wizard," \ + " to run the snf-image-creator in expert mode or " \ + "press ESC to quit the program." \ + % (dev.ostype if dev.ostype == dev.distro else "%s (%s)" % + (dev.ostype, dev.distro)) + + update_background_title(session) + + while True: + code = d.yesno(msg, width=WIDTH, height=12, yes_label="Wizard", + no_label="Expert") + if code == d.DIALOG_OK: + if wizard(session): + break + elif code == d.DIALOG_CANCEL: + main_menu(session) + break - if "token" not in session: - d.msgbox("You need to provide your ~okeanos account authentication " - "token before you can register an images to cyclades", - width=MSGBOX_WIDTH) - return False + if confirm_exit(d): + break - if "upload" not in session: - d.msgbox("You need to have an image uploaded to pithos+ before you " - "can register it to cyclades", - width=MSGBOX_WIDTH) - return False + d.infobox("Thank you for using snf-image-creator. Bye", width=53) + finally: + disk.cleanup() - return True + return 0 -def kamaki_menu(session): - d = session['dialog'] - default_item = "Account" +def select_file(d, media): + root = os.sep while 1: - account = session["account"] if "account" in session else "" - token = session["token"] if "token" in session else "" - upload = session["upload"] if "upload" in session else "" - (code, choice) = d.menu( - "Choose one of the following or press to go back.", - width=MENU_WIDTH, - choices=[("Account", "Change your ~okeanos username: %s" % - account), - ("Token", "Change your ~okeanos token: %s" % - token), - ("Upload", "Upload image to pithos+"), - ("Register", "Register image to cyclades: %s" % upload)], - cancel="Back", - default_item=default_item, - title="Image Registration Menu") - - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - return False - - if choice == "Account": - default_item = "Account" - (code, answer) = d.inputbox( - "Please provide your ~okeanos account e-mail address:", - init=session["account"] if "account" in session else '', - width=70) - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - continue - if len(answer) == 0 and "account" in session: - del session["account"] + if media is not None: + if not os.path.exists(media): + d.msgbox("The file `%s' you choose does not exist." % media, + width=SMALL_WIDTH) else: - session["account"] = answer.strip() - default_item = "Token" - elif choice == "Token": - default_item = "Token" - (code, answer) = d.inputbox( - "Please provide your ~okeanos account authetication token:", - init=session["token"] if "token" in session else '', - width=70) - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - continue - if len(answer) == 0 and "token" in session: - del session["token"] - else: - session["token"] = answer.strip() - default_item = "Upload" - elif choice == "Upload": - if upload_image(session): - default_item = "Register" - else: - default_item = "Upload" - elif choice == "Register": - if register_image(session): - return True - else: - default_item = "Register" - - -def main_menu(session): - d = session['dialog'] - dev = session['device'] - d.setBackgroundTitle("OS: %s, Distro: %s" % (dev.ostype, dev.distro)) - actions = {"Register": kamaki_menu} - default_item = "Customize" - - while 1: - (code, choice) = d.menu( - "Choose one of the following or press to exit.", - width=MENU_WIDTH, - choices=[("Customize", "Run various image customization tasks"), - ("Deploy", "Configure ~okeanos image deployment options"), - ("Register", "Register image to ~okeanos"), - ("Extract", "Dump image to local file system"), - ("Reset", "Reset everything and start over again"), - ("Help", "Get help for using snf-image-creator")], - cancel="Exit", - default_item=default_item, - title="Image Creator for ~okeanos (snf-image-creator version %s)" % - version) - - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - if confirm_exit(d): break - else: - continue - - if choice == "Reset": - if confirm_reset(d): - d.infobox("Resetting snf-image-creator. Please wait...") - raise Reset - else: - continue - elif choice in actions: - actions[choice](session) - -def select_file(d): - root = os.sep - while 1: - (code, path) = 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 + elif code == d.DIALOG_EXTRA: + media = '/' - if not os.path.exists(path): - d.msgbox("The file you choose does not exist", width=MSGBOX_WIDTH) - continue - else: - break + return media - return path +def main(): -def collect_metadata(dev, out): + d = dialog.Dialog(dialog="dialog") - dev.mount(readonly=True) - out.output("Collecting image metadata...") - cls = os_cls(dev.distro, dev.ostype) - image_os = cls(dev.root, dev.g, out) - out.success("done") - dev.umount() + # Add extra button in dialog library + dialog._common_args_syntax["extra_button"] = \ + lambda enable: dialog._simple_option("--extra-button", enable) - return image_os.meta + dialog._common_args_syntax["extra_label"] = \ + lambda string: ("--extra-label", string) + # Allow yes-no label overwriting + dialog._common_args_syntax["yes_label"] = \ + lambda string: ("--yes-label", string) -def image_creator(d): - basename = os.path.basename(sys.argv[0]) - usage = "Usage: %s [input_media]" % basename - if len(sys.argv) > 2: - sys.stderr.write("%s\n" % usage) - return 1 + dialog._common_args_syntax["no_label"] = \ + lambda string: ("--no-label", string) - if os.geteuid() != 0: - raise FatalError("You must run %s as root" % basename) + usage = "Usage: %prog [options] []" + parser = optparse.OptionParser(version=version, usage=usage) + parser.add_option("-l", "--logfile", type="string", dest="logfile", + default=None, help="log all messages to FILE", + metavar="FILE") - media = sys.argv[1] if len(sys.argv) == 2 else select_file(d) + options, args = parser.parse_args(sys.argv[1:]) - out = InitializationOutput(d) - disk = Disk(media, out) + if len(args) > 1: + parser.error("Wrong number of arguments") - def signal_handler(signum, fram): - out.cleanup() - disk.cleanup() + d.setBackgroundTitle('snf-image-creator') - signal.signal(signal.SIGINT, signal_handler) try: - snapshot = disk.snapshot() - dev = disk.get_device(snapshot) - - metadata = collect_metadata(dev, out) - out.cleanup() - - # Make sure the signal handler does not call out.cleanup again - def dummy(self): - pass - instancemethod = type(InitializationOutput.cleanup) - out.cleanup = instancemethod(dummy, out, InitializationOutput) - - session = {"dialog": d, - "disk": disk, - "device": dev, - "metadata": metadata} + if os.geteuid() != 0: + raise FatalError("You must run %s as root" % + parser.get_prog_name()) - main_menu(session) - d.infobox("Thank you for using snf-image-creator. Bye", width=53) - finally: - disk.cleanup() + media = select_file(d, args[0] if len(args) == 1 else None) - return 0 - - -def main(): - - d = dialog.Dialog(dialog="dialog") - - while 1: - try: + logfile = None + if options.logfile is not None: try: - ret = image_creator(d) - sys.exit(ret) - except FatalError as e: - msg = textwrap.fill(str(e), width=70) - d.infobox(msg, width=70, title="Fatal Error") - sys.exit(1) - except Reset: - continue + logfile = open(options.logfile, 'w') + except IOError as e: + raise FatalError( + "Unable to open logfile `%s' for writing. Reason: %s" % + (options.logfile, e.strerror)) + try: + log = SimpleOutput(False, logfile) if logfile is not None \ + else Output() + while 1: + try: + out = CompositeOutput([log]) + out.output("Starting %s v%s..." % + (parser.get_prog_name(), version)) + ret = image_creator(d, media, out) + sys.exit(ret) + except Reset: + log.output("Resetting everything...") + continue + finally: + if logfile is not None: + logfile.close() + except FatalError as e: + msg = textwrap.fill(str(e), width=WIDTH) + d.infobox(msg, width=WIDTH, title="Fatal Error") + sys.exit(1) # vim: set sta sts=4 shiftwidth=4 sw=4 et ai :