pymindaemon.py

   1 # vim: set encoding=utf-8 et sw=4 sts=4 :
   2
   3 r"""
   4 Python Administration Daemon.
   5
   6 Python Administration Daemon is an modular, extensible administration tool
   7 to administrate a set of services remotely (or localy) throw a simple
   8 command-line.
   9 """
  10
  11 import signal
  12 import socket
  13
  14 import dispatcher
  15 import eventloop
  16 import serializer
  17
  18 class PyminDaemon(eventloop.EventLoop):
  19     r"""PyminDaemon(bind_addr, routes) -> PyminDaemon instance
  20
  21     This class is well suited to run as a single process. It handles
  22     signals for controlled termination (SIGINT and SIGTERM), as well as
  23     a user signal to reload the configuration files (SIGUSR1).
  24
  25     bind_addr - is a tuple of (ip, port) where to bind the UDP socket to.
  26
  27     routes - is a dictionary where the key is a command string and the value
  28              is the command handler. This is passed directly to the Dispatcher.
  29
  30     Here is a simple usage example:
  31
  32     >>> def test_handler(*args): print 'test:', args
  33     >>> PyminDaemon(('', 9999), dict(test=test_handler)).run()
  34     """
  35
  36     def __init__(self, bind_addr, routes):
  37         r"""Initialize the PyminDaemon object.
  38
  39         See PyminDaemon class documentation for more info.
  40         """
  41         # Create and bind socket
  42         sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
  43         sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  44         sock.bind(bind_addr)
  45         # Create EventLoop
  46         eventloop.EventLoop.__init__(self, sock)
  47         # Create Dispatcher
  48         self.dispatcher = dispatcher.Dispatcher(routes)
  49         # Signal handling
  50         def quit(signum, frame):
  51             print "Shuting down ..."
  52             self.stop() # tell main event loop to stop
  53         def reload_config(signum, frame):
  54             print "Reloading configuration..."
  55             # TODO iterate handlers list propagating reload action
  56         signal.signal(signal.SIGINT, quit)
  57         signal.signal(signal.SIGTERM, quit)
  58         signal.signal(signal.SIGUSR1, reload_config)
  59
  60     def handle(self):
  61         r"handle() -> None :: Handle incoming events using the dispatcher."
  62         (msg, addr) = self.file.recvfrom(65535)
  63         try:
  64             result = self.dispatcher.dispatch(msg)
  65             if result is not None:
  66                 result = serializer.serialize(result)
  67             response = u'OK '
  68         except Exception, e:
  69             result = unicode(e)
  70             response = u'ERROR '
  71         if result is None:
  72             response += u'0'
  73         else:
  74             response += u'%d\n%s' % (len(result), result)
  75         self.file.sendto(response, addr)
  76
  77     def run(self):
  78         r"run() -> None :: Run the event loop (shortcut to loop())"
  79         try:
  80             return self.loop()
  81         except eventloop.LoopInterruptedError, e:
  82             pass
  83
  84 if __name__ == '__main__':
  85
  86     from dispatcher import handler
  87
  88     @handler
  89     def test_handler(*args):
  90         print 'test:', args
  91
  92     @handler
  93     def echo_handler(message):
  94         print 'echo:', message
  95         return message
  96
  97     PyminDaemon(('', 9999), dict(test=test_handler, echo=echo_handler)).run()
  98