Added option in demo to disable waypoint redirection (and fixed typos in subnet.py)

[affinity.git] / scripts / affinity_control.py

diff --git a/scripts/affinity_control.py b/scripts/affinity_control.py
index b3a3d5ba4ed8c922dd2b4c426302462fdf4c5ae3..540fea9d30fc048658c8a21186f1e09318b41cae 100644 (file)
--- a/scripts/affinity_control.py
+++ b/scripts/affinity_control.py
@@ -56,6 +56,13 @@ class AffinityControl:
         resp, content = self.http.request(self.url_prefix + "link/%s/enable" % link_name, "PUT")
         if (resp.status != 201):
             print "Waypoint could not be enabled for link %s" % link_name
-            print resp.status
             return
         print "Waypoint enabled for link %s" % link_name
+
+    # Disable waypoint
+    def disable_waypoint(self, link_name):
+        resp, content = self.http.request(self.url_prefix + "link/%s/disable" % link_name, "PUT")
+        if (resp.status != 201):
+            print "Waypoint could not be disabled for link %s" % link_name
+            return
+        print "Waypoint disabled for link %s" % link_name