Take ownership of pkgorder and splittree.py; call appropriately

These two files have been removed from anaconda sources, so we
are taking ownership of them.  We will likely rewrite some of these
but the plan is to keep them as 'utility' type tools so that they
can be used without creating full pungi objects.
This commit is contained in:
Jesse Keating 2008-06-23 10:27:30 -04:00
parent 897f97ee59
commit 15c5b80e87
3 changed files with 708 additions and 3 deletions

219
src/bin/pkgorder Executable file
View File

@ -0,0 +1,219 @@
#!/usr/bin/python
#
# pkgorder
#
# Copyright (C) 2005 Red Hat, Inc. All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# Author(s): Paul Nasrat <pnasrat@redhat.com>
#
import os.path
import glob
import rpm
import rpmUtils
import shutil
import string
import sys
import yum
sys.path.append("/usr/lib/anaconda")
sys.path.append("/usr/lib/booty")
import anaconda_log
import logging
logger = logging.getLogger("anaconda")
handler = logging.StreamHandler()
handler.setLevel(logging.ERROR)
logger.addHandler(handler)
from optparse import OptionParser
import yum
class PackageOrderer(yum.YumBase):
def __init__(self, arch=None):
yum.YumBase.__init__(self)
self.arch = arch
def _transactionDataFactory(self):
return yum.transactioninfo.SortableTransactionData()
def doFileLogSetup(self, uid, logfile):
pass
def doLoggingSetup(self, debuglevel, errorlevel):
pass
def setup(self, fn="/etc/yum.conf", root="/", excludes=[]):
self.doConfigSetup(fn, root, init_plugins = False)
self.conf.cache = 0
# if hasattr(self.repos, 'sqlite'):
# self.repos.sqlite = False
# self.repos._selectSackType()
exclude = self.conf.exclude
exclude.extend(excludes)
self.conf.exclude = exclude
cachedir = yum.misc.getCacheDir()
self.repos.setCacheDir(cachedir)
self.repos.setCache(0)
self.doRepoSetup()
self.doSackSetup(rpmUtils.arch.getArchList(self.arch))
self.doTsSetup()
self.doGroupSetup()
self.repos.populateSack('enabled', 'filelists')
def getDownloadPkgs(self):
pass
#XXX: sigh
processed = {}
def processTransaction(ds):
del ds.ts
ds.initActionTs()
ds.populateTs(keepold=0)
ds.ts.check()
ds.ts.order()
for (hdr, path) in ds.ts.ts.getKeys():
fname = os.path.basename(path)
fpattern = "%s*" % fname.rsplit('.', 2)[0]
printMatchingPkgs(fpattern)
def printMatchingPkgs(fpattern):
global processed
if os.path.isdir("%s/%s/RPMS" % (toppath, product)):
matches = glob.glob("%s/%s/RPMS/%s" % (toppath, product, fpattern))
elif os.path.isdir("%s/%s" %(toppath, product)):
matches = glob.glob("%s/%s/%s" % (toppath, product, fpattern))
else:
matches = glob.glob("%s/%s" % (toppath, fpattern))
for match in matches:
mname = os.path.basename(match)
if processed.has_key(mname): continue
processed[mname] = True
print mname
def addPackages(ds, pkgLst):
ds.initActionTs()
for pkg in pkgLst:
ds.install(pattern=pkg)
ds.resolveDeps()
processTransaction(ds)
def addGroups(ds, groupLst):
ds.initActionTs()
map(ds.selectGroup, filter(lambda x: ds.comps.has_group(x), groupLst))
ds.resolveDeps()
processTransaction(ds)
def createConfig(toppath):
yumconfstr = """
[main]
distroverpkg=redhat-release
gpgcheck=0
reposdir=/dev/null
exclude=*debuginfo*
[anaconda]
name=Anaconda
baseurl=file://%s
enabled=1
""" % (toppath)
try:
(fd, path) = tempfile.mkstemp("", "yum-conf-", toppath)
except (OSError, IOError), e:
print >> sys.stderr, "Error writing to %s" % (toppath,)
sys.exit(1)
os.write(fd, yumconfstr)
os.close(fd)
return path
def usage():
print >> sys.stderr, "pkgorder <toppath> <arch> <productpath>"
print >> sys.stderr, "<arch>: use rpm architecture for tree, eg i686"
if __name__ == "__main__":
import tempfile
parser = OptionParser()
parser.add_option("--debug", action="store_true", dest="debug", default=False)
parser.add_option("--file", action="store", dest="file")
parser.add_option("--product", action="store", dest="productPath", )
parser.add_option("--exclude", action="append", dest="excludeList",
default=[])
(options, args) = parser.parse_args()
if len(args) != 3:
usage()
sys.exit(1)
(toppath, arch, product) = args
config = createConfig(toppath)
# Boo.
if arch == "i386":
arch = "i686"
# print out kernel related packages first
#printMatchingPkgs("kernel-*")
if os.environ.has_key('TMPDIR'):
testpath = "%s/pkgorder-%d" %(os.environ['TMPDIR'],os.getpid(),)
else:
testpath = "/tmp/pkgorder-%d" %(os.getpid(),)
os.system("mkdir -p %s/var/lib/rpm" %(testpath,))
ds = PackageOrderer(arch=arch)
ds.setup(fn=config, excludes=options.excludeList, root = testpath)
# hack, hack, hack... make sure iscsi ends up on disc1 (#208832)
addPackages(ds, ["kernel-*","mkinitrd","mdadm"])
addGroups(ds, ["core", "base", "text-internet"])
addGroups(ds, ["base-x", "dial-up",
"graphical-internet", "editors",
"gnome-desktop", "sound-and-video", "printing",
"fonts", "hardware-support", "admin-tools",
"java", "legacy-fonts"])
addGroups(ds, ["office", "games", "graphics", "authoring-and-publishing"])
addGroups(ds, ["web-server", "ftp-server", "sql-server",
"mysql", "server-cfg", "dns-server",
"smb-server"])
addGroups(ds, ["kde-desktop", "development-tools", "development-libs",
"gnome-software-development", "eclipse",
"x-software-development",
"java-development", "kde-software-development",
"mail-server", "network-server", "legacy-network-server"])
addGroups(ds, ["news-server", "legacy-software-development",
"engineering-and-scientific"])
#Everthing else but kernels
for po in ds.pkgSack.returnPackages():
if po.name.find("kernel") == -1:
member = ds.tsInfo.addInstall(po)
ds.resolveDeps()
processTransaction(ds)
os.unlink(config)
shutil.rmtree(testpath)

