PlexKodiConnect/resources/lib/service_entry.py

536 lines
22 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2018-06-22 03:24:37 +10:00
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, unicode_literals
2018-06-22 03:24:37 +10:00
import logging
import sys
import xbmc
2018-11-19 00:59:17 +11:00
from . import utils, clientinfo, timing
from . import initialsetup
2018-06-22 03:24:37 +10:00
from . import kodimonitor
from . import sync, library_sync
2018-06-22 03:24:37 +10:00
from . import websocket_client
from . import plex_companion
2018-11-19 00:59:17 +11:00
from . import plex_functions as PF, playqueue as PQ
2018-06-22 03:24:37 +10:00
from . import playback_starter
from . import playqueue
from . import variables as v
2018-11-19 00:59:17 +11:00
from . import app
2018-06-22 03:24:37 +10:00
from . import loghandler
from . import backgroundthread
2018-11-19 00:59:17 +11:00
from .windows import userselect
2018-06-22 03:24:37 +10:00
###############################################################################
loghandler.config()
2018-11-19 00:59:17 +11:00
LOG = logging.getLogger("PLEX.service")
2018-06-22 03:24:37 +10:00
###############################################################################
2018-11-19 00:59:17 +11:00
WINDOW_PROPERTIES = (
"pms_token", "plex_token", "plex_authenticated", "plex_restricteduser",
"plex_allows_mediaDeletion", "plexkodiconnect.command", "plex_result")
2018-06-22 03:24:37 +10:00
2019-02-21 18:47:52 +11:00
class Service(object):
2018-06-22 03:24:37 +10:00
ws = None
2018-10-25 02:19:36 +11:00
sync = None
2018-06-22 03:24:37 +10:00
plexcompanion = None
def __init__(self):
self._init_done = False
# Kodi Version supported by PKC?
try:
v.database_paths()
except RuntimeError as err:
# Database does not exists
LOG.error('The current Kodi version is incompatible')
LOG.error('Error: %s', err)
# "The current Kodi version is not supported by PKC. Please consult
# the Plex forum."
utils.messageDialog(utils.lang(29999), utils.lang(39403))
return
2018-06-22 03:24:37 +10:00
# 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("Synching Plex artwork to Kodi: %s",
utils.settings('usePlexArtwork') == 'true')
2018-06-22 03:24:37 +10:00
LOG.info("Number of sync threads: %s",
utils.settings('syncThreadNumber'))
LOG.info('Playlist m3u encoding: %s', v.M3U_ENCODING)
2018-06-22 03:24:37 +10:00
LOG.info("Full sys.argv received: %s", sys.argv)
2018-08-06 02:13:16 +10:00
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'))
2018-11-03 20:36:37 +11:00
LOG.info("Db version: %s", utils.settings('dbCreatedWithVersion'))
LOG.info('Kodi video database version: %s', v.DB_VIDEO_VERSION)
LOG.info('Kodi music database version: %s', v.DB_MUSIC_VERSION)
LOG.info('Kodi texture database version: %s', v.DB_TEXTURE_VERSION)
2018-11-19 00:59:17 +11:00
# Reset some status in the PKC settings
# toggled to "No"
utils.settings('plex_status_fanarttv_lookup', value=utils.lang(106))
# toggled to "No"
utils.settings('plex_status_image_caching', value=utils.lang(106))
2018-11-19 00:59:17 +11:00
# Reset window props
for prop in WINDOW_PROPERTIES:
utils.window(prop, clear=True)
2018-06-22 03:24:37 +10:00
# Load/Reset PKC entirely - important for user/Kodi profile switch
2018-11-19 00:59:17 +11:00
# Clear video nodes properties
library_sync.VideoNodes().clearProperties()
2018-11-19 00:59:17 +11:00
clientinfo.getDeviceId()
# Init time-offset between Kodi and Plex
timing.KODI_PLEX_TIME_OFFSET = float(utils.settings('kodiplextimeoffset') or 0.0)
2018-11-19 00:59:17 +11:00
self.startup_completed = False
self.server_has_been_online = True
self.welcome_msg = True
self.connection_check_counter = 0
2019-02-08 23:52:33 +11:00
self.setup = None
self.alexa = None
self.playqueue = None
self.context_monitor = None
# Flags for other threads
self.connection_check_running = False
self.auth_running = False
self._init_done = True
2019-02-08 23:52:33 +11:00
@staticmethod
def isCanceled():
return xbmc.abortRequested or app.APP.stop_pkc
def on_connection_check(self, result):
"""
Call this method after PF.check_connection()
"""
try:
if result is False:
# Server is offline or cannot be reached
# Alert the user and suppress future warning
if app.CONN.online:
# PMS was online before
LOG.warn("Plex Media Server went offline")
app.CONN.online = False
app.APP.suspend_threads()
LOG.debug('Threads suspended')
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)
self.connection_check_counter += 1
# Periodically check if the IP changed every 15 seconds
if self.connection_check_counter > 150:
self.connection_check_counter = 0
server = self.setup.pick_pms()
if server:
LOG.debug('Found server: %s', server)
self.setup.save_pms_settings(server['baseURL'], server['token'])
self.setup.write_pms_to_settings(server)
app.CONN.load()
app.ACCOUNT.reset_session()
else:
# Server is online
self.connection_check_counter = 0
if not app.CONN.online:
# Server was offline before
if (self.welcome_msg is False and
utils.settings('show_pms_offline') == 'true'):
# Alert the user that server is online
utils.dialog('notification',
utils.lang(29999),
utils.lang(33003),
icon='{plex}',
time=5000,
sound=False)
LOG.info("Server is online and ready")
if app.ACCOUNT.authenticated:
# Server got offline when we were authenticated.
# Hence resume threads
app.APP.resume_threads()
app.CONN.online = True
finally:
self.connection_check_running = False
2018-06-22 03:24:37 +10:00
2019-02-08 23:52:33 +11:00
@staticmethod
def log_out():
2018-11-19 00:59:17 +11:00
"""
Ensures that lib sync threads are suspended; signs out user
"""
LOG.info('Log-out requested')
app.APP.suspend_threads()
LOG.info('Successfully suspended threads')
app.ACCOUNT.log_out()
2018-11-19 00:59:17 +11:00
LOG.info('User has been logged out')
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 not server:
2018-11-19 00:59:17 +11:00
LOG.info('We did not connect to a new PMS, aborting')
return False
LOG.info("User chose server %s with url %s",
server['name'], server['baseURL'])
if (server['machineIdentifier'] == app.CONN.machine_identifier and
server['baseURL'] == app.CONN.server):
2018-11-19 00:59:17 +11:00
LOG.info('User chose old PMS to connect to')
return False
# Save changes to to file
self.setup.save_pms_settings(server['baseURL'], server['token'])
2018-11-19 00:59:17 +11:00
self.setup.write_pms_to_settings(server)
self.log_out()
2018-11-19 00:59:17 +11:00
# Wipe Kodi and Plex database as well as playlists and video nodes
utils.wipe_database()
app.CONN.load()
app.ACCOUNT.reset_session()
app.ACCOUNT.set_unauthenticated()
self.server_has_been_online = False
self.welcome_msg = False
2018-11-27 02:58:25 +11:00
# Force a full sync
app.SYNC.run_lib_scan = 'full'
# Enable the main loop to continue
app.APP.suspend = False
2018-11-19 00:59:17 +11:00
LOG.info("Choosing new PMS complete")
return True
def switch_plex_user(self):
self.log_out()
2018-11-19 00:59:17 +11:00
# First remove playlists of old user
utils.delete_playlists()
# Remove video nodes
utils.delete_nodes()
app.ACCOUNT.set_unauthenticated()
2018-11-26 05:49:47 +11:00
# Force full sync after login
library_sync.force_full_sync()
2018-11-26 05:49:47 +11:00
app.SYNC.run_lib_scan = 'full'
# Enable the main loop to display user selection dialog
app.APP.suspend = False
2018-11-19 00:59:17 +11:00
return True
def toggle_plex_tv(self):
if app.ACCOUNT.plex_token:
LOG.info('Resetting plex.tv credentials in settings')
self.log_out()
2018-11-19 00:59:17 +11:00
app.ACCOUNT.clear()
else:
LOG.info('Login to plex.tv')
if self.setup.plex_tv_sign_in():
self.setup.write_credentials_to_settings()
app.ACCOUNT.load()
# Enable the main loop to continue
app.APP.suspend = False
2018-06-22 03:24:37 +10:00
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
"""
if self._do_auth():
if self.welcome_msg is True:
# Reset authentication warnings
self.welcome_msg = False
utils.dialog('notification',
utils.lang(29999),
"%s %s" % (utils.lang(33000),
app.ACCOUNT.plex_username),
icon='{plex}',
time=2000,
sound=False)
app.APP.resume_threads()
self.auth_running = False
def enter_new_pms_address(self):
server = self.setup.enter_new_pms_address()
if not server:
return
self.log_out()
# Save changes to to file
self.setup.save_pms_settings(server['baseURL'], server['token'])
self.setup.write_pms_to_settings(server)
if not v.KODIVERSION >= 18:
utils.settings('sslverify', value='false')
# Wipe Kodi and Plex database as well as playlists and video nodes
utils.wipe_database()
app.CONN.load()
app.ACCOUNT.reset_session()
app.ACCOUNT.set_unauthenticated()
self.server_has_been_online = False
self.welcome_msg = False
# Force a full sync
app.SYNC.run_lib_scan = 'full'
# Enable the main loop to continue
app.APP.suspend = False
LOG.info("Entering PMS address complete")
return True
def choose_plex_libraries(self):
if not app.CONN.online:
LOG.error('PMS not online to choose libraries')
# "{0} offline"
utils.dialog('notification',
utils.lang(29999),
utils.lang(39213).format(app.CONN.server_name or ''),
icon='{plex}')
return
if not app.ACCOUNT.authenticated:
LOG.error('Not yet authenticated for PMS to choose libraries')
# "Unauthorized for PMS"
utils.dialog('notification', utils.lang(29999), utils.lang(30017))
return
app.APP.suspend_threads()
from .library_sync import sections
try:
# Get newest sections from the PMS
if not sections.sync_from_pms(self):
return
if not sections.choose_libraries():
return
# Force a full sync
app.SYNC.run_lib_scan = 'full'
finally:
app.APP.resume_threads()
def _do_auth(self):
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 %s no longer has access - signing user out',
app.ACCOUNT.plex_username)
self.log_out()
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.APP.suspend = True
app.APP.suspend_threads()
LOG.debug('Threads suspended')
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('User %s does not have access to PMS %s on %s',
username, app.CONN.server_name, app.CONN.server)
# "User is unauthorized for server {0}"
utils.messageDialog(utils.lang(29999),
utils.lang(33010).format(app.CONN.server_name))
self.log_out()
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:
LOG.debug('Suspending threads')
app.APP.suspend = True
app.APP.suspend_threads()
LOG.debug('Threads suspended')
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
2018-06-22 03:24:37 +10:00
def ServiceEntryPoint(self):
if not self._init_done:
return
2018-06-22 03:24:37 +10:00
# Important: Threads depending on abortRequest will not trigger
# if profile switch happens more than once.
2018-11-19 00:59:17 +11:00
# Some plumbing
app.init()
app.APP.monitor = kodimonitor.KodiMonitor()
self.context_monitor = kodimonitor.ContextMonitor()
# Start immediately to catch user input even before auth
self.context_monitor.start()
2018-11-23 18:41:05 +11:00
app.APP.player = xbmc.Player()
2018-11-19 00:59:17 +11:00
# Initialize the PKC playqueues
PQ.init_playqueues()
2018-06-22 03:24:37 +10:00
# Server auto-detect
2018-11-19 00:59:17 +11:00
self.setup = initialsetup.InitialSetup()
self.setup.setup()
2018-06-22 03:24:37 +10:00
2018-11-19 00:59:17 +11:00
# Initialize important threads
2018-06-22 03:24:37 +10:00
self.ws = websocket_client.PMS_Websocket()
self.alexa = websocket_client.Alexa_Websocket()
2018-10-25 02:19:36 +11:00
self.sync = sync.Sync()
2018-06-22 03:24:37 +10:00
self.plexcompanion = plex_companion.PlexCompanion()
self.playqueue = playqueue.PlayqueueMonitor()
# Main PKC program loop
while not self.isCanceled():
2018-06-22 03:24:37 +10:00
# Check for PKC commands from other Python instances
plex_command = utils.window('plexkodiconnect.command')
2018-11-19 00:59:17 +11:00
if plex_command:
# Commands/user interaction received from other PKC Python
# instances (default.py and context.py instead of service.py)
utils.window('plexkodiconnect.command', clear=True)
task = None
2018-11-19 00:59:17 +11:00
if plex_command.startswith('PLAY-'):
# Add-on path playback!
task = playback_starter.PlaybackTask(
2018-11-19 00:59:17 +11:00
plex_command.replace('PLAY-', ''))
elif plex_command.startswith('CONTEXT_menu?'):
task = playback_starter.PlaybackTask(
2018-11-19 00:59:17 +11:00
'dummy?mode=context_menu&%s'
% plex_command.replace('CONTEXT_menu?', ''))
elif plex_command == 'choose_pms_server':
task = backgroundthread.FunctionAsTask(
self.choose_pms_server, None)
2018-11-19 00:59:17 +11:00
elif plex_command == 'switch_plex_user':
task = backgroundthread.FunctionAsTask(
self.switch_plex_user, None)
2018-11-19 00:59:17 +11:00
elif plex_command == 'enter_new_pms_address':
task = backgroundthread.FunctionAsTask(
self.enter_new_pms_address, None)
2018-11-19 00:59:17 +11:00
elif plex_command == 'toggle_plex_tv_sign_in':
task = backgroundthread.FunctionAsTask(
self.toggle_plex_tv, None)
2018-11-19 00:59:17 +11:00
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'
elif plex_command == 'select-libraries':
self.choose_plex_libraries()
2018-11-26 07:07:25 +11:00
elif plex_command == 'RESET-PKC':
utils.reset()
if task:
backgroundthread.BGThreader.addTasksToFront([task])
2018-11-19 00:59:17 +11:00
continue
if app.APP.suspend:
app.APP.monitor.waitForAbort(0.1)
continue
2018-06-22 03:24:37 +10:00
# Before proceeding, need to make sure:
# 1. Server is online
# 2. User is set
# 3. User has access to the server
if not app.CONN.online:
# Not online
2018-11-19 00:59:17 +11:00
server = app.CONN.server
if not server:
# No server info set in add-on settings
pass
elif not self.connection_check_running:
self.connection_check_running = True
task = backgroundthread.FunctionAsTask(
PF.check_connection,
self.on_connection_check,
server,
verifySSL=app.CONN.verify_ssl_cert)
backgroundthread.BGThreader.addTasksToFront([task])
continue
elif not app.ACCOUNT.authenticated:
# Plex server is online, but we're not yet authenticated
if not self.auth_running:
self.auth_running = True
task = backgroundthread.FunctionAsTask(
self.authenticate, None)
backgroundthread.BGThreader.addTasksToFront([task])
continue
elif not self.startup_completed:
self.startup_completed = True
self.ws.start()
self.sync.start()
self.plexcompanion.start()
self.playqueue.start()
if utils.settings('enable_alexa') == 'true':
self.alexa.start()
app.APP.monitor.waitForAbort(0.1)
# EXITING PKC
2018-06-22 03:24:37 +10:00
# Tell all threads to terminate (e.g. several lib sync threads)
LOG.debug('Aborting all threads')
2018-11-19 00:59:17 +11:00
app.APP.stop_pkc = True
# Will block until threads have quit
app.APP.stop_threads()
2018-06-22 03:24:37 +10:00
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()