Add tools/ directory, with list-packages-for-ks.py

Add a script which prints the explicitly specified package list
for a given comps file and kickstart.

1 files changed, 100 insertions, 0 deletions

diff --git a/tools/list-packages-for-ks.py b/tools/list-packages-for-ks.py
new file mode 100755
index 0000000..0ec0eb2
--- /dev/null
+++ b/tools/list-packages-for-ks.py
@@ -0,0 +1,100 @@
+#!/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()
+