stream.py: un-decorator Stream.register_method
authorIsaku Yamahata <yamahata@valinux.co.jp>
Wed, 17 Oct 2012 18:18:16 +0000 (03:18 +0900)
committerBen Pfaff <blp@nicira.com>
Wed, 17 Oct 2012 18:22:08 +0000 (11:22 -0700)
c38f8724aeb994fd840fa4283a07e38c3c114d68 made stream.py not use class
decorator. So Stream.register need not to be decorator any more.
So simplify it.

Signed-off-by: Isaku Yamahata <yamahata@valinux.co.jp>
Signed-off-by: Ben Pfaff <blp@nicira.com>
python/ovs/stream.py

index 6bd0ccbe8b22c931e07e22644de410631801f2d1..dad68483e7a8ea72735397a470be85e972f22a6b 100644 (file)
@@ -54,11 +54,8 @@ class Stream(object):
     _SOCKET_METHODS = {}
 
     @staticmethod
-    def register_method(method):
-        def _register_method(cls):
-            Stream._SOCKET_METHODS[method + ":"] = cls
-            return cls
-        return _register_method
+    def register_method(method, cls):
+        Stream._SOCKET_METHODS[method + ":"] = cls
 
     @staticmethod
     def _find_method(name):
@@ -350,7 +347,7 @@ class UnixStream(Stream):
         connect_path = suffix
         return  ovs.socket_util.make_unix_socket(socket.SOCK_STREAM,
                                                  True, None, connect_path)
-UnixStream = Stream.register_method("unix")(UnixStream)
+Stream.register_method("unix", UnixStream)
 
 
 class TCPStream(Stream):
@@ -361,4 +358,4 @@ class TCPStream(Stream):
         if not error:
             sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
         return error, sock
-TCPStream = Stream.register_method("tcp")(TCPStream)
+Stream.register_method("tcp", TCPStream)