if opt.sched and opt.nonsched: print >> sys.stderr, "--sched and --nonsched can't be used together" sys.exit(1) def namesort(resource): """Helper function to sort by resource name""" return resource["name"] # Adding/Removing are single operations (can't set attributes at same time) if opt.add: if len(args) == 0: print >> sys.stderr, "Must specify at least one resource name" sys.exit(1) specs = [{"name":name} for name in args] added = system.add_resources(specs) added.sort(key=namesort) if added == "KeyError": print >> sys.stderr, "One or more of the given names already exist" sys.exit(1) for r in added: print "Added resource with name '%s'" % r["name"] sys.exit(0) elif opt.remove: specs = [{"name":name} for name in args] removed = system.remove_resources(specs) removed.sort(key=namesort) for r in removed: print "Removed resource with name '%s'" % r["name"] sys.exit(0)
if opt.sched and opt.nonsched: print >> sys.stderr, "--sched and --nonsched can't be used together" sys.exit(1) def namesort(resource): """Helper function to sort by resource name""" return resource["name"] # Adding/Removing are single operations (can't set attributes at same time) if opt.add: if len(args) == 0: print >> sys.stderr, "Must specify at least one resource name" sys.exit(1) specs = [{"name": name} for name in args] added = system.add_resources(specs) added.sort(key=namesort) if added == "KeyError": print >> sys.stderr, "One or more of the given names already exist" sys.exit(1) for r in added: print "Added resource with name '%s'" % r["name"] sys.exit(0) elif opt.remove: specs = [{"name": name} for name in args] removed = system.remove_resources(specs) removed.sort(key=namesort) for r in removed: print "Removed resource with name '%s'" % r["name"] sys.exit(0)