#!/usr/bin/env python # -*- coding: utf-8 -*- from __future__ import absolute_import, division, unicode_literals import logging import sys import xbmc from . import utils, clientinfo, timing from . import initialsetup, artwork from . import kodimonitor from . import sync from . import websocket_client from . import plex_companion from . import plex_functions as PF, playqueue as PQ from . import playback_starter from . import playqueue from . import variables as v from . import app from . import loghandler from .windows import userselect ############################################################################### loghandler.config() LOG = logging.getLogger("PLEX.service") ############################################################################### WINDOW_PROPERTIES = ( "plex_online", "plex_command_processed", "plex_shouldStop", "plex_dbScan", "plex_customplayqueue", "plex_playbackProps", "pms_token", "plex_token", "pms_server", "plex_machineIdentifier", "plex_servername", "plex_authenticated", "PlexUserImage", "useDirectPaths", "countError", "countUnauthorized", "plex_restricteduser", "plex_allows_mediaDeletion", "plex_command", "plex_result", "plex_force_transcode_pix" ) class Service(): ws = None sync = None plexcompanion = None ws_running = False alexa_running = False sync_running = False plexcompanion_running = False kodimonitor_running = False playback_starter_running = False def __init__(self): # Initial logging LOG.info("======== START %s ========", v.ADDON_NAME) LOG.info("Platform: %s", v.PLATFORM) LOG.info("KODI Version: %s", v.KODILONGVERSION) LOG.info("%s Version: %s", v.ADDON_NAME, v.ADDON_VERSION) LOG.info("PKC Direct Paths: %s", utils.settings('useDirectPaths') == '1') LOG.info("Number of sync threads: %s", utils.settings('syncThreadNumber')) LOG.info('Playlist m3u encoding: %s', v.M3U_ENCODING) LOG.info("Full sys.argv received: %s", sys.argv) LOG.info('Sync playlists: %s', utils.settings('enablePlaylistSync')) LOG.info('Synching only specific Kodi playlists: %s', utils.settings('syncSpecificKodiPlaylists') == 'true') LOG.info('Kodi playlist prefix: %s', utils.settings('syncSpecificKodiPlaylistsPrefix')) LOG.info('Synching only specific Plex playlists: %s', utils.settings('syncSpecificPlexPlaylistsPrefix') == 'true') LOG.info('Play playlist prefix: %s', utils.settings('syncSpecificPlexPlaylistsPrefix')) LOG.info('XML decoding being used: %s', utils.ETREE) LOG.info("Db version: %s", utils.settings('dbCreatedWithVersion')) # Reset window props for prop in WINDOW_PROPERTIES: utils.window(prop, clear=True) # To detect Kodi profile switches utils.window('plex_kodiProfile', value=utils.try_decode(xbmc.translatePath("special://profile"))) self.monitor = xbmc.Monitor() # Load/Reset PKC entirely - important for user/Kodi profile switch # Clear video nodes properties from .library_sync import videonodes videonodes.VideoNodes().clearProperties() clientinfo.getDeviceId() # Init time-offset between Kodi and Plex timing.KODI_PLEX_TIME_OFFSET = utils.settings('kodiplextimeoffset') or 0.0 def isCanceled(self): return xbmc.abortRequested or app.APP.stop_pkc def log_out(self): """ Ensures that lib sync threads are suspended; signs out user """ LOG.info('Log-out requested') app.SYNC.suspend_library_thread = True i = 0 while app.SYNC.db_scan: i += 1 xbmc.sleep(50) if i > 100: LOG.error('Could not stop library sync, aborting log-out') # Failed to reset PMS and plex.tv connects. Try to restart Kodi utils.messageDialog(utils.lang(29999), utils.lang(39208)) # Resuming threads, just in case app.SYNC.suspend_library_thread = False return False LOG.info('Successfully stopped library sync') app.ACCOUNT.log_out() LOG.info('User has been logged out') return True def choose_pms_server(self, manual=False): LOG.info("Choosing PMS server requested, starting") if manual: if not self.setup.enter_new_pms_address(): return False else: server = self.setup.pick_pms(showDialog=True) if server is None: LOG.info('We did not connect to a new PMS, aborting') return False LOG.info("User chose server %s", server['name']) if server['baseURL'] == app.CONN.server: LOG.info('User chose old PMS to connect to') return False self.setup.write_pms_to_settings(server) if not self.log_out(): return False # Wipe Kodi and Plex database as well as playlists and video nodes utils.wipe_database() app.CONN.load() LOG.info("Choosing new PMS complete") return True def switch_plex_user(self): if not self.log_out(): return False # First remove playlists of old user utils.delete_playlists() # Remove video nodes utils.delete_nodes() return True def toggle_plex_tv(self): if utils.settings('plexToken'): LOG.info('Reseting plex.tv credentials in settings') app.ACCOUNT.clear() return True else: LOG.info('Login to plex.tv') return self.setup.plex_tv_sign_in() def authenticate(self): """ Authenticate the current user or prompt to log-in Returns True if successful, False if not. 'aborted' if user chose to abort """ LOG.info('Authenticating user') if app.ACCOUNT.plex_username and not app.ACCOUNT.force_login: # Found a user in the settings, try to authenticate LOG.info('Trying to authenticate with old settings') res = PF.check_connection(app.CONN.server, token=app.ACCOUNT.pms_token, verifySSL=app.CONN.verify_ssl_cert) if res is False: LOG.error('Something went wrong while checking connection') return False elif res == 401: LOG.error('User token no longer valid. Sign user out') app.ACCOUNT.clear() return False elif res >= 400: LOG.error('Answer from PMS is not as expected') return False LOG.info('Successfully authenticated using old settings') app.ACCOUNT.set_authenticated() return True while True: # Could not use settings - try to get Plex user list from plex.tv if app.ACCOUNT.plex_token: LOG.info("Trying to connect to plex.tv to get a user list") user, _ = userselect.start() if not user: LOG.info('No user received') app.CONN.pms_status = 'Stop' return False username = user.title user_id = user.id token = user.authToken else: LOG.info("Trying to authenticate without a token") username = '' user_id = '' token = '' res = PF.check_connection(app.CONN.server, token=token, verifySSL=app.CONN.verify_ssl_cert) if res is False: LOG.error('Something went wrong while checking connection') return False elif res == 401: if app.ACCOUNT.plex_token: LOG.error('Token no longer valid') # "Your Plex token is no longer valid. Logging-out of plex.tv" utils.messageDialog(utils.lang(29999), utils.lang(33010)) app.ACCOUNT.clear() return False else: # "Failed to authenticate. Did you login to plex.tv?" utils.messageDialog(utils.lang(29999), utils.lang(39023)) if self.setup.plex_tv_sign_in(): self.setup.write_credentials_to_settings() app.ACCOUNT.load() continue else: return False elif res >= 400: LOG.error('Answer from PMS is not as expected') return False LOG.info('Successfully authenticated') # Got new values that need to be saved utils.settings('username', value=username) utils.settings('userid', value=user_id) utils.settings('accessToken', value=token) app.ACCOUNT.load() app.ACCOUNT.set_authenticated() return True def ServiceEntryPoint(self): # Important: Threads depending on abortRequest will not trigger # if profile switch happens more than once. app.init() # Some plumbing artwork.IMAGE_CACHING_SUSPENDS = [ app.SYNC.suspend_library_thread, app.SYNC.stop_sync, app.SYNC.db_scan ] if not utils.settings('imageSyncDuringPlayback') == 'true': artwork.IMAGE_CACHING_SUSPENDS.append(app.SYNC.suspend_sync) # Initialize the PKC playqueues PQ.init_playqueues() # Server auto-detect self.setup = initialsetup.InitialSetup() self.setup.setup() # Initialize important threads self.ws = websocket_client.PMS_Websocket() self.alexa = websocket_client.Alexa_Websocket() self.sync = sync.Sync() self.plexcompanion = plex_companion.PlexCompanion() self.specialmonitor = kodimonitor.SpecialMonitor() self.playback_starter = playback_starter.PlaybackStarter() self.playqueue = playqueue.PlayqueueMonitor() server_online = True welcome_msg = True counter = 0 while not self.isCanceled(): if utils.window('plex_kodiProfile') != v.KODI_PROFILE: # Profile change happened, terminate this thread and others LOG.info("Kodi profile was: %s and changed to: %s. " "Terminating old PlexKodiConnect thread.", v.KODI_PROFILE, utils.window('plex_kodiProfile')) break plex_command = utils.window('plex_command') if plex_command: # Commands/user interaction received from other PKC Python # instances (default.py and context.py instead of service.py) utils.window('plex_command', clear=True) if plex_command.startswith('PLAY-'): # Add-on path playback! app.APP.command_pipeline_queue.put( plex_command.replace('PLAY-', '')) elif plex_command.startswith('NAVIGATE-'): app.APP.command_pipeline_queue.put( plex_command.replace('NAVIGATE-', '')) elif plex_command.startswith('CONTEXT_menu?'): app.APP.command_pipeline_queue.put( 'dummy?mode=context_menu&%s' % plex_command.replace('CONTEXT_menu?', '')) elif plex_command == 'choose_pms_server': if self.choose_pms_server(): utils.window('plex_online', clear=True) app.ACCOUNT.set_unauthenticated() server_online = False welcome_msg = False elif plex_command == 'switch_plex_user': if self.switch_plex_user(): app.ACCOUNT.set_unauthenticated() elif plex_command == 'enter_new_pms_address': if self.setup.enter_new_pms_address(): if self.log_out(): utils.window('plex_online', clear=True) app.ACCOUNT.set_unauthenticated() server_online = False welcome_msg = False elif plex_command == 'toggle_plex_tv_sign_in': if self.toggle_plex_tv(): app.ACCOUNT.set_unauthenticated() elif plex_command == 'repair-scan': app.SYNC.run_lib_scan = 'repair' elif plex_command == 'full-scan': app.SYNC.run_lib_scan = 'full' elif plex_command == 'fanart-scan': app.SYNC.run_lib_scan = 'fanart' elif plex_command == 'textures-scan': app.SYNC.run_lib_scan = 'textures' continue # Before proceeding, need to make sure: # 1. Server is online # 2. User is set # 3. User has access to the server if utils.window('plex_online') == "true": # Plex server is online if app.CONN.pms_status == 'Stop': xbmc.sleep(500) continue elif app.CONN.pms_status == '401': # Unauthorized access, revoke token LOG.info('401 received - revoking token') app.ACCOUNT.clear() app.CONN.pms_status = 'Auth' utils.window('plex_serverStatus', value='Auth') continue if not app.ACCOUNT.authenticated: LOG.info('Not yet authenticated') # Do authentication if not self.authenticate(): continue # Start up events if welcome_msg is True: # Reset authentication warnings welcome_msg = False utils.dialog('notification', utils.lang(29999), "%s %s" % (utils.lang(33000), app.ACCOUNT.plex_username), icon='{plex}', time=2000, sound=False) # Start monitoring kodi events if not self.kodimonitor_running: self.kodimonitor_running = kodimonitor.KodiMonitor() self.specialmonitor.start() # Start the Websocket Client if not self.ws_running: self.ws_running = True self.ws.start() # Start the Alexa thread if (not self.alexa_running and utils.settings('enable_alexa') == 'true'): self.alexa_running = True self.alexa.start() # Start the syncing thread if not self.sync_running: self.sync_running = True self.sync.start() # Start the Plex Companion thread if not self.plexcompanion_running: self.plexcompanion_running = True self.plexcompanion.start() if not self.playback_starter_running: self.playback_starter_running = True self.playback_starter.start() self.playqueue.start() else: # Wait until Plex server is online # or Kodi is shut down. server = app.CONN.server if not server: # No server info set in add-on settings pass elif PF.check_connection(server, verifySSL=True) is False: # Server is offline or cannot be reached # Alert the user and suppress future warning if server_online: server_online = False utils.window('plex_online', value="false") # Suspend threads app.SYNC.suspend_library_thread = True LOG.warn("Plex Media Server went offline") if utils.settings('show_pms_offline') == 'true': utils.dialog('notification', utils.lang(33001), "%s %s" % (utils.lang(29999), utils.lang(33002)), icon='{plex}', sound=False) counter += 1 # Periodically check if the IP changed, e.g. per minute if counter > 20: counter = 0 setup = initialsetup.InitialSetup() tmp = setup.pick_pms() if tmp: setup.write_pms_to_settings(tmp) app.CONN.load() else: # Server is online counter = 0 if not server_online: # Server was offline when Kodi started. server_online = True # Alert the user that server is online. if (welcome_msg is False and utils.settings('show_pms_offline') == 'true'): utils.dialog('notification', utils.lang(29999), utils.lang(33003), icon='{plex}', time=5000, sound=False) LOG.info("Server %s is online and ready.", server) utils.window('plex_online', value="true") if app.ACCOUNT.authenticated: # Server got offline when we were authenticated. # Hence resume threads app.SYNC.suspend_library_thread = False if self.monitor.waitForAbort(0.05): # Abort was requested while waiting. We should exit break # Tell all threads to terminate (e.g. several lib sync threads) app.APP.stop_pkc = True utils.window('plex_service_started', clear=True) LOG.info("======== STOP %s ========", v.ADDON_NAME) def start(): # Safety net - Kody starts PKC twice upon first installation! if utils.window('plex_service_started') == 'true': EXIT = True else: utils.window('plex_service_started', value='true') EXIT = False # Delay option DELAY = int(utils.settings('startupDelay')) LOG.info("Delaying Plex startup by: %s sec...", DELAY) if EXIT: LOG.error('PKC service.py already started - exiting this instance') elif DELAY and xbmc.Monitor().waitForAbort(DELAY): # Start the service LOG.info("Abort requested while waiting. PKC not started.") else: Service().ServiceEntryPoint()