#
# Select which disks to clear and which ones to just mount.
#
# Copyright (C) 2009 Red Hat, Inc.
# All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty 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, see .
#
import gtk, gobject
import gui
from DeviceSelector import *
from constants import *
import isys
from iw_gui import *
from storage.udev import *
import gettext
_ = lambda x: gettext.ldgettext("anaconda", x)
class ClearDisksWindow (InstallWindow):
windowTitle = N_("Clear Disks Selector")
def getNext (self):
# All the rows that are visible in the right hand side should be cleared.
cleardisks = []
for row in self.store:
if row[self.rightVisible]:
cleardisks.append(row[OBJECT_COL].name)
if len(cleardisks) == 0:
self.anaconda.intf.messageWindow(_("Error"),
_("You must select at least one "
"drive to be used for installation."),
custom_icon="error")
raise gui.StayOnScreen
# The selected row is the disk to boot from.
selected = self.rightDS.getSelected()
if len(selected) == 0:
self.anaconda.intf.messageWindow(_("Error"),
_("You must select one drive to "
"boot from."),
custom_icon="error")
raise gui.StayOnScreen
bootDisk = selected[0][OBJECT_COL].name
cleardisks.sort(self.anaconda.storage.compareDisks)
self.anaconda.storage.clearPartDisks = cleardisks
self.anaconda.bootloader.updateDriveList([bootDisk])
def getScreen (self, anaconda):
# We can't just use exclusiveDisks here because of kickstart. First,
# the kickstart file could have used ignoredisk --drives= in which case
# exclusiveDisks would be empty. Second, ignoredisk is entirely
# optional in which case neither list would be populated. Luckily,
# storage.disks takes isIgnored into account and that handles both these
# issues.
disks = filter(lambda d: not d.format.hidden, anaconda.storage.disks)
# Skip this screen as well if there's only one disk to use.
if len(disks) == 1:
anaconda.storage.clearPartDisks = [disks[0].name]
anaconda.bootloader.drivelist = [disks[0].name]
return None
(xml, self.vbox) = gui.getGladeWidget("cleardisks.glade", "vbox")
self.leftScroll = xml.get_widget("leftScroll")
self.rightScroll = xml.get_widget("rightScroll")
self.addButton = xml.get_widget("addButton")
self.removeButton = xml.get_widget("removeButton")
self.installTargetImage = xml.get_widget("installTargetImage")
self.installTargetTip = xml.get_widget("installTargetTip")
self.anaconda = anaconda
self.leftVisible = 1
self.leftActive = 2
self.rightVisible = 4
self.rightActive = 5
# One store for both views. First the obejct, then a visible/active for
# the left hand side, then a visible/active for the right hand side, then
# all the other stuff.
#
# NOTE: the third boolean is a placeholder. DeviceSelector uses the third
# slot in the store to determine whether the row is immutable or not. We
# just need to put False in there for everything.
self.store = gtk.TreeStore(gobject.TYPE_PYOBJECT,
gobject.TYPE_BOOLEAN, gobject.TYPE_BOOLEAN,
gobject.TYPE_BOOLEAN,
gobject.TYPE_BOOLEAN, gobject.TYPE_BOOLEAN,
gobject.TYPE_STRING, gobject.TYPE_STRING,
gobject.TYPE_STRING, gobject.TYPE_STRING,
gobject.TYPE_STRING)
self.store.set_sort_column_id(6, gtk.SORT_ASCENDING)
# The left view shows all the drives that will just be mounted, but
# can still be moved to the right hand side.
self.leftFilteredModel = self.store.filter_new()
self.leftSortedModel = gtk.TreeModelSort(self.leftFilteredModel)
self.leftTreeView = gtk.TreeView(self.leftSortedModel)
self.leftFilteredModel.set_visible_func(lambda model, iter, view: model.get_value(iter, self.leftVisible), self.leftTreeView)
self.leftScroll.add(self.leftTreeView)
self.leftDS = DeviceSelector(self.store, self.leftSortedModel,
self.leftTreeView, visible=self.leftVisible,
active=self.leftActive)
self.leftDS.createMenu()
self.leftDS.addColumn(_("Model"), 6)
self.leftDS.addColumn(_("Capacity"), 7)
self.leftDS.addColumn(_("Vendor"), 8)
self.leftDS.addColumn(_("Interconnect"), 9, displayed=False)
self.leftDS.addColumn(_("Serial Number"), 10, displayed=False)
# The right view show all the drives that will be wiped during install.
self.rightFilteredModel = self.store.filter_new()
self.rightSortedModel = gtk.TreeModelSort(self.rightFilteredModel)
self.rightTreeView = gtk.TreeView(self.rightSortedModel)
self.rightFilteredModel.set_visible_func(lambda model, iter, view: model.get_value(iter, self.rightVisible), self.rightTreeView)
self.rightScroll.add(self.rightTreeView)
self.rightDS = DeviceSelector(self.store, self.rightSortedModel,
self.rightTreeView, visible=self.rightVisible,
active=self.rightActive)
self.rightDS.createSelectionCol(title=_("Boot"), radioButton=True)
self.rightDS.addColumn(_("Model"), 6)
self.rightDS.addColumn(_("Capacity"), 7)
# Store the first disk (according to our detected BIOS order) for
# auto boot device selection
names = map(lambda d: d.name, disks)
self.bootDisk = sorted(names, self.anaconda.storage.compareDisks)[0]
# The device filtering UI set up exclusiveDisks as a list of the names
# of all the disks we should use later on. Now we need to go get those,
# look up some more information in the devicetree, and set up the
# selector.
for d in disks:
rightVisible = d.name in self.anaconda.storage.clearPartDisks
rightActive = rightVisible and \
d.name in self.anaconda.bootloader.drivelist[:1]
leftVisible = not rightVisible
self.store.append(None, (d,
leftVisible, True, False,
rightVisible, rightActive,
d.model,
str(int(d.size)) + " MB",
d.vendor, "", d.serial))
self.addButton.connect("clicked", self._add_clicked)
self.removeButton.connect("clicked", self._remove_clicked)
# Also allow moving devices back and forth with double click, enter, etc.
self.leftTreeView.connect("row-activated", self._add_clicked)
self.rightTreeView.connect("row-activated", self._remove_clicked)
# And let the user select multiple devices at a time.
self.leftTreeView.get_selection().set_mode(gtk.SELECTION_MULTIPLE)
self.rightTreeView.get_selection().set_mode(gtk.SELECTION_MULTIPLE)
if self.anaconda.storage.clearPartType == CLEARPART_TYPE_LINUX:
self.installTargetTip.set_markup(_("Tip: All Linux filesystems on install target devices will be reformatted and wiped of any data. Make sure you have backups."))
elif self.anaconda.storage.clearPartType == CLEARPART_TYPE_ALL:
self.installTargetTip.set_markup(_("Tip: Install target devices will be reformatted and wiped of any data. Make sure you have backups."))
else:
self.installTargetTip.set_markup(_("Tip: Your filesystems on install target devices will not be wiped unless you choose to do so during customization."))
return self.vbox
def _autoSelectBootDisk(self):
if self.rightDS.getSelected():
return
for row in self.store:
if row[OBJECT_COL].name == self.bootDisk and row[self.rightVisible]:
row[self.rightActive] = True
def _add_clicked(self, widget, *args, **kwargs):
(sortedModel, pathlist) = self.leftTreeView.get_selection().get_selected_rows()
if not pathlist:
return
for path in reversed(pathlist):
sortedIter = sortedModel.get_iter(path)
if not sortedIter:
continue
filteredIter = self.leftSortedModel.convert_iter_to_child_iter(None, sortedIter)
iter = self.leftFilteredModel.convert_iter_to_child_iter(filteredIter)
self.store.set_value(iter, self.leftVisible, False)
self.store.set_value(iter, self.rightVisible, True)
self.store.set_value(iter, self.rightActive, False)
self._autoSelectBootDisk()
self.leftFilteredModel.refilter()
self.rightFilteredModel.refilter()
def _remove_clicked(self, widget, *args, **kwargs):
(sortedModel, pathlist) = self.rightTreeView.get_selection().get_selected_rows()
if not pathlist:
return
for path in reversed(pathlist):
sortedIter = sortedModel.get_iter(path)
if not sortedIter:
continue
filteredIter = self.rightSortedModel.convert_iter_to_child_iter(None, sortedIter)
iter = self.rightFilteredModel.convert_iter_to_child_iter(filteredIter)
self.store.set_value(iter, self.leftVisible, True)
self.store.set_value(iter, self.rightVisible, False)
self.store.set_value(iter, self.rightActive, False)
self._autoSelectBootDisk()
self.leftFilteredModel.refilter()
self.rightFilteredModel.refilter()