From c057593be6d9c926ff83bcb21f880f6e6067e5b5 Mon Sep 17 00:00:00 2001 From: "jkeating@localhost.localdomain" <> Date: Tue, 10 Oct 2006 21:05:28 -0400 Subject: [PATCH] Move things into main that are for testing. --- gather/gather.py | 44 ++++++++++++++++---------------------------- 1 file changed, 16 insertions(+), 28 deletions(-) diff --git a/gather/gather.py b/gather/gather.py index 19b529c3..d296a039 100755 --- a/gather/gather.py +++ b/gather/gather.py @@ -17,32 +17,6 @@ import sys import yum from optparse import OptionParser - -def get_arguments(): -# hack job for now, I'm sure this could be better for our uses - usage = "usage: %s [options]" % sys.argv[0] - parser = OptionParser(usage=usage) - parser.add_option("--destdir", default=".", dest="destdir", - help='destination directory (defaults to current directory)') - parser.add_option("--comps", default="comps.xml", dest="comps", - help='comps file to use') - parser.add_option("--yumconf", default="yum.conf", dest="yumconf", - help='yum config file to use') - - - (opts, args) = parser.parse_args() - #if len(opts) < 1: - # parser.print_help() - # sys.exit(0) - return (opts, args) - -def get_packagelist(myComps): -# Get the list of packages from the comps file - pkglist = [] - for group in myComps.groups: - pkglist += group.packages - return pkglist - def create_yumobj(yumconf): # Create a yum object to act upon myYum = yum.yumBase() @@ -51,12 +25,26 @@ def create_yumobj(yumconf): return myYum def main(): - (opts, args) = get_arguments() +# This is used for testing the module + usage = "usage: %s [options]" % sys.argv[0] + parser = OptionParser(usage=usage) + parser.add_option("--destdir", default=".", dest="destdir", + help='destination directory (defaults to current directory)') + parser.add_option("--comps", default="comps.xml", dest="comps", + help='comps file to use') + parser.add_option("--yumconf", default="yum.conf", dest="yumconf", + help='yum config file to use') + (opts, args) = parser.parse_args() + + try: compsobj = yum.comps.Comps() compsobj.add(opts.comps) - print get_packagelist(compsobj) + pkglist = [] + for group in compsobj.groups: + pkglist += group.packages + print pkglist except IOError: print >> sys.stderr, "gather.py: No such file:\'%s\'" % opts.comps