product/installclasses/smeinstallclass.py

from installclass import BaseInstallClass
from rhpl.translate import N_, _
from constants import *
import os
import iutil
from fsset import *

from autopart import getAutopartitionBoot, autoCreatePartitionRequests, autoCreateLVMPartitionRequests
from rhpl.log import log
import string
import partRequests
import partedUtils

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):
    id = "smeserver"
    name = N_("New _SME Server Install")
    pixmap = "smeserver.png"
    description = N_("This option performs a new install of "
                     "SME Server.  All attached hard drives "
                     "will be repartitioned and formated.")
    sortPriority = 1

    parentClass = ( _("Install SME Server"), "smeserver.png" )

    def requiredDisplayMode(self):
        return 't'

    def setSteps(self, dispatch):
        dispatch.setStepList(
                 "language",
                 "keyboard",
                 "findrootparts",
                 "findinstall",
                 "installtype",
                 "partitionobjinit",
                 "autopartitionexecute",
                 "partition",
                 "partitiondone",
                 "bootloadersetup",
                 "languagesupport",
                 "timezone",
                 "readcomps",
                 "selectlangpackages",
                 "checkdeps",
                 "dependencies",
                 "install",
                 "enablefilesystems",
                 "migratefilesystems",
                 "setuptime",
                 "preinstallconfig",
                 "installpackages",
                 "postinstallconfig",
                 "writeconfig",
                 "instbootloader",
                 "dopostaction",
                 "methodcomplete",
                 "copylogs",
                 "writeksconfig",
                 "setfilecon",
                 "complete"
                )

        # 'partition' can be used on the command line to force 
        # verification of partitions.  useful in some cases...
        cmdline = open("/proc/cmdline", "r").read()
        cmdline = cmdline.split()
        if "partition" in cmdline:
            dispatch.skipStep("partition", skip = 0)
        else:
            dispatch.skipStep("partition", skip = 1)

    def setDefaultPartitioning(self, partitions, clear = CLEARPART_TYPE_LINUX, doClear = 1):
        uniqueID = 100
        (swapMin, swapMax) = iutil.swapSuggestion()
        diskset = partedUtils.DiskSet()
        drives = diskset.driveList()
        count = 0

        if "noraid" in cmdline:
            for drive in drives:
                if not isys.driveIsRemovable(drive):
                    filesystem = fileSystemTypeGet("ext3")
                    request = partRequests.PartitionSpec(filesystem, mountpoint="/boot", drive=[drive],
                            size=100, primary=1, format=1)
                    partitions.autoPartitionRequests.append(request)

                    if "nolvm" in cmdline:
                        filesystem = fileSystemTypeGet("ext3")
                        request = partRequests.PartitionSpec(filesystem, mountpoint="/", drive=[drive],
                                size=1300, grow=1, primary=1, format=1)
                        partitions.autoPartitionRequests.append(request)

                        filesystem = fileSystemTypeGet("swap")
                        request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin, 
                                maxSizeMB=swapMax, grow=1, primary=1, format=1)
                        partitions.autoPartitionRequests.append(request)

                    else:
                        filesystem = fileSystemTypeGet("physical volume (LVM)")
                        request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin+1500,
                                grow=1, primary=1, format=1)
                        request.uniqueID = 200
                        partitions.autoPartitionRequests.append(request)

                    count = count + 1
                    break

        else:
            raid1 = []
            raid2 = []
            raid3 = []

            for drive in drives:
                if not isys.driveIsRemovable(drive):
                    filesystem = fileSystemTypeGet("software RAID")
                    request = partRequests.PartitionSpec(filesystem, drive=[drive], size=100, primary=1, format=1)
                    request.uniqueID = uniqueID
                    raid1.append(uniqueID)
                    partitions.autoPartitionRequests.append(request)

                    if "nolvm" in cmdline:
                        filesystem = fileSystemTypeGet("software RAID")
                        request = partRequests.PartitionSpec(filesystem, drive=[drive], size=1500,
                                grow=1, primary=1, format=1)
                        request.uniqueID = uniqueID + 50
                        raid2.append(uniqueID + 50)
                        partitions.autoPartitionRequests.append(request)

                        filesystem = fileSystemTypeGet("software RAID")
                        request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin,
                                maxSizeMB=swapMax, grow=1, primary=1, format=1)
                        request.uniqueID = uniqueID + 51
                        raid3.append(uniqueID + 51)
                        partitions.autoPartitionRequests.append(request)

                    else:
                        filesystem = fileSystemTypeGet("software RAID")
                        request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin+1500,
                                grow=1, format=1)
                        request.uniqueID = uniqueID + 50
                        raid2.append(uniqueID + 50)
                        partitions.autoPartitionRequests.append(request)

                    uniqueID = uniqueID + 1

                    count = count + 1
                    if "raid1" in cmdline and len(raid2) > 1:
                        break

            filesystem = fileSystemTypeGet("ext3")
            request = partRequests.RaidRequestSpec(filesystem, mountpoint="/boot", raidminor=1,
                    raidmembers=raid1, raidlevel="RAID1", format=1, raidspares=0)
            partitions.autoPartitionRequests.append(request)

            if len(raid2) > 5:
                raidLevel = "RAID6"
            elif len(raid2) > 2:
                raidLevel = "RAID5"
            else:
                raidLevel = "RAID1"

            if "nolvm" in cmdline:
                filesystem = fileSystemTypeGet("ext3")
                request = partRequests.RaidRequestSpec(filesystem, mountpoint="/", raidmembers=raid2,  raidminor=2,
                        raidlevel=raidLevel, format=1, raidspares=0)
                partitions.autoPartitionRequests.append(request)

                filesystem = fileSystemTypeGet("swap")
                request = partRequests.RaidRequestSpec(filesystem, raidmembers=raid3,  raidminor=3,
                        raidlevel=raidLevel, format=1, raidspares=0)
                partitions.autoPartitionRequests.append(request)

            else:
                filesystem = fileSystemTypeGet("physical volume (LVM)")
                request = partRequests.RaidRequestSpec(filesystem, raidmembers=raid2,  raidminor=2,
                        raidlevel=raidLevel, format=1, raidspares=0)
                request.uniqueID = 200
                partitions.autoPartitionRequests.append(request)

        if count > 0:
            if not "nolvm" in cmdline:
                request = partRequests.VolumeGroupRequestSpec(vgname="main", physvols=[200], 
                        pesize=32768, format=1)
                request.uniqueID = 201
                partitions.autoPartitionRequests.append(request)

                filesystem = fileSystemTypeGet("ext3")
                request = partRequests.LogicalVolumeRequestSpec(filesystem, mountpoint="/", size=1300, 
                        volgroup=201, lvname="root", grow=1, format=1)
                partitions.autoPartitionRequests.append(request)

                filesystem = fileSystemTypeGet("swap")
                request = partRequests.LogicalVolumeRequestSpec(filesystem, size=swapMin, maxSizeMB=swapMax, 
                        volgroup=201, lvname="swap", grow=1, format=1)
                partitions.autoPartitionRequests.append(request)
        else:
            dispatch.skipStep("partition", skip = 0)
            BaseInstallClass.setDefaultPartitioning(self, partitions, clear, doClear)

    def setAsHeadless(self, dispatch, isHeadless = 0):
        if isHeadless == 0:
            pass
        else:
