cleaner API

[osm/vim-emu.git] / emuvim / dcemulator / net.py

diff --git a/emuvim/dcemulator/net.py b/emuvim/dcemulator/net.py
index 586b1e8..0efcafa 100644 (file)
--- a/emuvim/dcemulator/net.py
+++ b/emuvim/dcemulator/net.py
@@ -14,6 +14,12 @@ from node import Datacenter
 
 
 class DCNetwork(object):
+    """
+    Wraps the original Mininet class and provides
+    methods to add data centers, switches, etc.
+
+    This class is used by topology definition scripts.
+    """
 
     def __init__(self):
         self.dcs = {}
@@ -48,6 +54,10 @@ class DCNetwork(object):
         return s
 
     def addLink(self, node1, node2):
+        """
+        Able to handle Datacenter objects as link
+        end points.
+        """
         assert node1 is not None
         assert node2 is not None
         # ensure type of node1
@@ -74,6 +84,18 @@ class DCNetwork(object):
             raise Exception(
                 "one of the given nodes is not a Mininet switch or None")
 
+    def addDocker( self, name, **params ):
+        """
+        Wrapper for addDocker method provided by Dockernet.
+        """
+        return self.mnet.addDocker( name, **params)
+
+    def removeDocker( self, name, **params):
+        """
+        Wrapper for removeHost. Just to be complete.
+        """
+        return self.mnet.removeDocker(name, **params)
+
     def start(self):
         # start
         for dc in self.dcs.itervalues():
@@ -85,4 +107,3 @@ class DCNetwork(object):
 
     def CLI(self):
         CLI(self.mnet)
-