Utility for getting a slice's vnet (veeeery useful)

[nepi.git] / src / nepi / testbeds / planetlab / util.py

diff --git a/src/nepi/testbeds/planetlab/util.py b/src/nepi/testbeds/planetlab/util.py
index 1fd2880..41b32aa 100644 (file)
--- a/src/nepi/testbeds/planetlab/util.py
+++ b/src/nepi/testbeds/planetlab/util.py
@@ -27,9 +27,9 @@ import logging
 import metadata
 import weakref
 
-def getAPI(user, pass_):
+def getAPI(user, pass_, **kw):
     import plcapi
-    return plcapi.PLCAPI(username=user, password=pass_)
+    return plcapi.PLCAPI(username=user, password=pass_, **kw)
 
 def filterBlacklist(candidates):
     blpath = environ.homepath('plblacklist')
@@ -62,6 +62,16 @@ def appendBlacklist(node_ids):
     finally:
         bl.close()
 
+def getVnet(api, slicename):
+    slicetags = api.GetSliceTags(
+        name = slicename,
+        tagname = 'vsys_vnet',
+        fields=('value',))
+    if slicetags:
+        return slicetags[0]['value']
+    else:
+        return None
+
 def getNodes(api, num, **constraints):
     # Now do the backtracking search for a suitable solution
     # First with existing slice nodes