#           dispatch.skipStep("keyboard", permanent = 1)
#           dispatch.skipStep("mouse", permanent = 1)
            dispatch.skipStep("handleX11pkgs", permanent = 1)
            dispatch.skipStep("videocard", permanent = 1)
            dispatch.skipStep("monitor", permanent = 1)
            dispatch.skipStep("xcustom", permanent = 1)
            dispatch.skipStep("writexconfig", permanent = 1)

    def setGroupSelection(self, grpset, intf):
        grpset.unselectAll()
        grpset.selectGroup("Base")

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

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

    def setInstallData(self, id, intf = None):
        BaseInstallClass.setInstallData(self, id)
        self.setAuthentication(id, useShadow=1, useMd5=1)
        self.setRootPassword(id, pw="ThisIsGoingToBeDisabledAnyway", isCrypted=0)
        self.setZeroMbr(id, zeroMbr=1)
        self.setClearParts(id, clear=CLEARPART_TYPE_ALL, initAll=1)
        self.setDefaultPartitioning(id.partitions, doClear=0)
        self.setBootloader(id, useLilo=0, location="mbr", linear=1)

    def __init__(self, expert):
        BaseInstallClass.__init__(self, expert)
1	from installclass import BaseInstallClass
2	from rhpl.translate import N_, _
3	from constants import *
4	import os
5	import iutil
6	from fsset import *
7
8	from autopart import getAutopartitionBoot, autoCreatePartitionRequests, autoCreateLVMPartitionRequests
9	from rhpl.log import log
10	import string
11	import partRequests
12	import partedUtils
13
14	class Script:
15	def __repr__(self):
16	str = ("(s: '%s' i: %s c: %d)") % \
17	(self.script, self.interp, self.inChroot)
18	return string.replace(str, "\n", "\|")
19
20	def __init__(self, script, interp, inChroot, logfile = None):
21	self.script = script
22	self.interp = interp
23	self.inChroot = inChroot
24	self.logfile = logfile
25
26	def run(self, chroot, serial):
27	scriptRoot = "/"
28	if self.inChroot:
29	scriptRoot = chroot
30
31	path = scriptRoot + "/tmp/sme-script"
32
33	f = open(path, "w")
34	f.write(self.script)
35	f.close()
36	os.chmod(path, 0700)
37
38	if self.logfile is not None:
39	messages = self.logfile
40	elif serial:
41	messages = "/tmp/sme-script.log"
42	else:
43	messages = "/dev/tty3"
44
45	rc = iutil.execWithRedirect(self.interp,
46	[self.interp,"/tmp/sme-script"],
47	stdout = messages, stderr = messages,
48	root = scriptRoot)
49
50	if rc != 0:
51	log("WARNING - Error code %s encountered running a sme script", rc)
52
53	os.unlink(path)
54
55	class InstallClass(BaseInstallClass):
56	id = "smeserver"
57	name = N_("New _SME Server Install")
58	pixmap = "smeserver.png"
59	description = N_("This option performs a new install of "
60	"SME Server. All attached hard drives "
61	"will be repartitioned and formated.")
62	sortPriority = 1
63
64	parentClass = ( _("Install SME Server"), "smeserver.png" )
65
66	def requiredDisplayMode(self):
67	return 't'
68
69	def setSteps(self, dispatch):
70	dispatch.setStepList(
71	"language",
72	"keyboard",
73	"findrootparts",
74	"findinstall",
75	"installtype",
76	"partitionobjinit",
77	"autopartitionexecute",
78	"partition",
79	"partitiondone",
80	"bootloadersetup",
81	"languagesupport",
82	"timezone",
83	"readcomps",
84	"selectlangpackages",
85	"checkdeps",
86	"dependencies",
87	"install",
88	"enablefilesystems",
89	"migratefilesystems",
90	"setuptime",
91	"preinstallconfig",
92	"installpackages",
93	"postinstallconfig",
94	"writeconfig",
95	"instbootloader",
96	"dopostaction",
97	"methodcomplete",
98	"copylogs",
99	"writeksconfig",
100	"setfilecon",
101	"complete"
102	)
103
104	# 'partition' can be used on the command line to force
105	# verification of partitions. useful in some cases...
106	cmdline = open("/proc/cmdline", "r").read()
107	cmdline = cmdline.split()
108	if "partition" in cmdline:
109	dispatch.skipStep("partition", skip = 0)
110	else:
111	dispatch.skipStep("partition", skip = 1)
112
113	def setDefaultPartitioning(self, partitions, clear = CLEARPART_TYPE_LINUX, doClear = 1):
114	uniqueID = 100
115	(swapMin, swapMax) = iutil.swapSuggestion()
116	diskset = partedUtils.DiskSet()
117	drives = diskset.driveList()
118	count = 0
119
120	if "noraid" in cmdline:
121	for drive in drives:
122	if not isys.driveIsRemovable(drive):
123	filesystem = fileSystemTypeGet("ext3")
124	request = partRequests.PartitionSpec(filesystem, mountpoint="/boot", drive=[drive],
125	size=100, primary=1, format=1)
126	partitions.autoPartitionRequests.append(request)
127
128	if "nolvm" in cmdline:
129	filesystem = fileSystemTypeGet("ext3")
130	request = partRequests.PartitionSpec(filesystem, mountpoint="/", drive=[drive],
131	size=1300, grow=1, primary=1, format=1)
132	partitions.autoPartitionRequests.append(request)
133
134	filesystem = fileSystemTypeGet("swap")
135	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin,
136	maxSizeMB=swapMax, grow=1, primary=1, format=1)
137	partitions.autoPartitionRequests.append(request)
138
139	else:
140	filesystem = fileSystemTypeGet("physical volume (LVM)")
141	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin+1500,
142	grow=1, primary=1, format=1)
143	request.uniqueID = 200
144	partitions.autoPartitionRequests.append(request)
145
146	count = count + 1
147	break
148
149	else:
150	raid1 = []
151	raid2 = []
152	raid3 = []
153
154	for drive in drives:
155	if not isys.driveIsRemovable(drive):
156	filesystem = fileSystemTypeGet("software RAID")
157	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=100, primary=1, format=1)
158	request.uniqueID = uniqueID
159	raid1.append(uniqueID)
160	partitions.autoPartitionRequests.append(request)
161
162	if "nolvm" in cmdline:
163	filesystem = fileSystemTypeGet("software RAID")
164	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=1500,
165	grow=1, primary=1, format=1)
166	request.uniqueID = uniqueID + 50
167	raid2.append(uniqueID + 50)
168	partitions.autoPartitionRequests.append(request)
169
170	filesystem = fileSystemTypeGet("software RAID")
171	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin,
172	maxSizeMB=swapMax, grow=1, primary=1, format=1)
173	request.uniqueID = uniqueID + 51
174	raid3.append(uniqueID + 51)
175	partitions.autoPartitionRequests.append(request)
176
177	else:
178	filesystem = fileSystemTypeGet("software RAID")
179	request = partRequests.PartitionSpec(filesystem, drive=[drive], size=swapMin+1500,
180	grow=1, format=1)
181	request.uniqueID = uniqueID + 50
182	raid2.append(uniqueID + 50)
183	partitions.autoPartitionRequests.append(request)
184
185	uniqueID = uniqueID + 1
186
187	count = count + 1
188	if "raid1" in cmdline and len(raid2) > 1:
189	break
190
191	filesystem = fileSystemTypeGet("ext3")
192	request = partRequests.RaidRequestSpec(filesystem, mountpoint="/boot", raidminor=1,
193	raidmembers=raid1, raidlevel="RAID1", format=1, raidspares=0)
194	partitions.autoPartitionRequests.append(request)
195
196	if len(raid2) > 5:
197	raidLevel = "RAID6"
198	elif len(raid2) > 2:
199	raidLevel = "RAID5"
200	else:
201	raidLevel = "RAID1"
202
203	if "nolvm" in cmdline:
204	filesystem = fileSystemTypeGet("ext3")
205	request = partRequests.RaidRequestSpec(filesystem, mountpoint="/", raidmembers=raid2, raidminor=2,
206	raidlevel=raidLevel, format=1, raidspares=0)
207	partitions.autoPartitionRequests.append(request)
208
209	filesystem = fileSystemTypeGet("swap")
210	request = partRequests.RaidRequestSpec(filesystem, raidmembers=raid3, raidminor=3,
211	raidlevel=raidLevel, format=1, raidspares=0)
212	partitions.autoPartitionRequests.append(request)
213
214	else:
215	filesystem = fileSystemTypeGet("physical volume (LVM)")
216	request = partRequests.RaidRequestSpec(filesystem, raidmembers=raid2, raidminor=2,
217	raidlevel=raidLevel, format=1, raidspares=0)
218	request.uniqueID = 200
219	partitions.autoPartitionRequests.append(request)
220
221	if count > 0:
222	if not "nolvm" in cmdline:
223	request = partRequests.VolumeGroupRequestSpec(vgname="main", physvols=[200],
224	pesize=32768, format=1)
225	request.uniqueID = 201
226	partitions.autoPartitionRequests.append(request)
227
228	filesystem = fileSystemTypeGet("ext3")
229	request = partRequests.LogicalVolumeRequestSpec(filesystem, mountpoint="/", size=1300,
230	volgroup=201, lvname="root", grow=1, format=1)
231	partitions.autoPartitionRequests.append(request)
232
233	filesystem = fileSystemTypeGet("swap")
234	request = partRequests.LogicalVolumeRequestSpec(filesystem, size=swapMin, maxSizeMB=swapMax,
235	volgroup=201, lvname="swap", grow=1, format=1)
236	partitions.autoPartitionRequests.append(request)
237	else:
238	dispatch.skipStep("partition", skip = 0)
239	BaseInstallClass.setDefaultPartitioning(self, partitions, clear, doClear)
240
241	def setAsHeadless(self, dispatch, isHeadless = 0):
242	if isHeadless == 0:
243	pass
244	else:
245	# dispatch.skipStep("keyboard", permanent = 1)
246	# dispatch.skipStep("mouse", permanent = 1)
247	dispatch.skipStep("handleX11pkgs", permanent = 1)
248	dispatch.skipStep("videocard", permanent = 1)
249	dispatch.skipStep("monitor", permanent = 1)
250	dispatch.skipStep("xcustom", permanent = 1)
251	dispatch.skipStep("writexconfig", permanent = 1)
252
253	def setGroupSelection(self, grpset, intf):
254	grpset.unselectAll()
255	grpset.selectGroup("Base")
256
257	def postAction(self, rootPath, serial, intf):
258	win = intf.waitWindow(_("Post Install Script"),
259	_("The post installation script is running..."))
260
261	script = ( "/sbin/syslogd &\n"
262	"sleep 2\n"
263	"/sbin/e-smith/signal-event post-install\n" )
264	s = Script(script, interp="/bin/sh", inChroot=1)
265	log("%s", s)
266	s.run(rootPath, serial)
267	win.pop()
268
269	def setInstallData(self, id, intf = None):
270	BaseInstallClass.setInstallData(self, id)
271	self.setAuthentication(id, useShadow=1, useMd5=1)
272	self.setRootPassword(id, pw="ThisIsGoingToBeDisabledAnyway", isCrypted=0)
273	self.setZeroMbr(id, zeroMbr=1)
274	self.setClearParts(id, clear=CLEARPART_TYPE_ALL, initAll=1)
275	self.setDefaultPartitioning(id.partitions, doClear=0)
276	self.setBootloader(id, useLilo=0, location="mbr", linear=1)
277
278	def __init__(self, expert):
279	BaseInstallClass.__init__(self, expert)