forked from srbala/fedora-kickstarts
101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
|
#!/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()
|
||
|
|