small cleanup to sdn services, added loadservice line to daemon, and updated example service for how current services are coded

daemon/core/services/sdn.py

@@ -1,12 +1,13 @@
 """
 sdn.py defines services to start Open vSwitch and the Ryu SDN Controller.
 """
-import os
+
 import re
 
 from core.service import CoreService
 from core.service import ServiceManager
 
+
 class SdnService(CoreService):
     """
     Parent class for SDN services.
@@ -24,6 +25,7 @@ class SdnService(CoreService):
     def generateconfig(cls, node, filename, services):
         return ""
 
+
 class OvsService(SdnService):
     _name = "OvsService"
     _group = "SDN"
@@ -55,7 +57,7 @@ class OvsService(SdnService):
         cfg += "ifconfig ovsbr0 up\n"
 
         for ifc in node.netifs():
-            if hasattr(ifc, 'control') and ifc.control == True:
+            if hasattr(ifc, 'control') and ifc.control is True:
                 continue
             ifnumstr = re.findall(r"\d+", ifc.name)
             ifnum = ifnumstr[0]
@@ -77,20 +79,19 @@ class OvsService(SdnService):
                     if has_zebra == 0:
                         cfg += "ip -6 addr add %s dev rtr%s\n" % (ifcaddr, ifnum)
                 else:
-                    raise Value, "invalid address: %s", x
+                    raise ValueError("invalid address: %s" % ifcaddr)
 
             # add interfaces to bridge
             cfg += "ovs-vsctl add-port ovsbr0 eth%s\n" % ifnum
             cfg += "ovs-vsctl add-port ovsbr0 sw%s\n" % ifnum
 
-
         # Add rule for default controller if there is one local (even if the controller is not local, it finds it)
         cfg += "ovs-vsctl set-controller ovsbr0 tcp:127.0.0.1:6633\n"
 
         # Setup default flows
         portnum = 1
         for ifc in node.netifs():
-            if hasattr(ifc, 'control') and ifc.control == True:
+            if hasattr(ifc, 'control') and ifc.control is True:
                 continue
             cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum, portnum + 1)
             cfg += "ovs-ofctl add-flow ovsbr0 priority=1000,in_port=%d,action=output:%d\n" % (portnum + 1, portnum)
@@ -98,7 +99,8 @@ class OvsService(SdnService):
 
         return cfg
 
-class ryuService(SdnService):
+
+class RyuService(SdnService):
     _name = "ryuService"
     _group = "SDN"
     _depends = ()
@@ -106,21 +108,21 @@ class ryuService(SdnService):
     _configs = ('ryuService.sh',)
     _startindex = 50
     _startup = ('sh ryuService.sh',)
-    _shutdown = ('killall ryu-manager')
+    _shutdown = ('killall ryu-manager',)
 
     @classmethod
     def generateconfig(cls, node, filename, services):
-        ''' Return a string that will be written to filename, or sent to the
+        """
+        Return a string that will be written to filename, or sent to the
         GUI for user customization.
-        '''
+        """
         app_path = "/usr/local/lib/python2.7/dist-packages/ryu/app"
         cfg = "#!/bin/sh\n"
         cfg += "# auto-generated by ryuService (ryuService.py)\n"
         cfg += '/usr/local/bin/ryu-manager --observe-links %s/ofctl_rest.py %s/rest_topology.py' % (app_path, app_path)
         return cfg
 
+
 def load_services():
     ServiceManager.add(OvsService)
-    ServiceManager.add(ryuService)
-
-
+    ServiceManager.add(RyuService)

daemon/examples/myservices/sample.py

@@ -3,19 +3,19 @@
 # Copyright (c)2010-2012 the Boeing Company.
 # See the LICENSE file included in this distribution.
 #
-''' Sample user-defined service.
-'''
-
-import os
+"""
+Sample user-defined service.
+"""
 
 from core.misc.ipaddress import Ipv4Prefix
-from core.misc.ipaddress import Ipv6Prefix
 from core.service import CoreService
 from core.service import ServiceManager
 
+
 class MyService(CoreService):
-    ''' This is a sample user-defined service. 
-    '''
+    """
+    This is a sample user-defined service.
+    """
     # a unique name is required, without spaces
     _name = "MyService"
     # you can create your own group here
@@ -36,9 +36,10 @@ class MyService(CoreService):
 
     @classmethod
     def generateconfig(cls, node, filename, services):
-        ''' Return a string that will be written to filename, or sent to the
+        """
+        Return a string that will be written to filename, or sent to the
         GUI for user customization.
-        '''
+        """
         cfg = "#!/bin/sh\n"
         cfg += "# auto-generated by MyService (sample.py)\n"
 
@@ -51,16 +52,18 @@ class MyService(CoreService):
 
     @staticmethod
     def subnetentry(x):
-        ''' Generate a subnet declaration block given an IPv4 prefix string
+        """
+        Generate a subnet declaration block given an IPv4 prefix string
         for inclusion in the config file.
-        '''
+        """
         if x.find(":") >= 0:
             # this is an IPv6 address
             return ""
         else:
-            net = IPv4Prefix(x)
-            return 'echo "  network %s"' % (net)
+            net = Ipv4Prefix(x)
+            return 'echo "  network %s"' % net
 
+
+def load_services():
     # this line is required to add the above class to the list of available services
     ServiceManager.add(MyService)
-

daemon/sbin/core-daemon

@@ -43,6 +43,7 @@ from core.services import bird
 from core.services import dockersvc
 from core.services import nrl
 from core.services import quagga
+from core.services import sdn
 from core.services import security
 from core.services import startup
 from core.services import ucarp
@@ -364,5 +365,6 @@ if __name__ == "__main__":
     ucarp.load_services()
     dockersvc.load_services()
     startup.load_services()
+    sdn.load_services()
 
     main()