X-Git-Url: https://code.grnet.gr/git/snf-image-creator/blobdiff_plain/2a21ff92a5af881912920b22dda1cdcc49067300..84bc469c7f80281c0fb224f6de72dcdb859aa5f6:/image_creator/dialog_main.py diff --git a/image_creator/dialog_main.py b/image_creator/dialog_main.py index 700825c..7e6b33d 100644 --- a/image_creator/dialog_main.py +++ b/image_creator/dialog_main.py @@ -36,434 +36,200 @@ import dialog import sys import os +import stat import textwrap import signal -import StringIO +import optparse from image_creator import __version__ as version -from image_creator.util import FatalError, MD5 -from image_creator.output.dialog import InitializationOutput, GaugeOutput +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.kamaki_wrapper import Kamaki, ClientError +from image_creator.dialog_wizard import start_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 -INPUTBOX_WIDTH=70 +def create_image(d, media, out, tmp): + """Create an image out of `media'""" + d.setBackgroundTitle('snf-image-creator') -class Reset(Exception): - pass + gauge = GaugeOutput(d, "Initialization", "Initializing...") + out.add(gauge) + disk = Disk(media, out, tmp) + def signal_handler(signum, frame): + gauge.cleanup() + disk.cleanup() -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) - -def extract_image(session): - d = session['dialog'] - dir = os.getcwd() - while 1: - if dir and dir[-1] != os.sep: - dir = dir + os.sep - - (code, path) = d.fselect(dir, 10, 50, title="Save image as...") - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - return False - - if os.path.isdir(path): - dir=path - continue - - if os.path.isdir("%s.meta" % path): - d.msgbox("Can't overwrite directory `%s.meta'" % path, - width=MSGBOX_WIDTH) - continue - - if os.path.isdir("%s.md5sum" % path): - d.msgbox("Can't overwrite directory `%s.md5sum'" % path, - width=MSGBOX_WIDTH) - continue - - basedir = os.path.dirname(path) - name = os.path.basename(path) - if not os.path.exists(basedir): - d.msgbox("Directory `%s' does not exist" % basedir, - width=MSGBOX_WIDTH) - continue - - dir = basedir - if len(name) == 0: - continue - - files = ["%s%s" % (path, ext) for ext in ('', '.meta', '.md5sum')] - overwrite = filter(os.path.exists, files) - - if len(overwrite) > 0: - if d.yesno("The following file(s) exist:\n" - "%s\nDo you want to overwrite them?" % - "\n".join(overwrite), width=YESNO_WIDTH): - continue - - out = GaugeOutput(d, "Image Extraction", "Extracting image...") - try: - dev = session['device'] - if "checksum" not in session: - size = dev.meta['SIZE'] - md5 = MD5(out) - session['checksum'] = md5.compute(session['snapshot'], size) - - # Extract image file - dev.out = out - dev.dump(path) - - # Extract metadata file - out.output("Extracting metadata file...") - metastring = '\n'.join( - ['%s=%s' % (k, v) for (k, v) in session['metadata'].items()]) - metastring += '\n' - with open('%s.meta' % path, 'w') as f: - f.write(metastring) - out.success('done') - - # Extract md5sum file - out.output("Extracting md5sum file...") - md5str = "%s %s\n" % (session['checksum'], name) - with open('%s.md5sum' % path, 'w') as f: - f.write(md5str) - out.success("done") - - finally: - out.cleanup() - d.msgbox("Image file `%s' was successfully extracted!" % path, - width=MSGBOX_WIDTH) - break - - return True - - -def upload_image(session): - d = session["dialog"] - size = session['device'].meta['SIZE'] - - 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 - - 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 - - while 1: - init=session["upload"] if "upload" in session else '' - (code, answer) = d.inputbox("Please provide a filename:", init=init, - width=INPUTBOX_WIDTH) - - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - return False - - filename = answer.strip() - if len(filename) == 0: - d.msgbox("Filename cannot be empty", width=MSGBOX_WIDTH) - continue - - break - - out = GaugeOutput(d, "Image Upload", "Uploading...") - if 'checksum' not in session: - md5 = MD5(out) - session['checksum'] = md5.compute(session['snapshot'], size) + signal.signal(signal.SIGINT, signal_handler) + signal.signal(signal.SIGTERM, signal_handler) try: - kamaki = Kamaki(session['account'], session['token'], out) - try: - # Upload image file - with open(session['snapshot'], 'rb') as f: - session["upload"] = kamaki.upload(f, size, filename, - "Calculating block hashes", - "Uploading missing blocks") - # Upload metadata file - out.output("Uploading metadata file...") - metastring = '\n'.join( - ['%s=%s' % (k, v) for (k, v) in session['metadata'].items()]) - metastring += '\n' - kamaki.upload(StringIO.StringIO(metastring), size=len(metastring), - remote_path="%s.meta" % filename) - out.success("done") - - # Upload md5sum file - out.output("Uploading md5sum file...") - md5str = "%s %s\n" % (session['checksum'], filename) - kamaki.upload(StringIO.StringIO(md5str), size=len(md5str), - remote_path="%s.md5sum" % filename) - out.success("done") - - except ClientError as e: - d.msgbox("Error in pithos+ client: %s" % e.message, - title="Pithos+ Client Error", width=MSGBOX_WIDTH) - if 'upload' in session: - del session['upload'] - return False - finally: - out.cleanup() - - d.msgbox("Image file `%s' was successfully uploaded to pithos+" % filename, - width=MSGBOX_WIDTH) - return True + snapshot = disk.snapshot() + image = disk.get_image(snapshot) + out.output("Collecting image metadata...") + metadata = {} + for (key, value) in image.meta.items(): + metadata[str(key)] = str(value) -def register_image(session): - d = session["dialog"] + for (key, value) in image.os.meta.items(): + metadata[str(key)] = str(value) - 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 + out.success("done") + gauge.cleanup() + out.remove(gauge) - 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 + # Make sure the signal handler does not call gauge.cleanup again + def dummy(self): + pass + gauge.cleanup = type(GaugeOutput.cleanup)(dummy, gauge, GaugeOutput) - 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 + session = {"dialog": d, + "disk": disk, + "image": image, + "metadata": metadata} - while 1: - (code, answer) = d.inputbox("Please provide a registration name:" - " be registered:", width=INPUTBOX_WIDTH) - if code in (d.DIALOG_CANCEL, d.DIALOG_ESC): - 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." \ + % (image.ostype if image.ostype == image.distro or + image.distro == "unknown" else "%s (%s)" % + (image.ostype, image.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 start_wizard(session): + break + elif code == d.DIALOG_CANCEL: + main_menu(session) + break - name = answer.strip() - if len(name) == 0: - d.msgbox("Registration name cannot be empty", width=MSGBOX_WIDTH) - continue - break + if confirm_exit(d): + break - out = GaugeOutput(d, "Image Registration", "Registrating image...") - try: - out.output("Registring image to cyclades...") - try: - kamaki = Kamaki(session['account'], session['token'], out) - kamaki.register(name, session['upload'], session['metadata']) - out.success('done') - except ClientError as e: - d.msgbox("Error in pithos+ client: %s" % e.message) - return False + d.infobox("Thank you for using snf-image-creator. Bye", width=53) finally: - out.cleanup() - - d.msgbox("Image `%s' was successfully registered to cyclades as `%s'" % - (session['upload'], name), width=MSGBOX_WIDTH) - return True - - -def kamaki_menu(session): - d = session['dialog'] - default_item = "Account" - 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") + disk.cleanup() - 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"] - 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" + return 0 -def main_menu(session): - d = session['dialog'] - dev = session['device'] - d.setBackgroundTitle("OS: %s, Distro: %s" % (dev.ostype, dev.distro)) - actions = {"Register": kamaki_menu, "Extract": extract_image} - default_item = "Customize" +def select_file(d, media): + """Select a media file""" + if media == '/': + return '/' + default = os.getcwd() + os.sep 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 + 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: - 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") + mode = os.stat(media).st_mode + if not stat.S_ISDIR(mode): + break + default = media + + (code, media) = d.fselect(default, 10, 60, extra_button=1, + title="Please select an input media.", + extra_label="Bundle Host") 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: + return '/' - 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") - out.output("Collecting image metadata...") - metadata = dev.meta - dev.mount(readonly=True) - cls = os_cls(dev.distro, dev.ostype) - image_os = cls(dev.root, dev.g, out) - dev.umount() - metadata.update(image_os.meta) - out.success("done") + # Add extra button in dialog library + dialog._common_args_syntax["extra_button"] = \ + lambda enable: dialog._simple_option("--extra-button", enable) - return metadata + 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") + parser.add_option("--tmpdir", type="string", dest="tmp", default=None, + help="create large temporary image files under DIR", + metavar="DIR") - 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() + if os.geteuid() != 0: + raise FatalError("You must run %s as root" % + parser.get_prog_name()) - # 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) + if options.tmp is not None and not os.path.isdir(options.tmp): + raise FatalError("The directory `%s' specified with --tmpdir is " + "not valid" % options.tmp) - session = {"dialog": d, - "disk": disk, - "snapshot": snapshot, - "device": dev, - "metadata": metadata} - - main_menu(session) - d.infobox("Thank you for using snf-image-creator. Bye", width=53) - finally: - disk.cleanup() - - return 0 - - -def main(): + logfile = None + if options.logfile is not None: + try: + logfile = open(options.logfile, 'w') + except IOError as e: + raise FatalError( + "Unable to open logfile `%s' for writing. Reason: %s" % + (options.logfile, e.strerror)) - d = dialog.Dialog(dialog="dialog") + media = select_file(d, args[0] if len(args) == 1 else None) - while 1: try: - 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 + 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 = create_image(d, media, out, options.tmp) + 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 :