Changeset 023e79b for fedd/compose.py

Timestamp:

May 17, 2010 1:23:22 PM (14 years ago)

Author:

Ted Faber <faber@…>

Branches:

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

Children:

139e2e2

Parents:

4b6909e

Message:

Add testbed attributes

File:

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

fedd/compose.py

@@ -49 +49 @@
         self.add_option('--format', dest='format', default="xml",
                 help='Output file format')
+        self.add_option('--add_testbeds', dest='add_testbeds', default=False,
+                action='store_true',
+                help='add testbed attributes to each component')
         self.add_option('--lax', dest='lax', default=False,
                 action='store_true',
@@ -567 +570 @@
             provides, accepts)
 
+    # Add testbed attributes if requested
+    if opts.add_testbeds:
+        for i, t in enumerate(topos):
+            for e in [ e for e in t.elements if isinstance(e, topdl.Computer)]:
+                e.set_attribute('testbed', 'testbed%03d' % i)
+
     # Make a topology containing all elements and substrates from components
     # that had matches.
@@ -582 +591 @@
 elif topos == 1:
     comp = topos[0]
+    if opts.add_testbeds:
+        for e in [ e for e in comp.elements if isinstance(e, topdl.Computer)]:
+            e.set_attribute('testbed', 'testbed001')
+
 else:
     sys.exit("Did not read any components.")