| 54 |
54 |
INPUTBOX_WIDTH = 70
|
| 55 |
55 |
CHECKBOX_WIDTH = 70
|
| 56 |
56 |
HELP_WIDTH = 70
|
|
57 |
INFOBOX_WIDTH = 70
|
| 57 |
58 |
|
| 58 |
59 |
CONFIGURATION_TASKS = [
|
| 59 |
60 |
("Partition table manipulation", ["FixPartitionTable"],
|
| ... | ... | |
| 82 |
83 |
|
| 83 |
84 |
|
| 84 |
85 |
def confirm_reset(d):
|
| 85 |
|
return not d.yesno(
|
| 86 |
|
"Are you sure you want to reset everything?",
|
| 87 |
|
width=YESNO_WIDTH)
|
|
86 |
return not d.yesno("Are you sure you want to reset everything?",
|
|
87 |
width=YESNO_WIDTH)
|
| 88 |
88 |
|
| 89 |
89 |
|
| 90 |
90 |
def update_background_title(session):
|
| ... | ... | |
| 94 |
94 |
MB = 2 ** 20
|
| 95 |
95 |
|
| 96 |
96 |
size = (dev.meta['SIZE'] + MB - 1) // MB
|
| 97 |
|
title = "OS: %s, Distro: %s, Size: %dMB" % (dev.ostype, dev.distro, size)
|
|
97 |
shrinked = 'shrinked' in session and session['shrinked'] == True
|
|
98 |
postfix = " (shrinked)" if shrinked else ''
|
|
99 |
|
|
100 |
title = "OS: %s, Distro: %s, Size: %dMB%s" % \
|
|
101 |
(dev.ostype, dev.distro, size, postfix)
|
| 98 |
102 |
|
| 99 |
103 |
d.setBackgroundTitle(title)
|
| 100 |
104 |
|
| ... | ... | |
| 408 |
412 |
|
| 409 |
413 |
(code, choice) = d.menu(
|
| 410 |
414 |
"In this menu you can edit existing image properties or add new "
|
| 411 |
|
"ones. Be carefull! Most properties have special meaning and "
|
|
415 |
"ones. Be careful! Most properties have special meaning and "
|
| 412 |
416 |
"alter the image deployment behaviour. Press <HELP> to see more "
|
| 413 |
417 |
"information about image properties. Press <BACK> when done.",
|
| 414 |
418 |
height=18, width=MENU_WIDTH, choices=choices, menu_height=10,
|
| ... | ... | |
| 519 |
523 |
d = session['dialog']
|
| 520 |
524 |
image_os = session['image_os']
|
| 521 |
525 |
|
|
526 |
# Is the image already shrinked?
|
|
527 |
if 'shrinked' in session and session['shrinked'] == True:
|
|
528 |
msg = "It seems you have shrinked the image. Running system " \
|
|
529 |
"preparation tasks on a shrinked image is dangerous."
|
|
530 |
|
|
531 |
if d.yesno("%s\n\nDo you really want to continue?" % msg,
|
|
532 |
width=YESNO_WIDTH, defaultno=1):
|
|
533 |
return
|
|
534 |
|
| 522 |
535 |
wrapper = textwrap.TextWrapper(width=65)
|
| 523 |
536 |
|
| 524 |
537 |
help_title = "System Preperation Tasks"
|
| ... | ... | |
| 531 |
544 |
# Only give the user the choice between syspreps that have not ran yet
|
| 532 |
545 |
syspreps = [s for s in all_syspreps if s not in session['exec_syspreps']]
|
| 533 |
546 |
|
|
547 |
if len(syspreps) == 0:
|
|
548 |
d.msgbox("No system preparation task left to run!", width=MSGBOX_WIDTH)
|
|
549 |
return
|
|
550 |
|
| 534 |
551 |
while 1:
|
| 535 |
552 |
choices = []
|
| 536 |
553 |
index = 0
|
| ... | ... | |
| 576 |
593 |
image_os.out = out
|
| 577 |
594 |
image_os.do_sysprep()
|
| 578 |
595 |
|
|
596 |
for (k, v) in image_os.meta.items():
|
|
597 |
session['metadata'][str(k)] = str(v)
|
|
598 |
|
| 579 |
599 |
# Disable syspreps that have ran
|
| 580 |
600 |
for sysprep in session['exec_syspreps']:
|
| 581 |
601 |
image_os.disable_sysprep(sysprep)
|
| ... | ... | |
| 588 |
608 |
break
|
| 589 |
609 |
|
| 590 |
610 |
|
| 591 |
|
def customize_menu(session):
|
|
611 |
def shrink(session):
|
|
612 |
d = session['dialog']
|
|
613 |
dev = session['device']
|
|
614 |
|
|
615 |
shrinked = 'shrinked' in session and session['shrinked'] == True
|
|
616 |
|
|
617 |
if shrinked:
|
|
618 |
d.msgbox("You have already shrinked your image!")
|
|
619 |
return
|
|
620 |
|
|
621 |
msg = "This operation will shrink the last partition of the image to " \
|
|
622 |
"reduce the total image size. If the last partition is a swap " \
|
|
623 |
"partition, then this partition is removed and the partition " \
|
|
624 |
"before that is shrinked. The removed swap partition will be " \
|
|
625 |
"recreated during image deployment."
|
|
626 |
|
|
627 |
if not d.yesno("%s\n\nDo you want to continue?" % msg, width=70,
|
|
628 |
height=12, title="Image Shrinking"):
|
|
629 |
dev.out = InfoBoxOutput(d, "Image Shrinking", height=3)
|
|
630 |
session['metadata']['SIZE'] = str(dev.shrink())
|
|
631 |
session['shrinked'] = True
|
|
632 |
update_background_title(session)
|
|
633 |
dev.out.finalize()
|
|
634 |
|
|
635 |
|
|
636 |
def customization_menu(session):
|
| 592 |
637 |
d = session['dialog']
|
| 593 |
638 |
|
| 594 |
639 |
choices = [("Sysprep", "Run various image preperation tasks"),
|
| ... | ... | |
| 600 |
645 |
default_item = "Sysprep"
|
| 601 |
646 |
|
| 602 |
647 |
actions = {"Sysprep": sysprep,
|
|
648 |
"Shrink": shrink,
|
| 603 |
649 |
"View/Modify": modify_properties,
|
| 604 |
650 |
"Delete": delete_properties,
|
| 605 |
651 |
"Exclude": exclude_tasks}
|
| ... | ... | |
| 631 |
677 |
|
| 632 |
678 |
default_item = "Customize"
|
| 633 |
679 |
|
| 634 |
|
actions = {"Customize": customize_menu, "Register": kamaki_menu,
|
|
680 |
actions = {"Customize": customization_menu, "Register": kamaki_menu,
|
| 635 |
681 |
"Extract": extract_image}
|
| 636 |
682 |
while 1:
|
| 637 |
683 |
(code, choice) = d.menu(
|
| ... | ... | |
| 646 |
692 |
break
|
| 647 |
693 |
elif choice == "Reset":
|
| 648 |
694 |
if confirm_reset(d):
|
| 649 |
|
d.infobox("Resetting snf-image-creator. Please wait...")
|
|
695 |
d.infobox("Resetting snf-image-creator. Please wait...",
|
|
696 |
width=INFOBOX_WIDTH)
|
| 650 |
697 |
raise Reset
|
| 651 |
698 |
elif choice in actions:
|
| 652 |
699 |
actions[choice](session)
|
| ... | ... | |
| 681 |
728 |
sys.stderr.write("%s\n" % usage)
|
| 682 |
729 |
return 1
|
| 683 |
730 |
|
|
731 |
d.setBackgroundTitle('snf-image-creator')
|
|
732 |
|
| 684 |
733 |
if os.geteuid() != 0:
|
| 685 |
734 |
raise FatalError("You must run %s as root" % basename)
|
| 686 |
735 |
|
| 687 |
736 |
media = select_file(d, sys.argv[1] if len(sys.argv) == 2 else None)
|
| 688 |
737 |
|
| 689 |
|
d.setBackgroundTitle('snf-image-creator')
|
| 690 |
|
|
| 691 |
738 |
out = GaugeOutput(d, "Initialization", "Initializing...")
|
| 692 |
739 |
disk = Disk(media, out)
|
| 693 |
740 |
|
| ... | ... | |
| 701 |
748 |
dev = disk.get_device(snapshot)
|
| 702 |
749 |
|
| 703 |
750 |
out.output("Collecting image metadata...")
|
| 704 |
|
metadata = dev.meta
|
|
751 |
|
|
752 |
metadata = {}
|
|
753 |
for (key, value) in dev.meta.items():
|
|
754 |
metadata[str(key)] = str(value)
|
|
755 |
|
| 705 |
756 |
dev.mount(readonly=True)
|
| 706 |
757 |
cls = os_cls(dev.distro, dev.ostype)
|
| 707 |
758 |
image_os = cls(dev.root, dev.g, out)
|
| 708 |
759 |
dev.umount()
|
| 709 |
|
metadata.update(image_os.meta)
|
|
760 |
|
|
761 |
for (key, value) in image_os.meta.items():
|
|
762 |
metadata[str(key)] = str(value)
|
|
763 |
|
| 710 |
764 |
out.success("done")
|
| 711 |
765 |
out.cleanup()
|
| 712 |
766 |
|
| ... | ... | |
| 748 |
802 |
sys.exit(ret)
|
| 749 |
803 |
except FatalError as e:
|
| 750 |
804 |
msg = textwrap.fill(str(e), width=70)
|
| 751 |
|
d.infobox(msg, width=70, title="Fatal Error")
|
|
805 |
d.infobox(msg, width=INFOBOX_WIDTH, title="Fatal Error")
|
| 752 |
806 |
sys.exit(1)
|
| 753 |
807 |
except Reset:
|
| 754 |
808 |
continue
|