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