Allow lists of nodes to be passed to getNodeByName

....which should perhaps be renamed!!!
2012-04-09 23:54:17 +00:00
parent 149a1f5639
commit 548580d817
1 changed files with 5 additions and 3 deletions
@@ -214,9 +214,11 @@ class Mininet( object ):

    # BL: is this better than just using nameToNode[] ?
    # Should it have a better name?
-    def getNodeByName( self, nodeName ):
-        "Return node with given name"
-        return self.nameToNode[ nodeName ]
+    def getNodeByName( self, *args ):
+        "Return node(s) with given name(s)"
+        if len( args ) == 1:
+            return self.nameToNode[ args[ 0 ] ]
+        return [ self.nameToNode[ n ] for n in args ]

    def addLink( self, node1, node2, port1=None, port2=None,
                 cls=None, **params ):