a little nicer wrt pep8
[sfa.git] / clientbin / sfiAddSliver.py
1 #!/usr/bin/env python3
2
3 import sys
4
5 from sfa.util.sfalogging import logger
6 from sfa.client.sfi_commands import Commands
7 from sfa.rspecs.rspec import RSpec
8 from sfa.rspecs.version_manager import VersionManager
9
10 logger.enable_console()
11 command = Commands(usage="%prog [options] node1 node2...",
12                    description="Add slivers to the RSpec. " +
13                    "This command reads in an RSpec and outputs a modified " +
14                    "RSpec. Use this to add nodes to your slice.")
15 command.add_nodefile_option()
16 command.prep()
17
18 if not command.opts.nodefile:
19     print("Missing node list -- exiting")
20     command.parser.print_help()
21     sys.exit(1)
22
23 if command.opts.infile:
24     infile = open(command.opts.infile)
25 else:
26     infile = sys.stdin
27 if command.opts.outfile:
28     outfile = open(command.opts.outfile, "w")
29 else:
30     outfile = sys.stdout
31 ad_rspec = RSpec(infile)
32 nodes = open(command.opts.nodefile).read().split()
33 version_manager = VersionManager()
34 try:
35     type = ad_rspec.version.type
36     version_num = ad_rspec.version.version
37     request_version = version_manager._get_version(
38         type, version_num, 'request')
39     request_rspec = RSpec(version=request_version)
40     request_rspec.version.merge(ad_rspec)
41     request_rspec.version.add_slivers(nodes)
42 except:
43     logger.log_exc("sfiAddSliver failed with nodes %s" % nodes)
44     sys.exit(1)
45 print(request_rspec.toxml(), file=outfile)
46 sys.exit(0)