Add a missing handler decorator to PyminDaemon test.

[software/pymin.git] / pymindaemon.py

diff --git a/pymindaemon.py b/pymindaemon.py
index 76a31d75eae0bdc54bb5417102cabf156758d5c1..1ddef6c5d2bff19927580e11ceeb368368346a3d 100644 (file)
--- a/pymindaemon.py
+++ b/pymindaemon.py
@@ -58,7 +58,12 @@ class PyminDaemon(EventLoop):
     def handle(self):
         r"handle() -> None :: Handle incoming events using the dispatcher."
         (msg, addr) = self.file.recvfrom(65535)
-        self.dispatcher.dispatch(msg)
+        result = self.dispatcher.dispatch(msg)
+        if result is None:
+            msg = 'OK 0'
+        else:
+            msg = 'OK %d\n%s' % (len(str(result)), result)
+        self.file.sendto(msg, addr)
         #try:
         #    d.dispatch(msg)
         #except dis.BadRouteError, inst:
@@ -75,6 +80,7 @@ class PyminDaemon(EventLoop):
 
 if __name__ == '__main__':
 
+    @handler
     def test_handler(*args):
         print 'test:', args