service.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 # Module: service
   4 # Created on: 26.01.2017
   5
   6 import threading
   7 import SocketServer
   8 import socket
   9 from xbmc import Monitor
  10 from resources.lib.KodiHelper import KodiHelper
  11 from resources.lib.MSLHttpRequestHandler import MSLHttpRequestHandler
  12 from resources.lib.NetflixHttpRequestHandler import NetflixHttpRequestHandler
  13
  14 # helper function to select an unused port on the host machine
  15 def select_unused_port():
  16     sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  17     sock.bind(('127.0.0.1', 0))
  18     addr, port = sock.getsockname()
  19     sock.close()
  20     return port
  21
  22 kodi_helper = KodiHelper()
  23
  24 # pick & store a port for the MSL service
  25 msl_port = select_unused_port()
  26 kodi_helper.set_setting('msl_service_port', str(msl_port))
  27 kodi_helper.log(msg='[MSL] Picked Port: ' + str(msl_port))
  28
  29 # pick & store a port for the internal Netflix HTTP proxy service
  30 ns_port = select_unused_port()
  31 kodi_helper.set_setting('netflix_service_port', str(ns_port))
  32 kodi_helper.log(msg='[NS] Picked Port: ' + str(ns_port))
  33
  34 # server defaults
  35 SocketServer.TCPServer.allow_reuse_address = True
  36
  37 # configure the MSL Server
  38 msl_server = SocketServer.TCPServer(('127.0.0.1', msl_port), MSLHttpRequestHandler)
  39 msl_server.server_activate()
  40 msl_server.timeout = 1
  41
  42 # configure the Netflix Data Server
  43 nd_server = SocketServer.TCPServer(('127.0.0.1', ns_port), NetflixHttpRequestHandler)
  44 nd_server.server_activate()
  45 nd_server.timeout = 1
  46
  47 if __name__ == '__main__':
  48     monitor = Monitor()
  49
  50     # start thread for MLS servie
  51     msl_thread = threading.Thread(target=msl_server.serve_forever)
  52     msl_thread.daemon = True
  53     msl_thread.start()
  54
  55     # start thread for Netflix HTTP service
  56     nd_thread = threading.Thread(target=nd_server.serve_forever)
  57     nd_thread.daemon = True
  58     nd_thread.start()
  59
  60     # kill the services if kodi monitor tells us to
  61     while not monitor.abortRequested():
  62         if monitor.waitForAbort(5):
  63             msl_server.shutdown()
  64             nd_server.shutdown()
  65             break
  66
  67     # MSL service shutdown sequence
  68     msl_server.server_close()
  69     msl_server.socket.close()
  70     msl_server.shutdown()
  71     kodi_helper.log(msg='Stopped MSL Service')
  72
  73     # Netflix service shutdown sequence
  74     nd_server.server_close()
  75     nd_server.socket.close()
  76     nd_server.shutdown()
  77     kodi_helper.log(msg='Stopped HTTP Service')