Changeset 9e38ded for fedd

Timestamp:

Jun 3, 2010 12:40:38 AM (14 years ago)

Author:

Ted Faber <faber@…>

Branches:

axis_example, compt_changes, info-ops, master, version-3.01, version-3.02

Children:

05e8da8

Parents:

2c51061

Message:

Parse options from a file.

File:

• fedd/compose.py (modified) (3 diffs)

fedd/compose.py

@@ -13 +13 @@
 from federation.service_error import service_error
 from federation import topdl
+
+class config_error(RuntimeError): pass
 
 class constraint:
@@ -180 +182 @@
                 help='Allow multiple links between the same nodes " + \
                         "to be created.')
+        self.add_option('--config', dest='config', default=None,
+                help='Configuration file of options')
 
     @staticmethod
@@ -624 +628 @@
     return (components, topos, names, constraints, provides, accepts)
 
+def parse_config(fn, opts):
+    """
+    Pull configuration options from a file given in fn.  These mirror the
+    command line options.
+    """
+    # These functions make up a jump table for parsing lines of config.
+    # They're dispatched from directives below.
+    def file(args, opts):
+        i = args.find(',')
+        if i == -1: opts.files.append((args, 1))
+        else: 
+            try:
+                opts.files.append((args[0:i], int(args[i+1:])))
+            except ValueError:
+                raise config_error("Cannot convert %s to an integer" % \
+                        args[i+1:])
+
+    def url(args, opts):
+        opts.url=args
+
+    def output(args, opts):
+        opts.outfile=args
+
+    def format(args, opts):
+        valid = ("xml", "topdl", "tcl", "ns")
+        if args in valid:
+            opts.format = args
+        else: 
+            raise config_error("Bad format %s. Must be one of %s" % \
+                    (args, valid))
+
+    def tb_out(args, opts):
+        opts.output_testbeds = True
+
+    def tb_add(args, opts):
+        opts.add_testbeds = True
+
+    def seed(args, opts):
+        try:
+            opts.seed = int(args)
+        except ValueError:
+            raise config_error("Cannot convert %s to an integer" % args)
+
+    def lax(args, opts):
+        opts.lax = True
+
+    def same_node(args, opts):
+        opts.same_node = True
+
+    def same_topo(args, opts):
+        opts.same_topo = True
+
+    def same_pair(args, opts):
+        opts.multi_pair = True
+
+    directives = {
+            'file': file,
+            'multifile': file,
+            'output': output,
+            'url': url,
+            'format': format,
+            'output_testbeds': tb_out,
+            'add_testbeds': tb_add,
+            'seed': seed,
+            'lax': lax,
+            'same_node': same_node,
+            'same_topo': same_topo,
+            'same_pair': same_pair,
+            }
+    comment_re = re.compile('^\s*#')
+    blank_re = re.compile('^\s*$')
+
+    # Parse_config code begins
+    f = open(fn, "r")
+    try:
+        for i, l in enumerate(f):
+            if blank_re.match(l) or comment_re.match(l):
+                continue
+
+            if l.find('=') != -1: dir, args = l.split('=', 1)
+            else: dir, args = (l, "")
+
+            dir, args = (dir.strip(), args.strip())
+
+            try:
+                if dir in directives: directives[dir](args, opts)
+                else: file(dir, opts)
+            except config_error, e:
+                raise config_error("%s in line %d" % (e, i+1))
+    finally:
+        if f: f.close()
+
+
 # Main line begins
 parser = ComposeOptionParser()
 
 opts, args = parser.parse_args()
+
+if opts.config:
+    try:
+        parse_config(opts.config, opts)
+    except EnvironmentError, e:
+        sys.exit("Can't open configuration: %s" %e)
+    except config_error, e:
+        sys.exit(e)
 
 if opts.cert: