adding new client cli tools

[sfa.git] / sfa / client / sfiAddAttribute.py

diff --git a/sfa/client/sfiAddAttribute.py b/sfa/client/sfiAddAttribute.py
new file mode 100755 (executable)
index 0000000..33b94a3
--- /dev/null
+++ b/sfa/client/sfiAddAttribute.py
@@ -0,0 +1,36 @@
+#! /usr/bin/env python
+
+import sys
+from sfa.util.rspecHelper import RSpec, Commands
+
+command = Commands(usage="%prog [options] [node1 node2...]",
+                   description="Add sliver attributes to the RSpec. " +
+                   "This command reads in an RSpec and outputs a modified " +
+                   "RSpec. Use this to add attributes to individual nodes " +
+                   "in your slice.  If no nodes are specified, the " +
+                   "attributes will be added to ALL nodes.",
+                   epilog="NOTE: Only admins can actually set these " +
+                   "attributes, with the exception of --delegations")
+command.add_nodefile_option()
+command.add_attribute_options()
+command.prep()
+
+attrs = command.get_attribute_dict()
+for name in attrs:
+    for value in attrs[name]:
+        if not command.nodes:
+            try:
+                command.rspec.add_default_sliver_attribute(name, value)
+            except:
+                print >> sys.stderr, "FAILED: on all nodes: %s=%s" % (name, value)
+        else:
+            for node in command.nodes:
+                try:
+                    command.rspec.add_sliver_attribute(node, name, value)
+                except:
+                    print >> sys.stderr, "FAILED: on node %s: %s=%s" % (node, name, value)
+
+print command.rspec
+    
+
+