Initial import from Sabayon (ver 0.9.9.56)

1 files changed, 284 insertions, 0 deletions

diff --git a/iw/autopart_type.py b/iw/autopart_type.py
new file mode 100644
index 0000000..2ba1075
--- /dev/null
+++ b/iw/autopart_type.py
@@ -0,0 +1,284 @@
+#
+# autopart_type.py: Allows the user to choose how they want to partition
+#
+# Copyright (C) 2005, 2006  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 <http://www.gnu.org/licenses/>.
+#
+# Author(s): Jeremy Katz <katzj@redhat.com>
+#
+
+import gtk
+import gobject
+import math
+
+from constants import *
+import gui
+from partition_ui_helpers_gui import *
+from pixmapRadioButtonGroup_gui import pixmapRadioButtonGroup
+
+from iw_gui import *
+from flags import flags
+from storage.deviceaction import *
+
+import gettext
+_ = lambda x: gettext.ldgettext("anaconda", x)
+
+def whichToShrink(storage, intf):
+    def getActive(combo):
+        act = combo.get_active_iter()
+        return combo.get_model().get_value(act, 1)
+
+    def comboCB(combo, shrinkSB):
+        # partition to resize changed, let's update our spinbutton
+        newSize = shrinkSB.get_value_as_int()
+
+        part = getActive(combo)
+        reqlower = long(math.ceil(part.format.minSize))
+        requpper = long(math.floor(part.format.currentSize))
+
+        adj = shrinkSB.get_adjustment()
+        adj.lower = reqlower
+        adj.upper = requpper
+        adj.set_value(reqlower)
+
+
+    (dxml, dialog) = gui.getGladeWidget("autopart.glade", "shrinkDialog")
+
+    store = gtk.TreeStore(gobject.TYPE_STRING, gobject.TYPE_PYOBJECT)
+    combo = dxml.get_widget("shrinkPartCombo")
+    combo.set_model(store)
+    crt = gtk.CellRendererText()
+    combo.pack_start(crt, True)
+    combo.set_attributes(crt, text = 0)
+    combo.connect("changed", comboCB, dxml.get_widget("shrinkSB"))
+
+    biggest = -1
+    for part in storage.partitions:
+        if not part.exists:
+            continue
+
+        entry = None
+        if part.resizable and part.format.resizable:
+            entry = ("%s (%s, %d MB)" % (part.name,
+                                         part.format.name,
+                                         math.floor(part.format.size)),
+                     part)
+
+        if entry:
+            i = store.append(None)
+            store[i] = entry
+            combo.set_active_iter(i)
+
+            if biggest == -1:
+                biggest = i
+            else:
+                current = store.get_value(biggest, 1)
+                if part.format.targetSize > current.format.targetSize:
+                    biggest = i
+
+    if biggest > -1:
+        combo.set_active_iter(biggest)
+
+    if len(store) == 0:
+        dialog.destroy()
+        intf.messageWindow(_("Error"),
+                           _("No partitions are available to resize.  Only "
+                             "physical partitions with specific filesystems "
+                             "can be resized."),
+                             type="warning", custom_icon="error")
+        return (gtk.RESPONSE_CANCEL, [])
+
+    gui.addFrame(dialog)
+    dialog.show_all()
+    runResize = True
+
+    while runResize:
+        rc = dialog.run()
+        if rc != gtk.RESPONSE_OK:
+            dialog.destroy()
+            return (rc, [])
+
+        request = getActive(combo)
+        newSize = dxml.get_widget("shrinkSB").get_value_as_int()
+        actions = []
+
+        try:
+            actions.append(ActionResizeFormat(request, newSize))
+        except ValueError as e:
+            intf.messageWindow(_("Resize FileSystem Error"),
+                               _("%(device)s: %(msg)s")
+                                 % {'device': request.format.device,
+                                    'msg': e.message},
+                               type="warning", custom_icon="error")
+            continue
+
+        try:
+            actions.append(ActionResizeDevice(request, newSize))
+        except ValueError as e:
+            intf.messageWindow(_("Resize Device Error"),
+                               _("%(name)s: %(msg)s")
+                                 % {'name': request.name, 'msg': e.message},
+                               type="warning", custom_icon="error")
+            continue
+
+        runResize = False
+
+    dialog.destroy()
+    return (rc, actions)
+
+class PartitionTypeWindow(InstallWindow):
+    def __init__(self, ics):
+        InstallWindow.__init__(self, ics)
+        ics.setTitle("Automatic Partitioning")
+        ics.setNextEnabled(True)
+
+    def getNext(self):
+        if self.storage.checkNoDisks():
+            raise gui.StayOnScreen
+
+        if self.buttonGroup.getCurrent() == "custom":
+            self.dispatch.skipStep("autopartitionexecute", skip = 1)
+            self.dispatch.skipStep("partition", skip = 0)
+            self.dispatch.skipStep("bootloader", skip = 0)
+
+            self.storage.clearPartType = CLEARPART_TYPE_NONE
+        else:
+            if self.buttonGroup.getCurrent() == "shrink":
+                (rc, actions) = whichToShrink(self.storage, self.intf)
+                if rc == gtk.RESPONSE_OK:
+                    for action in actions:
+                        self.storage.devicetree.registerAction(action)
+                else:
+                    raise gui.StayOnScreen
+
+                # we're not going to delete any partitions in the resize case
+                self.storage.clearPartType = CLEARPART_TYPE_NONE
+            elif self.buttonGroup.getCurrent() == "all":
+                self.storage.clearPartType = CLEARPART_TYPE_ALL
+            elif self.buttonGroup.getCurrent() == "replace":
+                self.storage.clearPartType = CLEARPART_TYPE_LINUX
+            elif self.buttonGroup.getCurrent() == "freespace":
+                self.storage.clearPartType = CLEARPART_TYPE_NONE
+
+            self.dispatch.skipStep("autopartitionexecute", skip = 0)
+
+            if self.encryptButton.get_active():
+                self.storage.encryptedAutoPart = True
+            else:
+                self.storage.encryptionPassphrase = ""
+                self.storage.retrofitPassphrase = False
+                self.storage.encryptedAutoPart = False
+
+            self.storage.doAutoPart = True
+
+            if self.reviewButton.get_active():
+                self.dispatch.skipStep("partition", skip = 0)
+                self.dispatch.skipStep("bootloader", skip = 0)
+            else:
+                self.dispatch.skipStep("partition")
+                self.dispatch.skipStep("bootloader")
+                self.dispatch.skipStep("bootloaderadvanced")
+
+        return None
+
+    def typeChanged(self, *args):
+        if self.buttonGroup.getCurrent() == "custom":
+            if not self.prevrev:
+                self.prevrev = self.reviewButton.get_active()
+
+            self.reviewButton.set_active(True)
+            self.reviewButton.set_sensitive(False)
+            self.encryptButton.set_sensitive(False)
+        else:
+            if self.prevrev:
+                self.reviewButton.set_active(self.prevrev)
+                self.prevrev = None
+
+            self.reviewButton.set_sensitive(True)
+            self.encryptButton.set_sensitive(True)
+
+    def getScreen(self, anaconda):
+        self.anaconda = anaconda
+        self.storage = anaconda.storage
+        self.intf = anaconda.intf
+        self.dispatch = anaconda.dispatch
+
+        (self.xml, vbox) = gui.getGladeWidget("autopart.glade", "parttypeTable")
+        self.encryptButton = self.xml.get_widget("encryptButton")
+        self.reviewButton = self.xml.get_widget("reviewButton")
+        self.table = self.xml.get_widget("parttypeTable")
+
+        self.prevrev = None
+        self.reviewButton.set_active(not self.dispatch.stepInSkipList("partition"))
+        self.encryptButton.set_active(self.storage.encryptedAutoPart)
+
+        self.buttonGroup = pixmapRadioButtonGroup()
+        self.buttonGroup.addEntry("all", _("Use All Space"),
+                                  pixmap=gui.readImageFromFile("partscheme-all.png"),
+                                  descr=_("Removes all partitions on the selected "
+                                          "device(s).  This includes partitions "
+                                          "created by other operating systems.\n\n"
+                                          "<b>Tip:</b> This option will remove "
+                                          "data from the selected device(s).  Make "
+                                          "sure you have backups."))
+        self.buttonGroup.addEntry("replace", _("Replace Existing Linux System(s)"),
+                                  pixmap=gui.readImageFromFile("partscheme-replace.png"),
+                                  descr=_("Removes all Linux partitions on the "
+                                          "selected device(s). This does "
+                                          "not remove other partitions you may have "
+                                          "on your storage device(s) (such as VFAT or "
+                                          "FAT32).\n\n"
+                                          "<b>Tip:</b> This option will remove "
+                                          "data from the selected device(s).  Make "
+                                          "sure you have backups."))
+        self.buttonGroup.addEntry("shrink", _("Shrink Current System"),
+                                  pixmap=gui.readImageFromFile("partscheme-shrink.png"),
+                                  descr=_("Shrinks existing partitions to create free "
+                                          "space for the default layout."))
+        self.buttonGroup.addEntry("freespace", _("Use Free Space"),
+                                  pixmap=gui.readImageFromFile("partscheme-freespace.png"),
+                                  descr=_("Retains your current data and partitions and "
+                                          "uses only the unpartitioned space on the "
+                                          "selected device(s), assuming you have enough "
+                                          "free space available."))
+        self.buttonGroup.addEntry("custom", _("Create Custom Layout"),
+                                  pixmap=gui.readImageFromFile("partscheme-custom.png"),
+                                  descr=_("Manually create your own custom layout on "
+                                          "the selected device(s) using our partitioning "
+                                          "tool."))
+
+        self.buttonGroup.setToggleCallback(self.typeChanged)
+
+        widget = self.buttonGroup.render()
+        self.table.attach(widget, 0, 1, 1, 2)
+
+        # if not set in ks, use UI default
+        if self.storage.clearPartType is None or self.storage.clearPartType == CLEARPART_TYPE_LINUX:
+            self.buttonGroup.setCurrent("replace")
+        elif self.storage.clearPartType == CLEARPART_TYPE_NONE:
+            self.buttonGroup.setCurrent("freespace")
+        elif self.storage.clearPartType == CLEARPART_TYPE_ALL:
+            self.buttonGroup.setCurrent("all")
+
+        if self.buttonGroup.getCurrent() == "custom":
+            # make sure reviewButton is active and not sensitive
+            if self.prevrev == None:
+                self.prevrev = self.reviewButton.get_active()
+
+            self.reviewButton.set_active(True)
+            self.reviewButton.set_sensitive(False)
+            self.encryptButton.set_sensitive(False)
+
+        return vbox