qubes-installer-qubes-os/anaconda/pyanaconda/ui/gui/spokes/storage.py

983 lines
40 KiB
Python
Raw Normal View History

# Storage configuration spoke classes
#
# Copyright (C) 2011, 2012 Red Hat, Inc.
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# the GNU General Public License v.2, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY expressed or implied, including the implied warranties of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
# Public License for more details. You should have received a copy of the
# GNU General Public License along with this program; if not, write to the
# Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA. Any Red Hat trademarks that are incorporated in the
# source code or documentation are not subject to the GNU General Public
# License and may only be used or replicated with the express permission of
# Red Hat, Inc.
#
# Red Hat Author(s): David Lehman <dlehman@redhat.com>
#
"""
TODO:
- add button within sw_needs text in options dialogs 2,3
- udev data gathering
- udev fwraid, mpath would sure be nice
- status/completed
- what are noteworthy status events?
- disks selected
- exclusiveDisks non-empty
- sufficient space for software selection
- autopart selected
- custom selected
- performing custom configuration
- storage configuration complete
- spacing and border width always 6
"""
from gi.repository import Gdk, GLib, AnacondaWidgets
from pyanaconda.ui.communication import hubQ
from pyanaconda.ui.lib.disks import getDisks, isLocalDisk, size_str
from pyanaconda.ui.gui import GUIObject
from pyanaconda.ui.gui.spokes import NormalSpoke
from pyanaconda.ui.gui.spokes.lib.cart import SelectedDisksDialog
from pyanaconda.ui.gui.spokes.lib.passphrase import PassphraseDialog
from pyanaconda.ui.gui.spokes.lib.detailederror import DetailedErrorDialog
from pyanaconda.ui.gui.spokes.lib.resize import ResizeDialog
from pyanaconda.ui.gui.categories.system import SystemCategory
from pyanaconda.ui.gui.utils import enlightbox
from pyanaconda.kickstart import doKickstartStorage, getAvailableDiskSpace
from blivet import empty_device
from blivet.size import Size
from blivet.devices import MultipathDevice
from blivet.errors import StorageError
from blivet.platform import platform
from blivet.devicelibs import swap as swap_lib
from pyanaconda.threads import threadMgr, AnacondaThread
from pyanaconda.product import productName
from pyanaconda.flags import flags
from pyanaconda.i18n import _, N_, P_
from pyanaconda import constants
from pyanaconda.bootloader import BootLoaderError
from pykickstart.constants import CLEARPART_TYPE_NONE, AUTOPART_TYPE_LVM
from pykickstart.errors import KickstartValueError
import sys
import logging
log = logging.getLogger("anaconda")
__all__ = ["StorageSpoke"]
class InstallOptions1Dialog(GUIObject):
builderObjects = ["options1_dialog"]
mainWidgetName = "options1_dialog"
uiFile = "spokes/storage.glade"
# Response ID codes for all the various buttons on all the dialogs.
RESPONSE_CANCEL = 0
RESPONSE_CONTINUE = 1
RESPONSE_MODIFY_SW = 2
RESPONSE_RECLAIM = 3
RESPONSE_QUIT = 4
RESPONSE_CUSTOM = 5
# Which radiobutton is selected on the options1 dialog?
RESPONSE_CONTINUE_NONE = -1
RESPONSE_CONTINUE_AUTOPART = 0
RESPONSE_CONTINUE_RECLAIM = 1
RESPONSE_CONTINUE_CUSTOM = 2
def __init__(self, *args, **kwargs):
self.payload = kwargs.pop("payload", None)
self.showReclaim = kwargs.pop("showReclaim", None)
GUIObject.__init__(self, *args, **kwargs)
self.autoPartType = None
self.encrypted = False
self._grabObjects()
def _grabObjects(self):
self.autoPartTypeCombo = self.builder.get_object("options1_combo")
self.encryptCheckbutton = self.builder.get_object("encryption1_checkbutton")
def run(self):
rc = self.window.run()
self.window.destroy()
return rc
# pylint: disable-msg=W0221
def refresh(self, required_space, auto_swap, disk_free, fs_free, autoPartType, encrypted):
self.autoPartType = autoPartType
self.autoPartTypeCombo.set_active(self.autoPartType)
self.encrypted = encrypted
self.encryptCheckbutton.set_active(self.encrypted)
options_label = self.builder.get_object("options1_label")
options_text = _("You have <b>%(freeSpace)s</b> of free space, which is "
"enough to install %(productName)s. What would you "
"like to do?") % {"freeSpace": disk_free, "productName": productName}
options_label.set_markup(options_text)
label = self.builder.get_object("options1_autopart_radio").get_children()[0]
label.set_markup(_("<span font-desc=\"Cantarell 11\">A_utomatically "
"configure my %(productName)s installation to the "
"disk(s) I selected and return me to the main "
"menu.</span>") % {"productName": productName})
label.set_line_wrap(True)
label.set_use_underline(True)
radio = self.builder.get_object("options1_reclaim_radio")
if self.showReclaim:
label = radio.get_children()[0]
label.set_markup(_("<span font-desc=\"Cantarell 11\">I want more space. "
"_Guide me through shrinking and/or removing partitions "
"so I can have more space for %(productName)s.</span>") % {"productName": productName})
label.set_line_wrap(True)
label.set_use_underline(True)
else:
radio.hide()
label = self.builder.get_object("options1_custom_radio").get_children()[0]
label.set_markup(_("<span font-desc=\"Cantarell 11\">I want to review/_modify "
"my disk partitions before continuing.</span>"))
label.set_line_wrap(True)
label.set_use_underline(True)
@property
def continue_response(self):
if self.builder.get_object("options1_autopart_radio").get_active():
return self.RESPONSE_CONTINUE_AUTOPART
elif self.builder.get_object("options1_reclaim_radio").get_active():
return self.RESPONSE_CONTINUE_RECLAIM
elif self.builder.get_object("options1_custom_radio").get_active():
return self.RESPONSE_CONTINUE_CUSTOM
else:
return self.RESPONSE_CONTINUE_NONE
def _modify_sw_link_clicked(self, label, uri):
if self._software_is_ready():
self.window.response(self.RESPONSE_MODIFY_SW)
return True
def _get_sw_needs_text(self, required_space, auto_swap):
sw_text = (_("Your current <b>%(product)s</b> software "
"selection requires <b>%(total)s</b> of available "
"space, including <b>%(software)s</b> for software and "
"<b>%(swap)s</b> for swap space.")
% {"product": productName,
"total": required_space + auto_swap,
"software": required_space, "swap": auto_swap})
return sw_text
# Methods to handle sensitivity of the modify button.
def _software_is_ready(self):
# FIXME: Would be nicer to just ask the spoke if it's ready.
return (not threadMgr.get(constants.THREAD_PAYLOAD) and
not threadMgr.get(constants.THREAD_PAYLOAD_MD) and
not threadMgr.get(constants.THREAD_SOFTWARE_WATCHER) and
not threadMgr.get(constants.THREAD_CHECK_SOFTWARE) and
self.payload.baseRepo is not None)
def _check_for_storage_thread(self, button):
if self._software_is_ready():
button.set_has_tooltip(False)
button.show_all()
# False means this function should never be called again.
return False
else:
return True
def _add_modify_watcher(self, widgetName):
# If the payload fetching thread is still running, the user can't go to
# modify the software selection screen. Thus, we have to set the button
# insensitive and wait until software selection is ready to go.
modify_widget = self.builder.get_object(widgetName)
if not self._software_is_ready():
GLib.timeout_add_seconds(1, self._check_for_storage_thread, modify_widget)
# signal handlers
def on_type_changed(self, combo):
self.autoPartType = combo.get_active()
def on_encrypt_toggled(self, checkbox):
self.encrypted = checkbox.get_active()
class InstallOptions2Dialog(InstallOptions1Dialog):
builderObjects = ["options2_dialog"]
mainWidgetName = "options2_dialog"
def _grabObjects(self):
self.autoPartTypeCombo = self.builder.get_object("options2_combo")
self.encryptCheckbutton = self.builder.get_object("encryption2_checkbutton")
self.disk_free_label = self.builder.get_object("options2_disk_free_label")
self.fs_free_label = self.builder.get_object("options2_fs_free_label")
def _set_free_space_labels(self, disk_free, fs_free):
disk_free_text = size_str(disk_free)
self.disk_free_label.set_text(disk_free_text)
fs_free_text = size_str(fs_free)
self.fs_free_label.set_text(fs_free_text)
def refresh(self, required_space, auto_swap, disk_free, fs_free, autoPartType, encrypted):
self.autoPartType = autoPartType
self.autoPartTypeCombo.set_active(self.autoPartType)
self.encrypted = encrypted
self.encryptCheckbutton.set_active(self.encrypted)
sw_text = self._get_sw_needs_text(required_space, auto_swap)
label_text = _("%s The disks you've selected have the following "
"amounts of free space:") % sw_text
label = self.builder.get_object("options2_label1")
label.set_markup(label_text)
self._set_free_space_labels(disk_free, fs_free)
label_text = _("<b>You don't have enough space available to install "
"%s</b>. You can shrink or remove existing partitions "
"via our guided reclaim space tool, or you can adjust your "
"partitions on your own in the custom partitioning "
"interface.") % productName
self.builder.get_object("options2_label2").set_markup(label_text)
self._add_modify_watcher("options2_label1")
@property
def continue_response(self):
return self.RESPONSE_CONTINUE_NONE
class InstallOptions3Dialog(InstallOptions1Dialog):
builderObjects = ["options3_dialog"]
mainWidgetName = "options3_dialog"
def _grabObjects(self):
self.disk_free_label = self.builder.get_object("options3_disk_free_label")
self.fs_free_label = self.builder.get_object("options3_fs_free_label")
def _set_free_space_labels(self, disk_free, fs_free):
disk_free_text = size_str(disk_free)
self.disk_free_label.set_text(disk_free_text)
fs_free_text = size_str(fs_free)
self.fs_free_label.set_text(fs_free_text)
def refresh(self, required_space, auto_swap, disk_free, fs_free, autoPartType, encrypted):
sw_text = self._get_sw_needs_text(required_space, auto_swap)
label_text = (_("%(sw_text)s You don't have enough space available to install "
"<b>%(product)s</b>, even if you used all of the free space "
"available on the selected disks.")
% {"sw_text": sw_text, "product": productName})
label = self.builder.get_object("options3_label1")
label.set_markup(label_text)
label.set_tooltip_text(_("Please wait... software metadata still loading."))
label.connect("activate-link", self._modify_sw_link_clicked)
self._set_free_space_labels(disk_free, fs_free)
label_text = _("<b>You don't have enough space available to install "
"%(productName)s</b>, even if you used all of the free space "
"available on the selected disks. You could add more "
"disks for additional space, "
"modify your software selection to install a smaller "
"version of <b>%(productName)s</b>, or quit the installer.") % \
{"productName": productName}
self.builder.get_object("options3_label2").set_markup(label_text)
self._add_modify_watcher("options3_label1")
@property
def continue_response(self):
return self.RESPONSE_CONTINUE_NONE
class StorageChecker(object):
errors = []
warnings = []
_mainSpokeClass = "StorageSpoke"
def __init__(self):
# This is provided by the StorageSpoke class, which is a subclass of
# this one. Backwards, I know.
self.storage = None
def run(self):
threadMgr.add(AnacondaThread(name=constants.THREAD_CHECK_STORAGE,
target=self.checkStorage))
def checkStorage(self):
threadMgr.wait(constants.THREAD_EXECUTE_STORAGE)
hubQ.send_not_ready(self._mainSpokeClass)
hubQ.send_message(self._mainSpokeClass, _("Checking storage configuration..."))
(StorageChecker.errors,
StorageChecker.warnings) = self.storage.sanityCheck()
hubQ.send_ready(self._mainSpokeClass, True)
for e in StorageChecker.errors:
log.error(e)
for w in StorageChecker.warnings:
log.warn(w)
class StorageSpoke(NormalSpoke, StorageChecker):
builderObjects = ["storageWindow", "addSpecializedImage"]
mainWidgetName = "storageWindow"
uiFile = "spokes/storage.glade"
category = SystemCategory
# other candidates: computer-symbolic, folder-symbolic
icon = "drive-harddisk-symbolic"
title = N_("INSTALLATION _DESTINATION")
def __init__(self, *args, **kwargs):
StorageChecker.__init__(self)
NormalSpoke.__init__(self, *args, **kwargs)
self.applyOnSkip = True
self._ready = False
self.autoPartType = None
self.encrypted = False
self.passphrase = ""
self.selected_disks = self.data.ignoredisk.onlyuse[:]
# This list contains all possible disks that can be included in the install.
# All types of advanced disks should be set up for us ahead of time, so
# there should be no need to modify this list.
self.disks = []
if not flags.automatedInstall:
# default to using autopart for interactive installs
self.data.autopart.autopart = True
self.autopart = self.data.autopart.autopart
self.autoPartType = None
self.clearPartType = CLEARPART_TYPE_NONE
self._previous_autopart = False
self._last_clicked_overview = None
self._cur_clicked_overview = None
def _applyDiskSelection(self, use_names):
onlyuse = use_names[:]
for disk in (d for d in self.storage.disks if d.name in onlyuse):
onlyuse.extend(d.name for d in disk.ancestors
if d.name not in onlyuse)
self.data.ignoredisk.onlyuse = onlyuse
self.data.clearpart.drives = use_names[:]
def apply(self):
self._applyDiskSelection(self.selected_disks)
self.data.autopart.autopart = self.autopart
self.data.autopart.type = self.autoPartType
self.data.autopart.encrypted = self.encrypted
self.data.autopart.passphrase = self.passphrase
self.clearPartType = CLEARPART_TYPE_NONE
if self.data.bootloader.bootDrive and \
self.data.bootloader.bootDrive not in self.selected_disks:
self.data.bootloader.bootDrive = ""
self.storage.bootloader.reset()
self.data.clearpart.initAll = True
self.data.clearpart.type = self.clearPartType
self.storage.config.update(self.data)
self.storage.autoPartType = self.data.autopart.type
self.storage.encryptedAutoPart = self.data.autopart.encrypted
self.storage.encryptionPassphrase = self.data.autopart.passphrase
# If autopart is selected we want to remove whatever has been
# created/scheduled to make room for autopart.
# If custom is selected, we want to leave alone any storage layout the
# user may have set up before now.
self.storage.config.clearNonExistent = self.data.autopart.autopart
# refresh the autopart swap size suggestion with currently selected disks
for request in self.storage.autoPartitionRequests:
if request.fstype == "swap":
disk_space = getAvailableDiskSpace(self.storage)
request.size = swap_lib.swapSuggestion(disk_space=disk_space)
break
def execute(self):
# Spawn storage execution as a separate thread so there's no big delay
# going back from this spoke to the hub while StorageChecker.run runs.
# Yes, this means there's a thread spawning another thread. Sorry.
threadMgr.add(AnacondaThread(name=constants.THREAD_EXECUTE_STORAGE,
target=self._doExecute))
def _doExecute(self):
self._ready = False
hubQ.send_not_ready(self.__class__.__name__)
hubQ.send_message(self.__class__.__name__, _("Saving storage configuration..."))
try:
doKickstartStorage(self.storage, self.data, self.instclass)
except (StorageError, KickstartValueError) as e:
log.error("storage configuration failed: %s", e)
StorageChecker.errors = str(e).split("\n")
hubQ.send_message(self.__class__.__name__, _("Failed to save storage configuration..."))
self.data.bootloader.bootDrive = ""
self.data.ignoredisk.drives = []
self.data.ignoredisk.onlyuse = []
self.storage.config.update(self.data)
self.storage.reset()
self.disks = getDisks(self.storage.devicetree)
# now set ksdata back to the user's specified config
self._applyDiskSelection(self.selected_disks)
except BootLoaderError as e:
log.error("BootLoader setup failed: %s", e)
StorageChecker.errors = str(e).split("\n")
hubQ.send_message(self.__class__.__name__, _("Failed to save storage configuration..."))
self.data.bootloader.bootDrive = ""
else:
if self.autopart:
# this was already run as part of doAutoPartition. dumb.
StorageChecker.errors = []
StorageChecker.warnings = []
self.run()
finally:
self._ready = True
hubQ.send_ready(self.__class__.__name__, True)
@property
def completed(self):
retval = (threadMgr.get(constants.THREAD_EXECUTE_STORAGE) is None and
threadMgr.get(constants.THREAD_CHECK_STORAGE) is None and
self.storage.rootDevice is not None and
not self.errors)
return retval
@property
def ready(self):
# By default, the storage spoke is not ready. We have to wait until
# storageInitialize is done.
return self._ready
@property
def showable(self):
return not flags.dirInstall
@property
def status(self):
""" A short string describing the current status of storage setup. """
msg = _("No disks selected")
if flags.automatedInstall and not self.storage.rootDevice:
return msg
elif self.data.ignoredisk.onlyuse:
msg = P_(("%d disk selected"),
("%d disks selected"),
len(self.data.ignoredisk.onlyuse)) % len(self.data.ignoredisk.onlyuse)
if self.errors:
msg = _("Error checking storage configuration")
elif self.warnings:
msg = _("Warning checking storage configuration")
elif self.data.autopart.autopart:
msg = _("Automatic partitioning selected")
else:
msg = _("Custom partitioning selected")
return msg
@property
def localOverviews(self):
return self.local_disks_box.get_children()
@property
def advancedOverviews(self):
return filter(lambda child: isinstance(child, AnacondaWidgets.DiskOverview),
self.specialized_disks_box.get_children())
def _on_disk_clicked(self, overview, event):
# This handler only runs for these two kinds of events, and only for
# activate-type keys (space, enter) in the latter event's case.
if not event.type in [Gdk.EventType.BUTTON_PRESS, Gdk.EventType.KEY_RELEASE]:
return
if event.type == Gdk.EventType.KEY_RELEASE and \
event.keyval not in [Gdk.KEY_space, Gdk.KEY_Return, Gdk.KEY_ISO_Enter, Gdk.KEY_KP_Enter, Gdk.KEY_KP_Space]:
return
if event.type == Gdk.EventType.BUTTON_PRESS and \
event.state & Gdk.ModifierType.SHIFT_MASK:
# clicked with Shift held down
if self._last_clicked_overview is None:
# nothing clicked before, cannot apply Shift-click
return
local_overviews = self.localOverviews
advanced_overviews = self.advancedOverviews
# find out which list of overviews the clicked one belongs to
if overview in local_overviews:
from_overviews = local_overviews
elif overview in advanced_overviews:
from_overviews = advanced_overviews
else:
# should never happen, but if it does, no other actions should be done
return
if self._last_clicked_overview in from_overviews:
# get index of the last clicked overview
last_idx = from_overviews.index(self._last_clicked_overview)
else:
# overview from the other list clicked before, cannot apply "Shift-click"
return
# get index and state of the clicked overview
cur_idx = from_overviews.index(overview)
state = self._last_clicked_overview.get_chosen()
if cur_idx > last_idx:
copy_to = from_overviews[last_idx:cur_idx+1]
else:
copy_to = from_overviews[cur_idx:last_idx]
# copy the state of the last clicked overview to the ones between it and the
# one clicked with the Shift held down
for disk_overview in copy_to:
disk_overview.set_chosen(state)
self._update_disk_list()
self._update_summary()
def _on_disk_focus_in(self, overview, event):
self._last_clicked_overview = self._cur_clicked_overview
self._cur_clicked_overview = overview
def refresh(self):
self.disks = getDisks(self.storage.devicetree)
# synchronize our local data store with the global ksdata
disk_names = [d.name for d in self.disks]
# don't put disks with hidden formats in selected_disks
self.selected_disks = [d for d in self.data.ignoredisk.onlyuse
if d in disk_names]
self.autopart = self.data.autopart.autopart
self.autoPartType = self.data.autopart.type
if self.autoPartType is None:
self.autoPartType = AUTOPART_TYPE_LVM
self.encrypted = self.data.autopart.encrypted
self.passphrase = self.data.autopart.passphrase
self._previous_autopart = self.autopart
# First, remove all non-button children.
for child in self.localOverviews + self.advancedOverviews:
child.destroy()
# Then deal with local disks, which are really easy. They need to be
# handled here instead of refresh to take into account the user pressing
# the rescan button on custom partitioning.
for disk in filter(isLocalDisk, self.disks):
self._add_disk_overview(disk, self.local_disks_box)
# Advanced disks are different. Because there can potentially be a lot
# of them, we do not display them in the box by default. Instead, only
# those selected in the filter UI are displayed. This means refresh
# needs to know to create and destroy overviews as appropriate.
for name in self.data.ignoredisk.onlyuse:
if name not in disk_names:
continue
obj = self.storage.devicetree.getDeviceByName(name, hidden=True)
if isLocalDisk(obj):
continue
self._add_disk_overview(obj, self.specialized_disks_box)
# update the selections in the ui
for overview in self.localOverviews + self.advancedOverviews:
name = overview.get_property("name")
overview.set_chosen(name in self.selected_disks)
self._update_summary()
if self.errors:
self.set_warning(_("Error checking storage configuration. Click for details."))
elif self.warnings:
self.set_warning(_("Warning checking storage configuration. Click for details."))
def initialize(self):
NormalSpoke.initialize(self)
# Wouldn't it be nice if glade knew how to do this?
label = self.builder.get_object("summary_button").get_children()[0]
markup = "<span foreground='blue'><u>%s</u></span>" % label.get_text()
label.set_use_markup(True)
label.set_markup(markup)
specializedButton = self.builder.get_object("addSpecializedButton")
# It's uh... uh... it's down there somewhere, let me take another look.
label = specializedButton.get_children()[0].get_children()[0].get_children()[1]
markup = "<span size='large'><b>%s</b></span>" % label.get_text()
label.set_use_markup(True)
label.set_markup(markup)
specializedButton.show_all()
self.local_disks_box = self.builder.get_object("local_disks_box")
self.specialized_disks_box = self.builder.get_object("specialized_disks_box")
threadMgr.add(AnacondaThread(name=constants.THREAD_STORAGE_WATCHER,
target=self._initialize))
def _add_disk_overview(self, disk, box):
if disk.removable:
kind = "drive-removable-media"
else:
kind = "drive-harddisk"
size = size_str(disk.size)
if disk.serial:
popup_info = "%s" % disk.serial
else:
popup_info = None
# We don't want to display the whole huge WWID for a multipath device.
# That makes the DO way too wide.
if isinstance(disk, MultipathDevice):
desc = disk.wwid.split(":")
description = ":".join(desc[0:3]) + "..." + ":".join(desc[-5:-1])
else:
description = disk.description
free = self.storage.getFreeSpace(disks=[disk])[disk.name][0]
overview = AnacondaWidgets.DiskOverview(description,
kind,
size,
_("%s free") % size_str(free),
disk.name,
popup=popup_info)
box.pack_start(overview, False, False, 0)
# FIXME: this will need to get smarter
#
# maybe a little function that resolves each item in onlyuse using
# udev_resolve_devspec and compares that to the DiskDevice?
overview.set_chosen(disk.name in self.selected_disks)
overview.connect("button-press-event", self._on_disk_clicked)
overview.connect("key-release-event", self._on_disk_clicked)
overview.connect("focus-in-event", self._on_disk_focus_in)
overview.show_all()
def _initialize(self):
hubQ.send_message(self.__class__.__name__, _("Probing storage..."))
threadMgr.wait(constants.THREAD_STORAGE)
threadMgr.wait(constants.THREAD_CUSTOM_STORAGE_INIT)
self.disks = getDisks(self.storage.devicetree)
# if there's only one disk, select it by default
if len(self.disks) == 1 and not self.selected_disks:
self._applyDiskSelection([self.disks[0].name])
self._ready = True
hubQ.send_ready(self.__class__.__name__, False)
def _update_summary(self):
""" Update the summary based on the UI. """
count = 0
capacity = 0
free = Size(bytes=0)
# pass in our disk list so hidden disks' free space is available
free_space = self.storage.getFreeSpace(disks=self.disks)
selected = [d for d in self.disks if d.name in self.selected_disks]
for disk in selected:
capacity += disk.size
free += free_space[disk.name][0]
count += 1
summary = (P_("%(count)d disk selected; %(capacity)s capacity; %(free)s free",
"%(count)d disks selected; %(capacity)s capacity; %(free)s free",
count) % {"count" : count,
"capacity" : str(Size(en_spec="%f MB" % capacity)),
"free" : free})
summary_label = self.builder.get_object("summary_label")
summary_label.set_text(summary)
summary_label.set_sensitive(count > 0)
summary_button = self.builder.get_object("summary_button")
summary_button.set_visible(count > 0)
if len(self.disks) == 0:
self.set_warning(_("No disks detected. Please shut down the computer, connect at least one disk, and restart to complete installation."))
elif count == 0:
self.set_warning(_("No disks selected; please select at least one disk to install to."))
else:
self.clear_info()
def _update_disk_list(self):
""" Update self.selected_disks based on the UI. """
for overview in self.localOverviews + self.advancedOverviews:
selected = overview.get_chosen()
name = overview.get_property("name")
if selected and name not in self.selected_disks:
self.selected_disks.append(name)
if not selected and name in self.selected_disks:
self.selected_disks.remove(name)
# signal handlers
def on_summary_clicked(self, button):
# show the selected disks dialog
# pass in our disk list so hidden disks' free space is available
free_space = self.storage.getFreeSpace(disks=self.disks)
dialog = SelectedDisksDialog(self.data,)
dialog.refresh([d for d in self.disks if d.name in self.selected_disks],
free_space)
self.run_lightbox_dialog(dialog)
# update selected disks since some may have been removed
self.selected_disks = [d.name for d in dialog.disks]
# update the UI to reflect changes to self.selected_disks
for overview in self.localOverviews:
name = overview.get_property("name")
overview.set_chosen(name in self.selected_disks)
self._update_summary()
self.data.bootloader.seen = True
if self.data.bootloader.location == "none":
self.set_warning(_("You have chosen to skip bootloader installation. Your system may not be bootable."))
self.window.show_all()
else:
self.clear_info()
def run_lightbox_dialog(self, dialog):
with enlightbox(self.window, dialog.window):
rc = dialog.run()
return rc
def _check_encrypted(self):
# even if they're not doing autopart, setting autopart.encrypted
# establishes a default of encrypting new devices
if not self.encrypted:
return True
dialog = PassphraseDialog(self.data)
rc = self.run_lightbox_dialog(dialog)
if rc == 0:
return False
self.passphrase = dialog.passphrase
return True
def on_back_clicked(self, button):
# We can't exit early if it looks like nothing has changed because the
# user might want to change settings presented in the dialogs shown from
# within this method.
# Remove all non-existing devices if autopart was active when we last
# refreshed.
if self._previous_autopart:
self._previous_autopart = False
for partition in self.storage.partitions[:]:
# check if it's been removed in a previous iteration
if not partition.exists and \
partition in self.storage.partitions:
self.storage.recursiveRemove(partition)
# hide/unhide disks as requested
for disk in self.disks:
if disk.name not in self.selected_disks and \
disk in self.storage.devices:
self.storage.devicetree.hide(disk)
elif disk.name in self.selected_disks and \
disk not in self.storage.devices:
self.storage.devicetree.unhide(disk)
# show the installation options dialog
disks = [d for d in self.disks if d.name in self.selected_disks]
disks_size = sum(Size(en_spec="%f MB" % d.size) for d in disks)
# No disks selected? The user wants to back out of the storage spoke.
if not disks:
NormalSpoke.on_back_clicked(self, button)
return
# Figure out if the existing disk labels will work on this platform
# you need to have at least one of the platform's labels in order for
# any of the free space to be useful.
disk_labels = set(disk.format.labelType for disk in disks
if hasattr(disk.format, "labelType"))
platform_labels = set(platform.diskLabelTypes)
if disk_labels and platform_labels.isdisjoint(disk_labels):
disk_free = 0
fs_free = 0
log.debug("Need disklabel: %s have: %s", ", ".join(platform_labels),
", ".join(disk_labels))
else:
free_space = self.storage.getFreeSpace(disks=disks,
clearPartType=CLEARPART_TYPE_NONE)
disk_free = sum(f[0] for f in free_space.itervalues())
fs_free = sum(f[1] for f in free_space.itervalues())
required_space = self.payload.spaceRequired
auto_swap = Size(bytes=0)
for autoreq in self.storage.autoPartitionRequests:
if autoreq.fstype == "swap":
auto_swap += Size(en_spec="%d MB" % autoreq.size)
log.debug("disk free: %s fs free: %s sw needs: %s auto swap: %s",
disk_free, fs_free, required_space, auto_swap)
if disk_free >= required_space + auto_swap:
showReclaim = not all(map(lambda dev: empty_device(dev, self.storage.devicetree),
self.disks))
dialog = InstallOptions1Dialog(self.data, showReclaim=showReclaim)
elif disks_size >= required_space:
dialog = InstallOptions2Dialog(self.data, payload=self.payload)
else:
dialog = InstallOptions3Dialog(self.data, payload=self.payload)
dialog.refresh(required_space, auto_swap, disk_free, fs_free, self.autoPartType,
self.encrypted)
rc = self.run_lightbox_dialog(dialog)
if rc == dialog.RESPONSE_CONTINUE:
self.autoPartType = dialog.autoPartType
self.encrypted = dialog.encrypted
if not self._check_encrypted():
return
if dialog.continue_response == dialog.RESPONSE_CONTINUE_AUTOPART:
self.autopart = True
elif dialog.continue_response == dialog.RESPONSE_CONTINUE_RECLAIM:
self.apply()
if not self._show_resize_dialog(disks):
# User pressed cancel on the reclaim dialog, so don't leave
# the storage spoke.
return
# if the user did not press cancel that means they want to
# proceed with autopart
self.autopart = True
elif dialog.continue_response == dialog.RESPONSE_CONTINUE_CUSTOM:
self.autopart = False
self.skipTo = "CustomPartitioningSpoke"
elif rc == dialog.RESPONSE_CANCEL:
# stay on this spoke
return
elif rc == dialog.RESPONSE_MODIFY_SW:
# go to software spoke
self.skipTo = "SoftwareSelectionSpoke"
elif rc == dialog.RESPONSE_RECLAIM:
self.autoPartType = dialog.autoPartType
self.encrypted = dialog.encrypted
if not self._check_encrypted():
return
self.apply()
if not self._show_resize_dialog(disks):
# User pressed cancel on the reclaim dialog, so don't leave
# the storage spoke.
return
# if the user did not press cancel that means they want to
# proceed with autopart
self.autopart = True
elif rc == dialog.RESPONSE_QUIT:
raise SystemExit("user-selected exit")
elif rc == dialog.RESPONSE_CUSTOM:
self.autopart = False
self.autoPartType = dialog.autoPartType
self.encrypted = dialog.encrypted
self.skipTo = "CustomPartitioningSpoke"
self.applyOnSkip = True
NormalSpoke.on_back_clicked(self, button)
def _show_resize_dialog(self, disks):
resizeDialog = ResizeDialog(self.data, self.storage, self.payload)
resizeDialog.refresh(disks)
rc = self.run_lightbox_dialog(resizeDialog)
return rc
def on_specialized_clicked(self, button):
# Don't want to run apply or execute in this case, since we have to
# collect some more disks first. The user will be back to this spoke.
self.applyOnSkip = False
# However, we do want to apply current selections so the disk cart off
# the filter spoke will display the correct information.
self._applyDiskSelection(self.selected_disks)
self.skipTo = "FilterSpoke"
NormalSpoke.on_back_clicked(self, button)
def on_info_bar_clicked(self, *args):
if self.errors:
label = _("The following errors were encountered when checking your storage "
"configuration. You can modify your storage layout or quit the "
"installer.")
dialog = DetailedErrorDialog(self.data, buttons=[_("_Quit"), _("_Modify Storage Layout")], label=label)
with enlightbox(self.window, dialog.window):
errors = "\n".join(self.errors)
dialog.refresh(errors)
rc = dialog.run()
dialog.window.destroy()
if rc == 0:
# Quit.
sys.exit(0)
elif self.warnings:
label = _("The following warnings were encountered when checking your storage "
"configuration. These are not fatal, but you may wish to make "
"changes to your storage layout.")
dialog = DetailedErrorDialog(self.data, buttons=[_("_OK")], label=label)
with enlightbox(self.window, dialog.window):
warnings = "\n".join(self.warnings)
dialog.refresh(warnings)
rc = dialog.run()
dialog.window.destroy()
def on_disks_key_released(self, box, event):
# we want to react only on Ctrl-A being pressed
if not bool(event.state & Gdk.ModifierType.CONTROL_MASK) or \
(event.keyval not in (Gdk.KEY_a, Gdk.KEY_A)):
return
# select disks in the right box
if box is self.local_disks_box:
overviews = self.localOverviews
elif box is self.specialized_disks_box:
overviews = self.advancedOverviews
else:
# no other box contains disk overviews
return
for overview in overviews:
overview.set_chosen(True)
self._update_disk_list()