Make Host iterable so it can be used as a sequence when serializing.

[software/pymin.git] / pymindaemon.py

diff --git a/pymindaemon.py b/pymindaemon.py
index 1e9589aa242478b49e1e06990e2fefbe51142298..b8369b9d67622467cc7e8b36187459a664d899ac 100644 (file)
--- a/pymindaemon.py
+++ b/pymindaemon.py
@@ -58,18 +58,17 @@ class PyminDaemon(EventLoop):
     def handle(self):
         r"handle() -> None :: Handle incoming events using the dispatcher."
         (msg, addr) = self.file.recvfrom(65535)
     def handle(self):
         r"handle() -> None :: Handle incoming events using the dispatcher."
         (msg, addr) = self.file.recvfrom(65535)

-        result = self.dispatcher.dispatch(msg)
+        try:
+            result = self.dispatcher.dispatch(msg)
+            response = 'OK '
+        except Exception, e:
+            result = str(e)
+            response = 'ERROR '

         if result is None:
         if result is None:

-            msg = 'OK 0'
+            response += '0'

         else:
         else:

-            msg = 'OK %d\n%s' % (len(str(result)), result)
-        self.file.sendto(msg, addr)
-        #try:
-        #    d.dispatch(msg)
-        #except dis.BadRouteError, inst:
-        #    sock.sendto('Bad route from : ' + inst.cmd + '\n', addr)
-        #except dis.CommandNotFoundError, inst:
-        #    sock.sendto('Command not found : ' + inst.cmd + '\n', addr)
+            response += '%d\n%s' % (len(str(result)), result)
+        self.file.sendto(response, addr)

 
     def run(self):
         r"run() -> None :: Run the event loop (shortcut to loop())"
 
     def run(self):
         r"run() -> None :: Run the event loop (shortcut to loop())"


@@ -80,6 +79,7 @@ class PyminDaemon(EventLoop):
 
 if __name__ == '__main__':
 
 
 if __name__ == '__main__':
 

+    @handler

     def test_handler(*args):
         print 'test:', args
 
     def test_handler(*args):
         print 'test:', args