Generate headers on split media methods

4 files changed, 7 insertions, 2 deletions

diff --git a/harddrive.py b/harddrive.py
index 5a5c96672..a25027d31 100644
--- a/harddrive.py
+++ b/harddrive.py
@@ -156,6 +156,7 @@ class HardDriveInstallMethod(InstallMethod):
     def __init__(self, method, rootPath, intf):
         """@param method hd://device:fstype:/path"""
 	InstallMethod.__init__(self, method, rootPath, intf)
+        self.splitmethod = True
         method = method[5:]
 
         device = method[0:method.index(":")]
diff --git a/image.py b/image.py
index ecdd738cb..ca69da728 100644
--- a/image.py
+++ b/image.py
@@ -120,6 +120,7 @@ class ImageInstallMethod(InstallMethod):
     def __init__(self, tree, rootPath, intf):
 	InstallMethod.__init__(self, tree, rootPath, intf)
 	self.tree = tree
+        self.splitmethod = True
         self.currentIso = None
 
 class CdromInstallMethod(ImageInstallMethod):
diff --git a/installmethod.py b/installmethod.py
index e8613e029..ca7489400 100644
--- a/installmethod.py
+++ b/installmethod.py
@@ -84,6 +84,7 @@ class InstallMethod:
         self.rootPath = rootpath
         self.intf = intf
         self.tree = None
+        self.splitmethod = False
 
     def getMethodUri(self):
         pass
diff --git a/yuminstall.py b/yuminstall.py
index 21a9a35de..9e9d029bb 100644
--- a/yuminstall.py
+++ b/yuminstall.py
@@ -311,8 +311,10 @@ class YumBackend(AnacondaBackend):
         self.ac = AnacondaYumConf(self.method.getMethodUri(), 
                                  configfile="/tmp/yum.conf", root=instPath)
         self.ac.write()
-
-        self.ayum = AnacondaYum(fn="/tmp/yum.conf", root=instPath)
+        if self.method.splitmethod:
+            self.ayum = AnacondaYumMedia(fn="/tmp/yum.conf", root=instPath)
+        else:
+            self.ayum = AnacondaYum(fn="/tmp/yum.conf", root=instPath)
         # FIXME: this is a bad hack until we can get something better into yum
         self.anaconda_grouplist = []