diff --git a/gather/gather.py b/gather/gather.py index d296a039..310d2ee3 100755 --- a/gather/gather.py +++ b/gather/gather.py @@ -26,30 +26,40 @@ def create_yumobj(yumconf): def main(): # 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() - - + (opts, args) = get_arguments() try: compsobj = yum.comps.Comps() compsobj.add(opts.comps) - pkglist = [] - for group in compsobj.groups: - pkglist += group.packages - print pkglist + print get_packagelist(compsobj) except IOError: print >> sys.stderr, "gather.py: No such file:\'%s\'" % opts.comps sys.exit(1) if __name__ == '__main__': - main() + 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 + main()