fedora-kickstarts/tools/list-packages-for-ks.py

101 lines
3.1 KiB
Python
Raw Normal View History

#!/usr/bin/python
# Given as input a comps definition and a .ks file, list
# the packages which would be installed. This list is
# NOT depsolved, it's merely a textual list of explicitly
# added or removed packages from the combined set.
#
# Licensed under the new-BSD license (http://www.opensource.org/licenses/bsd-license.php)
# Copyright (C) 2010 Red Hat, Inc.
# Written by Colin Walters <walters@verbum.org>
import os
import sys
import getopt
import xml.etree.cElementTree as ElementTree
import pykickstart
import pykickstart.parser
import pykickstart.version
def usage(ecode):
print "Usage: %s [--version=VERSION] COMPS.xml KICKSTART.ks" % (sys.argv[0], )
print "List packages installed by KICKSTART."
sys.exit(ecode)
def main():
try:
opts,args = getopt.getopt(sys.argv[1:], 'h', ['help', 'version=', 'debug'])
except getopt.GetoptError, e:
usage(1)
version = None
debug = False
for o,a in opts:
if o in ('-h', '--help'):
usage(0)
elif o in ('--debug', ):
debug = True
elif o in ('--version', ):
version = a
if len(args) != 2:
usage(1)
comps_filename = args[0]
kickstart_filename = args[1]
if version:
version_cls = pykickstart.version.stringToVersion(version)
else:
version_cls = pykickstart.version.makeVersion()
comps = ElementTree.parse(comps_filename)
# pykickstart is going to look for files relative to
# the current directory of the kickstart file
working_directory = os.path.dirname(kickstart_filename)
if working_directory:
os.chdir(working_directory)
parser = pykickstart.parser.KickstartParser(version_cls)
parser.readKickstart(kickstart_filename)
comps_groups = comps.findall('group')
packages_for_group = {}
for group in comps_groups:
group_id = group.find('id')
assert group_id is not None
pkglist_node = group.find('packagelist')
assert pkglist_node is not None
reqs = pkglist_node.findall('packagereq')
pkglist = []
for req in reqs:
if req.attrib['type'] in ('default', 'mandatory'):
pkglist.append(req.text)
packages_for_group[group_id.text] = pkglist
pkg_list = set()
for group in parser.handler.packages.groupList:
if debug:
print >>sys.stderr, "# Including %d packages from group %r" % (len(packages_for_group[group.name]), group.name)
for pkg in packages_for_group[group.name]:
pkg_list.add(pkg)
if debug:
print >>sys.stderr, "# Adding %d explicitly specified packages" % (len(parser.handler.packages.packageList), )
for pkg in parser.handler.packages.packageList:
pkg_list.add(pkg)
if debug:
print >>sys.stderr, "# Processing %d explicitly excluded packages" % (len(parser.handler.packages.excludedList), )
for pkg in parser.handler.packages.excludedList:
if pkg in pkg_list:
pkg_list.remove(pkg)
for pkg in sorted(pkg_list):
print "%s" % (pkg, )
sys.exit(0)
if __name__ == '__main__':
main()