From 548580d817e80d80719e3cfb4cf7e842bd63fe1c Mon Sep 17 00:00:00 2001 From: Bob Lantz Date: Mon, 9 Apr 2012 23:54:17 +0000 Subject: [PATCH] Allow lists of nodes to be passed to getNodeByName ....which should perhaps be renamed!!! --- mininet/net.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/mininet/net.py b/mininet/net.py index 82ec2b0..d7a673d 100755 --- a/mininet/net.py +++ b/mininet/net.py @@ -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 ):