pyTivoService.py

   1 import beacon, httpserver
   2 import win32serviceutil
   3 import win32service
   4 import win32event
   5 import select, sys
   6 import config
   7
   8 class PyTivoService(win32serviceutil.ServiceFramework):
   9     _svc_name_ = 'pyTivo'
  10     _svc_display_name_ = 'pyTivo'
  11
  12     def __init__(self, args):
  13         win32serviceutil.ServiceFramework.__init__(self, args)
  14         self.stop_event = win32event.CreateEvent(None, 0, 0, None)
  15
  16     def SvcDoRun(self):
  17
  18         import sys, os
  19
  20         p = os.path.dirname(__file__)
  21
  22         f = open(os.path.join(p, 'log.txt'), 'w')
  23         sys.stdout = f
  24         sys.stderr = f
  25
  26
  27         port = config.getPort()
  28
  29         httpd = httpserver.TivoHTTPServer(('', int(port)),
  30                                           httpserver.TivoHTTPHandler)
  31
  32         for section, settings in config.getShares():
  33             httpd.add_container(section, settings)
  34
  35         b = beacon.Beacon()
  36         b.add_service('TiVoMediaServer:' + str(port) + '/http')
  37         b.start()
  38         if 'listen' in config.getBeaconAddresses():
  39             b.listen()
  40
  41         httpd.set_beacon(b)
  42
  43         while 1:
  44             sys.stdout.flush()
  45             (rx, tx, er) = select.select((httpd,), (), (), 5)
  46             for sck in rx:
  47                 sck.handle_request()
  48             rc = win32event.WaitForSingleObject(self.stop_event, 5)
  49             if rc == win32event.WAIT_OBJECT_0:
  50                 b.stop()
  51                 break
  52
  53     def SvcStop(self):
  54         win32event.SetEvent(self.stop_event)
  55
  56 if __name__ == '__main__':
  57     win32serviceutil.HandleCommandLine(PyTivoService)