handle kernel-enterprise if it's selected

1 files changed, 13 insertions, 5 deletions

diff --git a/lilo.py b/lilo.py
index 524d5bc13..a88d0bfc2 100644
--- a/lilo.py
+++ b/lilo.py
@@ -298,14 +298,22 @@ class LiloConfiguration:
             elif label:
                 otherList.append (label, "/dev/" + drive)
 
-        lilo.addEntry("default", self.default)        
-
-	label = main
+	upKernelLabel = main
+	defaultKernel = main
+
+	if (hdList.has_key('kernel-enterprise') and 
+                        hdList['kernel-enterprise'].selected):
+	    kernelList.append(('linux-ent', 
+			      hdList['kernel-enterprise'], "enterprise"))
+	    upKernelLabel = main + "-up"
+	    defaultKernel = 'linux-ent'
 	if (smpInstalled):
 	    kernelList.append((main, hdList['kernel-smp'], "smp"))
-	    label = main + "-up"
+	    upKernelLabel = main + "-up"
+
+	kernelList.append((upKernelLabel, hdList['kernel'], ""))
 
-	kernelList.append((label, hdList['kernel'], ""))
+        lilo.addEntry("default", defaultKernel)
 
 	for (label, kernel, tag) in kernelList:
 	    kernelTag = "-%s-%s%s" % (kernel[rpm.RPMTAG_VERSION],