cdrom.image/updates/smeupgradeclass.py

from installclass import BaseInstallClass
from rhpl.translate import N_, _
from rhpl.log import log
from constants import *

import os
import iutil
import string
import security

class Script:
    def __repr__(self):
        str = ("(s: '%s' i: %s c: %d)") %  \
            (self.script, self.interp, self.inChroot)
        return string.replace(str, "\n", "|")

    def __init__(self, script, interp, inChroot, logfile = None):
        self.script = script
        self.interp = interp
        self.inChroot = inChroot
        self.logfile = logfile

    def run(self, chroot, serial):
        scriptRoot = "/"
        if self.inChroot:
            scriptRoot = chroot

        path = scriptRoot + "/tmp/sme-script"

        f = open(path, "w")
        f.write(self.script)
        f.close()
        os.chmod(path, 0700)

        if self.logfile is not None:
            messages = self.logfile
        elif serial:
            messages = "/tmp/sme-script.log"
        else:
            messages = "/dev/tty3"

        rc = iutil.execWithRedirect(self.interp,
                                    [self.interp,"/tmp/sme-script"],
                                    stdout = messages, stderr = messages,
                                    root = scriptRoot)

        if rc != 0:
            log("WARNING - Error code %s encountered running a sme script", rc)

        os.unlink(path)

class InstallClass(BaseInstallClass):
    name = N_("Upgrade Existing System")
    pixmap = "upgrade.png"
    sortPriority = 999999

    parentClass = ( _("Upgrade"), "upgrade.png" )

    def requiredDisplayMode(self):
        return 't'

    def setSteps(self, dispatch):
        dispatch.setStepList(
                    "language",
                    "keyboard",
                    "installtype",
                    "findrootparts",
                    "findinstall",
                    "partitionobjinit",
                    "upgrademount",
                    "upgrademigfind",
                    "upgrademigratefs",
                    "upgradecontinue",
                    "readcomps",
                    "findpackages",
                    "checkdeps",
                    "dependencies",
                    "install",
                    "migratefilesystems",
                    "preinstallconfig",
                    "installpackages",
                    "postinstallconfig",
                    "instbootloader",
                    "dopostaction",
                    "methodcomplete",
                    "copylogs",
                    "complete"
                )

        if iutil.getPPCMachine() == "iSeries":
            dispatch.skipStep("bootloadersetup", skip = 0)

        if iutil.getArch() != "i386" and iutil.getArch() != "x86_64":
            dispatch.skipStep("bootloader")
            dispatch.skipStep("bootloaderadvanced")

        if iutil.getArch() != "i386" and iutil.getArch() != "x86_64":
            dispatch.skipStep("upgbootloader")            

    def postAction(self, rootPath, serial, intf):
        win = intf.waitWindow(_("Post Upgrade Script"),
           _("The post upgrade script is running..."))

        script = ( "/sbin/syslogd &\n"
                   "sleep 2\n"
                   "/sbin/e-smith/signal-event post-upgrade\n" )
        s = Script(script, interp="/bin/sh", inChroot=1)
        log("%s", s)
        s.run(rootPath, serial)
        win.pop()

    def setInstallData(self, id):
        BaseInstallClass.setInstallData(self, id)
        id.upgrade.set(1)
    
    def __init__(self, expert):
        BaseInstallClass.__init__(self, expert)
1	slords	1.1	from installclass import BaseInstallClass
2			from rhpl.translate import N_, _
3			from rhpl.log import log
4	slords	1.2	from constants import *
5	slords	1.1
6			import os
7	slords	1.8	import iutil
8	slords	1.2	import string
9			import security
10
11			class Script:
12			def __repr__(self):
13			str = ("(s: '%s' i: %s c: %d)") % \
14			(self.script, self.interp, self.inChroot)
15			return string.replace(str, "\n", "\|")
16
17			def __init__(self, script, interp, inChroot, logfile = None):
18			self.script = script
19			self.interp = interp
20			self.inChroot = inChroot
21			self.logfile = logfile
22
23			def run(self, chroot, serial):
24			scriptRoot = "/"
25			if self.inChroot:
26			scriptRoot = chroot
27
28			path = scriptRoot + "/tmp/sme-script"
29
30			f = open(path, "w")
31			f.write(self.script)
32			f.close()
33			os.chmod(path, 0700)
34
35			if self.logfile is not None:
36			messages = self.logfile
37			elif serial:
38			messages = "/tmp/sme-script.log"
39			else:
40			messages = "/dev/tty3"
41
42			rc = iutil.execWithRedirect(self.interp,
43			[self.interp,"/tmp/sme-script"],
44			stdout = messages, stderr = messages,
45			root = scriptRoot)
46
47			if rc != 0:
48			log("WARNING - Error code %s encountered running a sme script", rc)
49
50			os.unlink(path)
51	slords	1.1
52			class InstallClass(BaseInstallClass):
53			name = N_("Upgrade Existing System")
54			pixmap = "upgrade.png"
55			sortPriority = 999999
56
57			parentClass = ( _("Upgrade"), "upgrade.png" )
58
59			def requiredDisplayMode(self):
60			return 't'
61
62			def setSteps(self, dispatch):
63			dispatch.setStepList(
64			"language",
65			"keyboard",
66	slords	1.12	"installtype",
67	slords	1.1	"findrootparts",
68	slords	1.2	"findinstall",
69	slords	1.1	"partitionobjinit",
70			"upgrademount",
71			"upgrademigfind",
72			"upgrademigratefs",
73			"upgradecontinue",
74			"readcomps",
75			"findpackages",
76			"checkdeps",
77			"dependencies",
78			"install",
79			"migratefilesystems",
80			"preinstallconfig",
81			"installpackages",
82			"postinstallconfig",
83			"instbootloader",
84			"dopostaction",
85			"methodcomplete",
86			"copylogs",
87			"complete"
88			)
89
90	slords	1.8	if iutil.getPPCMachine() == "iSeries":
91			dispatch.skipStep("bootloadersetup", skip = 0)
92
93			if iutil.getArch() != "i386" and iutil.getArch() != "x86_64":
94			dispatch.skipStep("bootloader")
95			dispatch.skipStep("bootloaderadvanced")
96
97			if iutil.getArch() != "i386" and iutil.getArch() != "x86_64":
98			dispatch.skipStep("upgbootloader")
99
100	gordonr	1.7	def postAction(self, rootPath, serial, intf):
101			win = intf.waitWindow(_("Post Upgrade Script"),
102			_("The post upgrade script is running..."))
103
104	slords	1.3	script = ( "/sbin/syslogd &\n"
105			"sleep 2\n"
106			"/sbin/e-smith/signal-event post-upgrade\n" )
107	slords	1.2	s = Script(script, interp="/bin/sh", inChroot=1)
108			log("%s", s)
109			s.run(rootPath, serial)
110	gordonr	1.7	win.pop()
111	slords	1.1
112			def setInstallData(self, id):
113			BaseInstallClass.setInstallData(self, id)
114			id.upgrade.set(1)
115
116			def __init__(self, expert):
117			BaseInstallClass.__init__(self, expert)