From 4c97c19f37aa22fb90a7540ebaf73e5d15f52b3f Mon Sep 17 00:00:00 2001
From: Chris Lumens <clumens@redhat.com>
Date: Fri, 26 Oct 2007 15:03:54 +0000
Subject: Added a --disc-size= option to set the size of the media in MB. 
 Defaults to the old value of 640 (#149234)

---
 scripts/splittree.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'scripts')

diff --git a/scripts/splittree.py b/scripts/splittree.py
index 5ac78cd80..4546c9ad4 100644
--- a/scripts/splittree.py
+++ b/scripts/splittree.py
@@ -79,10 +79,11 @@ and RPMs. Set to 1 to turn on.
 self.reserve_size : Additional size needed to be reserved on the first disc.
 """
         
-        self.target_size = 640.0 * 1024.0 * 1024
+        self.reserve_size = 0
+        self.disc_size = 640.0
+        self.target_size = self.disc_size * 1024.0 * 1024
         self.fudge_factor = 1.2 * 1024.0 * 1024
         self.comps_size = 10.0 * 1024 * 1024
-        self.reserve_size = 0
         self.release_str = None
         self.package_order_file = None
         self.arch = None
@@ -388,6 +389,10 @@ self.reserve_size : Additional size needed to be reserved on the first disc.
 
     def main(self):
         """Just runs everything"""
+
+        # Recalculate this here in case the disc_size changed.
+        self.target_size = self.disc_size * 1024.0 * 1024
+
         self.createSplitDirs()
         self.splitRPMS()
         if (self.src_discs != 0):
@@ -407,7 +412,7 @@ if "__main__" == __name__:
     
     timber = Timber()
 
-    theargs = ["arch=", "total-discs=", "bin-discs=",
+    theargs = ["arch=", "total-discs=", "bin-discs=", 'disc-size=',
                "src-discs=", "release-string=", "pkgorderfile=",
                "distdir=", "srcdir=", "productpath=", "reserve-size="]
 
@@ -468,6 +473,9 @@ if "__main__" == __name__:
     if options.has_key("--reserve-size"):
         timber.reserve_size = float(options["--reserve-size"])
 
+    if options.has_key("--disc-size"):
+        timber.disc_size = float(options["--disc-size"])
+
     logfile = timber.main()
     
     for logentry in range(0, len(logfile)):
-- 
cgit