View File

@ -23,8 +23,7 @@ import logging
import urlgrabber.progress import urlgrabber.progress
import subprocess import subprocess
import createrepo import createrepo
sys.path.append('/usr/lib/anaconda-runtime') import pypungi.splittree
import splittree
class PungiBase(object): class PungiBase(object):
"""The base Pungi class. Set up config items and logging here""" """The base Pungi class. Set up config items and logging here"""
@ -610,7 +609,7 @@ class Pungi(pypungi.PungiBase):
pkgorderfile = open(os.path.join(self.workdir, 'pkgorder-%s' % self.config.get('default', 'arch')), 'w') pkgorderfile = open(os.path.join(self.workdir, 'pkgorder-%s' % self.config.get('default', 'arch')), 'w')
# setup the command # setup the command
pkgorder = ['/usr/lib/anaconda-runtime/pkgorder'] pkgorder = ['/usr/bin/pkgorder']
#pkgorder.append('TMPDIR=%s' % self.workdir) #pkgorder.append('TMPDIR=%s' % self.workdir)
pkgorder.append(self.topdir) pkgorder.append(self.topdir)
pkgorder.append(self.config.get('default', 'arch')) pkgorder.append(self.config.get('default', 'arch'))

487
src/pypungi/splittree.py Normal file
View File

@ -0,0 +1,487 @@
#!/usr/bin/env python
#
# splittree.py
#
# Copyright (C) 2003, 2004, 2005 Red Hat, Inc. All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import sys
import os
import os.path
import string
import getopt
import time
import types
import rpm
global _ts
_ts = None
# returns n-v-r.a from file filename
def nvra(pkgfile):
global _ts
if _ts is None:
_ts = rpm.TransactionSet()
_ts.setVSFlags(-1)
fd = os.open(pkgfile, os.O_RDONLY)
h = _ts.hdrFromFdno(fd)
os.close(fd)
return "%s-%s-%s.%s.rpm" %(h['name'], h['version'], h['release'],
h['arch'])
class Timber:
"""Split trees like no other"""
def __init__(self):
"""self.release_str : the name and version of the product"
self.package_order_file : the location of the file which has
the package ordering
self.arch : the arch the tree is intended for
self.real_arch : the arch found in the unified tree's
.discinfo file
self.dist_dir : the loaction of the unified tree
self.src_dir : the location of the unified SRPM dir
self.start_time : the timestamp that's in .discinfo files
self.dir_info : The info other than start_time that goes
into the .discinfo files. The info should already exist
after running buildinstall in the unified tree
self.total_discs : total number of discs
self.total_bin_discs : total number of discs with RPMs
self.total_srpm_discs : total number of discs with SRPMs
self.reverse_sort_srpms : sort the srpms in reverse order to
fit. Usually only needed if we share a disc between SRPMs
and RPMs. Set to 1 to turn on.
self.reserve_size : Additional size needed to be reserved on the first disc.
"""
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.release_str = None
self.package_order_file = None
self.arch = None
self.real_arch = None
self.dist_dir = None
self.src_dir = None
self.start_time = None
self.dir_info = None
self.total_discs = None
self.bin_discs = None
self.src_discs = None
self.product_path = "anaconda"
self.bin_list = []
self.src_list = []
self.shared_list = []
self.reverse_sort_srpms=None
self.common_files = ['beta_eula.txt', 'EULA', 'README', 'GPL', 'RPM-GPG-KEY', 'RPM-GPG-KEY-beta', 'RPM-GPG-KEY-fedora']
self.logfile = []
def getSize(self, path, blocksize=None):
"""Gets the size as reported by du -sL"""
if blocksize:
p = os.popen("du -sL --block-size=1 %s" % path, 'r')
thesize = p.read()
p.close()
thesize = long(string.split(thesize)[0])
return thesize
else:
p = os.popen("du -sLh %s" % path, 'r')
thesize = p.read()
p.close()
thesize = string.split(thesize)[0]
return thesize
def reportSizes(self, disc, firstpkg=None, lastpkg=None):
"""appends to self.logfile"""
if firstpkg:
self.logfile.append("First package on disc%d: %s" % (disc, firstpkg))
if lastpkg:
self.logfile.append("Last package on disc%d : %s" % (disc, lastpkg))
discsize = self.getSize("%s-disc%d" % (self.dist_dir, disc))
self.logfile.append("%s-disc%d size: %s" % (self.arch, disc, discsize))
def createDiscInfo(self, discnumber):
"""creates the .discinfo files in the split trees"""
if not os.path.exists("%s/.discinfo" % self.dist_dir):
raise RuntimeError, "CRITICAL ERROR : .discinfo doesn't exist in the unified tree, not splitting"
# if start_time isn't set then we haven't got this info yet
if not self.start_time:
file = open("%s/.discinfo" % (self.dist_dir), 'r')
self.start_time = file.readline()[:-1]
self.release_str = file.readline()[:-1]
self.real_arch = file.readline()[:-1]
if self.real_arch != self.arch:
raise RuntimeError, "CRITICAL ERROR : self.real_arch is not the same as self.arch"
# skip the disc number line from the unified tree
file.readline()
# basedir, packagedir, and pixmapdir
self.dir_info = [file.readline()[:-1], file.readline()[:-1], file.readline()[:-1]]
file.close()
discinfo_file = open("%s-disc%d/.discinfo" % (self.dist_dir, discnumber), 'w')
discinfo_file.write("%s\n" % self.start_time)
discinfo_file.write(self.release_str + '\n')
discinfo_file.write(self.real_arch + '\n')
discinfo_file.write("%s\n" % discnumber)
for i in range(0, len(self.dir_info)):
discinfo_file.write(self.dir_info[i] + '\n')
discinfo_file.close()
def linkFiles(self, src_dir, dest_dir, filelist):
"""Creates hardlinks from files in the unified dir to files in the split dirs. This is not for RPMs or SRPMs"""
for file in filelist:
src = "%s/%s" % (src_dir, file)
dest = "%s/%s" % (dest_dir, file)
try:
os.link(src, dest)
except OSError, (errno, msg):
pass
def createSplitDirs(self):
"""Figures out which discs are for RPMs, which are for SRPMs,
and which are shared. Also creates links for files on disc1
and files which are common across all discs"""
if self.bin_discs > self.total_discs or self.src_discs > self.total_discs:
raise RuntimeError, "CRITICAL ERROR : Number of discs specified exceeds the total number of discs"
# get a list of discs for each type of disc. shared_list will
# be returned for sorting out which discs SRPMS can land on
self.bin_list = range(1, self.bin_discs + 1)
self.src_list = range(self.total_discs - self.src_discs + 1, self.total_discs + 1)
self.shared_list = range(self.total_discs - self.src_discs + 1, self.bin_discs + 1)
for i in range(self.bin_list[0], self.bin_list[-1] + 1):
if i == 1:
p = os.popen('find %s/ -type f -not -name .discinfo -not -name "*\.rpm"' % self.dist_dir, 'r')
filelist = p.read()
p.close()
filelist = string.split(filelist)
p = os.popen('find %s/ -type d -not -name SRPMS' % self.dist_dir, 'r')
dirlist = p.read()
p.close()
dirlist = string.split(dirlist)
dont_create = []
# we need to clean up the dirlist first. We don't want everything yet
for j in range(0, len(dirlist)):
dirlist[j] = string.replace(dirlist[j], self.dist_dir, '')
# now create the dirs for disc1
for j in range(0, len(dirlist)):
os.makedirs("%s-disc%d/%s" % (self.dist_dir, i, dirlist[j]))
for j in range(0, len(filelist)):
filelist[j] = string.replace(filelist[j], self.dist_dir, '')
try:
os.link(os.path.normpath("%s/%s" % (self.dist_dir, filelist[j])),
os.path.normpath("%s-disc%d/%s" % (self.dist_dir, i, filelist[j])))
except OSError, (errno, msg):
pass
else:
os.makedirs("%s-disc%d/%s" % (self.dist_dir, i, self.product_path))
self.linkFiles(self.dist_dir, "%s-disc%d" %(self.dist_dir, i), self.common_files)
self.createDiscInfo(i)
if (self.src_discs != 0):
for i in range(self.src_list[0], self.src_list[-1] + 1):
os.makedirs("%s-disc%d/SRPMS" % (self.dist_dir, i))
self.linkFiles(self.dist_dir,
"%s-disc%d" %(self.dist_dir, i),
self.common_files)
self.createDiscInfo(i)
def splitRPMS(self, reportSize = 1):
"""Creates links in the split dirs for the RPMs"""
packages = {}
pkgdir = "%s" %(self.product_path,)
rpmlist = os.listdir("%s/%s" %(self.dist_dir, pkgdir))
rpmlist.sort()
# create the packages dictionary in this format: n-v-r.a:['n-v-r.arch.rpm']
for filename in rpmlist:
filesize = os.path.getsize("%s/%s/%s" % (self.dist_dir, pkgdir, filename))
try:
pkg_nvr = nvra("%s/%s/%s" %(self.dist_dir, pkgdir, filename))
except rpm.error, e:
continue
if packages.has_key(pkg_nvr):
# append in case we have multiple packages with the
# same n-v-r. Ex: the kernel has multiple n-v-r's for
# different arches
packages[pkg_nvr].append(filename)
else:
packages[pkg_nvr] = [filename]
orderedlist = []
# read the ordered pacakge list into orderedlist
file = open(self.package_order_file, 'r')
for pkg_nvr in file.readlines():
pkg_nvr = string.rstrip(pkg_nvr)
if pkg_nvr[0:8] != "warning:":
orderedlist.append(pkg_nvr)
file.close()
# last package is the last package placed on the disc
firstpackage = ''
lastpackage = ''
# packagenum resets when we change discs. It's used to
# determine the first package in the split tree and that's
# about it
packagenum = 0
disc = self.bin_list[0]
for rpm_nvr in orderedlist:
if not packages.has_key(rpm_nvr):
continue
for file_name in packages[rpm_nvr]:
curused = self.getSize("%s-disc%s" % (self.dist_dir, disc), blocksize=1)
filesize = self.getSize("%s/%s/%s" % (self.dist_dir, pkgdir, file_name), blocksize=1)
newsize = filesize + curused
# compensate for the size of the comps package which has yet to be created
if disc == 1:
maxsize = self.target_size - self.comps_size - self.reserve_size
else:
maxsize = self.target_size
packagenum = packagenum + 1
if packagenum == 1:
firstpackage = file_name
# move to the next disc if true
if newsize > maxsize:
self.reportSizes(disc, firstpkg=firstpackage, lastpkg=lastpackage)
# try it, if we are already on the last disc then complain loudly
try:
nextdisc=self.bin_list.index(disc+1)
disc = self.bin_list[nextdisc]
os.link("%s/%s/%s" % (self.dist_dir, pkgdir, file_name),
"%s-disc%d/%s/%s" % (self.dist_dir, disc, pkgdir, file_name))
packagenum = 1
firstpackage = file_name
except:
# back down to the last RPM disc and complain about the overflow
disc = disc - 1
self.logfile.append("No more discs to put packages, overflowing on disc%d" % disc)
continue
else:
os.link("%s/%s/%s" % (self.dist_dir, pkgdir, file_name),
"%s-disc%d/%s/%s" % (self.dist_dir, disc, pkgdir, file_name))
lastpackage = file_name
if reportSize == 1:
if firstpackage == '':
raise RuntimeError, "CRITICAL ERROR : Packages do not fit in given CD size"
self.reportSizes(disc, firstpkg=firstpackage, lastpkg=lastpackage)
def getLeastUsedTree(self):
"""Returns the least full tree to use for SRPMS"""
sizes = []
for i in range(0, len(self.src_list)):
sizes.append([self.getSize("%s-disc%d" % (self.dist_dir, self.src_list[i]), blocksize=1), self.src_list[i]])
sizes.sort()
return sizes[0]
def splitSRPMS(self):
"""Puts the srpms onto the SRPM split discs. The packages are
ordered by size, and placed one by one on the disc with the
most space available"""
srpm_list = []
srpm_disc_list = self.src_list
# create a list of [[size, srpm]]
for srpm in os.listdir("%s" % self.src_dir):
if not srpm.endswith('.rpm'):
continue
srpm_size = self.getSize("%s/%s" % (self.src_dir, srpm), blocksize=1)
srpm_list.append([srpm_size, srpm])
srpm_list.sort()
srpm_list.reverse()
for i in range(0, len(srpm_list)):
# make sure that the src disc is within the size limits,
# if it isn't, pull it out of the list. If there's only
# one disk make loud noises over the overflow
for disc in self.src_list:
if self.getSize("%s-disc%s" % (self.dist_dir, disc), blocksize=1) > self.target_size:
if len(self.src_list) < 2:
self.logfile.append("Overflowing %s on disc%d" % (srpm_list[i][1], disc))
break
else:
discsize = self.getSize("%s-disc%d" % (self.dist_dir, disc))
self.logfile.append("%s-disc%d size: %s" % (self.arch, disc, discsize))
self.src_list.pop(self.src_list.index(disc))
os.link("%s/%s" % (self.src_dir, srpm_list[i][1]),
"%s-disc%d/SRPMS/%s" % (self.dist_dir, self.getLeastUsedTree()[1], srpm_list[i][1]))
for i in range(0, len(srpm_disc_list)):
self.reportSizes(srpm_disc_list[i])
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):
self.splitSRPMS()
return self.logfile
def usage(theerror):
print theerror
print """Usage: %s --arch=i386 --total-discs=8 --bin-discs=4 --src-discs=4 --release-string="distro name" --pkgorderfile=/tmp/pkgorder.12345 --distdir=/usr/src/someunifiedtree --srcdir=/usr/src/someunifiedtree/SRPMS --productpath=product""" % sys.argv[0]
sys.exit(1)
if "__main__" == __name__:
import getopt
timber = Timber()
theargs = ["arch=", "total-discs=", "bin-discs=", 'disc-size=',
"src-discs=", "release-string=", "pkgorderfile=",
"distdir=", "srcdir=", "productpath=", "reserve-size="]
try:
options, args = getopt.getopt(sys.argv[1:], '', theargs)
except getopt.error, error:
usage(error)
myopts = {}
for i in options:
myopts[i[0]] = i[1]
options = myopts
if options.has_key("--arch"):
timber.arch = options['--arch']
else:
usage("You forgot to specify --arch")
if options.has_key("--total-discs"):
timber.total_discs = int(options['--total-discs'])
else:
usage("You forgot to specify --total-discs")
if options.has_key("--bin-discs"):
timber.bin_discs = int(options['--bin-discs'])
else:
usage("You forgot to specify --bin-discs")
if options.has_key("--src-discs"):
timber.src_discs = int(options['--src-discs'])
else:
usage("You forgot to specify --src-discs")
if options.has_key("--release-string"):
timber.release_str = options["--release-string"]
else:
usage("You forgot to specify --release-string")
if options.has_key("--pkgorderfile"):
timber.package_order_file = options["--pkgorderfile"]
else:
usage("You forgot to specify --pkgorderfile")
if options.has_key("--distdir"):
timber.dist_dir = options["--distdir"]
else:
usage("You forgot to specify --distdir")
if options.has_key("--srcdir"):
timber.src_dir = options["--srcdir"]
else:
usage("You forgot to specify --srcdir")
if options.has_key("--productpath"):
timber.product_path = options["--productpath"]
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)):
print logfile[logentry]
sys.exit(0)