1735 lines
No EOL
64 KiB
Python
1735 lines
No EOL
64 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
##################################################################################################
|
|
|
|
import threading
|
|
from datetime import datetime
|
|
import Queue
|
|
|
|
import xbmc
|
|
import xbmcgui
|
|
import xbmcvfs
|
|
|
|
import api
|
|
import utils
|
|
import clientinfo
|
|
import downloadutils
|
|
import itemtypes
|
|
import embydb_functions as embydb
|
|
import kodidb_functions as kodidb
|
|
import read_embyserver as embyserver
|
|
import userclient
|
|
import videonodes
|
|
|
|
import PlexAPI
|
|
import PlexFunctions
|
|
|
|
##################################################################################################
|
|
|
|
|
|
@utils.ThreadMethodsAdditionalStop('emby_shouldStop')
|
|
@utils.ThreadMethods
|
|
class ThreadedGetMetadata(threading.Thread):
|
|
"""
|
|
Threaded download of Plex XML metadata for a certain library item.
|
|
Fills the out_queue with the downloaded etree XML objects
|
|
|
|
Input:
|
|
queue Queue.Queue() object that you'll need to fill up
|
|
with Plex itemIds
|
|
out_queue Queue.Queue() object where this thread will store
|
|
the downloaded metadata XMLs as etree objects
|
|
lock threading.Lock(), used for counting where we are
|
|
"""
|
|
def __init__(self, queue, out_queue, lock):
|
|
self.queue = queue
|
|
self.out_queue = out_queue
|
|
self.lock = lock
|
|
threading.Thread.__init__(self)
|
|
|
|
def run(self):
|
|
# cache local variables because it's faster
|
|
queue = self.queue
|
|
out_queue = self.out_queue
|
|
lock = self.lock
|
|
threadStopped = self.threadStopped
|
|
global getMetadataCount
|
|
while threadStopped() is False:
|
|
# grabs Plex item from queue
|
|
try:
|
|
updateItem = queue.get(block=False)
|
|
# Empty queue
|
|
except Queue.Empty:
|
|
xbmc.sleep(100)
|
|
continue
|
|
# Download Metadata
|
|
plexXML = PlexFunctions.GetPlexMetadata(updateItem['itemId'])
|
|
if plexXML is None:
|
|
# Did not receive a valid XML - skip that item for now
|
|
queue.task_done()
|
|
continue
|
|
|
|
updateItem['XML'] = plexXML
|
|
# place item into out queue
|
|
out_queue.put(updateItem)
|
|
# Keep track of where we are at
|
|
with lock:
|
|
getMetadataCount += 1
|
|
# signals to queue job is done
|
|
queue.task_done()
|
|
|
|
|
|
@utils.ThreadMethodsAdditionalStop('emby_shouldStop')
|
|
@utils.ThreadMethods
|
|
class ThreadedProcessMetadata(threading.Thread):
|
|
"""
|
|
Not yet implemented - if ever. Only to be called by ONE thread!
|
|
Processes the XML metadata in the queue
|
|
|
|
Input:
|
|
queue: Queue.Queue() object that you'll need to fill up with
|
|
the downloaded XML eTree objects
|
|
itemType: as used to call functions in itemtypes.py
|
|
e.g. 'Movies' => itemtypes.Movies()
|
|
lock: threading.Lock(), used for counting where we are
|
|
"""
|
|
def __init__(self, queue, itemType, lock):
|
|
self.queue = queue
|
|
self.lock = lock
|
|
self.itemType = itemType
|
|
threading.Thread.__init__(self)
|
|
|
|
def run(self):
|
|
# Constructs the method name, e.g. itemtypes.Movies
|
|
itemFkt = getattr(itemtypes, self.itemType)
|
|
# cache local variables because it's faster
|
|
queue = self.queue
|
|
lock = self.lock
|
|
threadStopped = self.threadStopped
|
|
global processMetadataCount
|
|
global processingViewName
|
|
with itemFkt() as item:
|
|
while threadStopped() is False:
|
|
# grabs item from queue
|
|
try:
|
|
updateItem = queue.get(block=False)
|
|
except Queue.Empty:
|
|
xbmc.sleep(100)
|
|
continue
|
|
# Do the work; lock to be sure we've only got 1 Thread
|
|
plexitem = updateItem['XML']
|
|
method = updateItem['method']
|
|
viewName = updateItem['viewName']
|
|
viewId = updateItem['viewId']
|
|
title = updateItem['title']
|
|
itemSubFkt = getattr(item, method)
|
|
with lock:
|
|
# Get the one child entry in the xml and process
|
|
for child in plexitem:
|
|
itemSubFkt(child,
|
|
viewtag=viewName,
|
|
viewid=viewId)
|
|
# Keep track of where we are at
|
|
processMetadataCount += 1
|
|
processingViewName = title
|
|
del plexitem
|
|
del updateItem
|
|
# signals to queue job is done
|
|
queue.task_done()
|
|
|
|
|
|
@utils.ThreadMethodsAdditionalStop('emby_shouldStop')
|
|
@utils.ThreadMethods
|
|
class ThreadedShowSyncInfo(threading.Thread):
|
|
"""
|
|
Threaded class to show the Kodi statusbar of the metadata download.
|
|
|
|
Input:
|
|
dialog xbmcgui.DialogProgressBG() object to show progress
|
|
locks = [downloadLock, processLock] Locks() to the other threads
|
|
total: Total number of items to get
|
|
"""
|
|
def __init__(self, dialog, locks, total, itemType):
|
|
self.locks = locks
|
|
self.total = total
|
|
self.addonName = clientinfo.ClientInfo().getAddonName()
|
|
self.dialog = dialog
|
|
self.itemType = itemType
|
|
threading.Thread.__init__(self)
|
|
|
|
def run(self):
|
|
# cache local variables because it's faster
|
|
total = self.total
|
|
dialog = self.dialog
|
|
threadStopped = self.threadStopped
|
|
downloadLock = self.locks[0]
|
|
processLock = self.locks[1]
|
|
dialog.create("%s: Sync %s: %s items"
|
|
% (self.addonName, self.itemType, str(total)),
|
|
"Starting")
|
|
global getMetadataCount
|
|
global processMetadataCount
|
|
global processingViewName
|
|
total = 2 * total
|
|
totalProgress = 0
|
|
while threadStopped() is False:
|
|
with downloadLock:
|
|
getMetadataProgress = getMetadataCount
|
|
with processLock:
|
|
processMetadataProgress = processMetadataCount
|
|
viewName = processingViewName
|
|
totalProgress = getMetadataProgress + processMetadataProgress
|
|
try:
|
|
percentage = int(float(totalProgress) / float(total)*100.0)
|
|
except ZeroDivisionError:
|
|
percentage = 0
|
|
try:
|
|
dialog.update(
|
|
percentage,
|
|
message="Downloaded: %s. Processed: %s: %s"
|
|
% (getMetadataProgress, processMetadataProgress,
|
|
viewName))
|
|
except:
|
|
# Wierd formating of the string viewName?!?
|
|
pass
|
|
# Sleep for x milliseconds
|
|
xbmc.sleep(500)
|
|
dialog.close()
|
|
|
|
|
|
@utils.logging
|
|
@utils.ThreadMethodsAdditionalSuspend('suspend_LibraryThread')
|
|
@utils.ThreadMethodsAdditionalStop('emby_shouldStop')
|
|
@utils.ThreadMethods
|
|
class LibrarySync(threading.Thread):
|
|
|
|
_shared_state = {}
|
|
|
|
# Track websocketclient updates
|
|
addedItems = []
|
|
updateItems = []
|
|
userdataItems = []
|
|
removeItems = []
|
|
forceLibraryUpdate = False
|
|
refresh_views = False
|
|
|
|
def __init__(self):
|
|
|
|
self.__dict__ = self._shared_state
|
|
|
|
# How long should we look into the past for fast syncing items (in s)
|
|
self.syncPast = 60
|
|
|
|
self.clientInfo = clientinfo.ClientInfo()
|
|
self.doUtils = downloadutils.DownloadUtils()
|
|
self.user = userclient.UserClient()
|
|
self.emby = embyserver.Read_EmbyServer()
|
|
self.vnodes = videonodes.VideoNodes()
|
|
self.syncThreadNumber = int(utils.settings('syncThreadNumber'))
|
|
|
|
self.installSyncDone = True if \
|
|
utils.settings('SyncInstallRunDone') == 'true' else False
|
|
|
|
threading.Thread.__init__(self)
|
|
|
|
def progressDialog(self, title, forced=False):
|
|
|
|
dialog = None
|
|
|
|
if utils.settings('dbSyncIndicator') == "true" or forced:
|
|
dialog = xbmcgui.DialogProgressBG()
|
|
dialog.create(self.addonName, title)
|
|
self.logMsg("Show progress dialog: %s" % title, 2)
|
|
|
|
return dialog
|
|
|
|
def startSync(self):
|
|
# Only run fastSync AFTER startup when SyncInstallRunDone has already
|
|
# been set
|
|
utils.window('emby_dbScan', value="true")
|
|
completed = self.fastSync()
|
|
if not completed:
|
|
# Fast sync failed or server plugin is not found
|
|
self.logMsg("Something went wrong, starting full sync", 0)
|
|
completed = self.fullSync(manualrun=True)
|
|
utils.window('emby_dbScan', value="false")
|
|
return completed
|
|
|
|
def fastSync(self):
|
|
"""
|
|
Fast incremential lib sync
|
|
|
|
Using /library/recentlyAdded is NOT working as changes to lib items are
|
|
not reflected
|
|
"""
|
|
self.compare = True
|
|
# Get last sync time
|
|
lastSync = self.lastSync - self.syncPast
|
|
if not lastSync:
|
|
# Original Emby format:
|
|
# lastSync = "2016-01-01T00:00:00Z"
|
|
# January 1, 2015 at midnight:
|
|
lastSync = 1420070400
|
|
# Set new timestamp NOW because sync might take a while
|
|
self.saveLastSync()
|
|
|
|
# Get all PMS items already saved in Kodi
|
|
# Also get checksums of every Plex items already saved in Kodi
|
|
allKodiElementsId = {}
|
|
with embydb.GetEmbyDB() as emby_db:
|
|
for itemtype in PlexFunctions.EmbyItemtypes():
|
|
try:
|
|
allKodiElementsId.update(
|
|
dict(emby_db.getChecksum(itemtype)))
|
|
except ValueError:
|
|
pass
|
|
|
|
self.allKodiElementsId = allKodiElementsId
|
|
|
|
# Run through views and get latest changed elements using time diff
|
|
self.updatelist = []
|
|
self.allPlexElementsId = {}
|
|
updateKodiVideoLib = False
|
|
for view in self.views:
|
|
if self.threadStopped():
|
|
return True
|
|
# Get items per view
|
|
items = PlexFunctions.GetAllPlexLeaves(
|
|
view['id'], updatedAt=lastSync)
|
|
if not items:
|
|
continue
|
|
# Get one itemtype, because they're the same in the PMS section
|
|
plexType = items[0].attrib['type']
|
|
# Populate self.updatelist
|
|
self.GetUpdatelist(items,
|
|
PlexFunctions.GetItemClassFromType(plexType),
|
|
PlexFunctions.GetMethodFromPlexType(plexType),
|
|
view['name'],
|
|
view['id'],
|
|
dontCheck=True)
|
|
# Process self.updatelist
|
|
if self.updatelist:
|
|
if self.updatelist[0]['itemType'] in ['Movies', 'TVShows']:
|
|
updateKodiVideoLib = True
|
|
self.GetAndProcessXMLs(
|
|
PlexFunctions.GetItemClassFromType(plexType))
|
|
self.updatelist = []
|
|
# Let Kodi grab the artwork now
|
|
if updateKodiVideoLib:
|
|
xbmc.executebuiltin('UpdateLibrary(video)')
|
|
# Update userdata
|
|
for view in self.views:
|
|
self.PlexUpdateWatched(
|
|
view['id'],
|
|
PlexFunctions.GetItemClassFromType(view['itemtype']),
|
|
lastViewedAt=lastSync)
|
|
# Reset and return
|
|
self.allKodiElementsId = {}
|
|
self.allPlexElementsId = {}
|
|
return True
|
|
|
|
def saveLastSync(self):
|
|
# Save last sync time
|
|
self.lastSync = utils.getUnixTimestamp()
|
|
|
|
def initializeDBs(self):
|
|
"""
|
|
Run once during startup to verify that emby db exists.
|
|
"""
|
|
embyconn = utils.kodiSQL('emby')
|
|
embycursor = embyconn.cursor()
|
|
# Create the tables for the emby database
|
|
# emby, view, version
|
|
embycursor.execute(
|
|
"""CREATE TABLE IF NOT EXISTS emby(
|
|
emby_id TEXT UNIQUE, media_folder TEXT, emby_type TEXT, media_type TEXT, kodi_id INTEGER,
|
|
kodi_fileid INTEGER, kodi_pathid INTEGER, parent_id INTEGER, checksum INTEGER)""")
|
|
embycursor.execute(
|
|
"""CREATE TABLE IF NOT EXISTS view(
|
|
view_id TEXT UNIQUE, view_name TEXT, media_type TEXT, kodi_tagid INTEGER)""")
|
|
embycursor.execute("CREATE TABLE IF NOT EXISTS version(idVersion TEXT)")
|
|
embyconn.commit()
|
|
|
|
# content sync: movies, tvshows, musicvideos, music
|
|
embyconn.close()
|
|
return
|
|
|
|
def fullSync(self, manualrun=False, repair=False):
|
|
# Only run once when first setting up. Can be run manually.
|
|
self.compare = manualrun or repair
|
|
music_enabled = utils.settings('enableMusic') == "true"
|
|
|
|
# Add sources
|
|
utils.sourcesXML()
|
|
|
|
if manualrun:
|
|
message = "Manual sync"
|
|
elif repair:
|
|
message = "Repair sync"
|
|
else:
|
|
message = "Initial sync"
|
|
utils.window('emby_initialScan', value="true")
|
|
# Set new timestamp NOW because sync might take a while
|
|
self.saveLastSync()
|
|
starttotal = datetime.now()
|
|
|
|
# Ensure that DBs exist if called for very first time
|
|
self.initializeDBs()
|
|
# Set views
|
|
self.maintainViews()
|
|
|
|
# Sync video library
|
|
# process = {
|
|
|
|
# 'movies': self.movies,
|
|
# 'musicvideos': self.musicvideos,
|
|
# 'tvshows': self.tvshows
|
|
# }
|
|
|
|
process = {
|
|
'movies': self.PlexMovies,
|
|
'tvshows': self.PlexTVShows
|
|
}
|
|
|
|
for itemtype in process:
|
|
startTime = datetime.now()
|
|
completed = process[itemtype]()
|
|
if not completed:
|
|
return False
|
|
else:
|
|
elapsedTime = datetime.now() - startTime
|
|
self.logMsg(
|
|
"SyncDatabase (finished %s in: %s)"
|
|
% (itemtype, str(elapsedTime).split('.')[0]), 1)
|
|
|
|
# # sync music
|
|
# if music_enabled:
|
|
|
|
# musicconn = utils.kodiSQL('music')
|
|
# musiccursor = musicconn.cursor()
|
|
|
|
# startTime = datetime.now()
|
|
# completed = self.music(embycursor, musiccursor, pDialog)
|
|
# if not completed:
|
|
|
|
# utils.window('emby_dbScan', clear=True)
|
|
|
|
# embycursor.close()
|
|
# musiccursor.close()
|
|
# return False
|
|
# else:
|
|
# musicconn.commit()
|
|
# embyconn.commit()
|
|
# elapsedTime = datetime.now() - startTime
|
|
# self.logMsg(
|
|
# "SyncDatabase (finished music in: %s)"
|
|
# % (str(elapsedTime).split('.')[0]), 1)
|
|
# musiccursor.close()
|
|
|
|
xbmc.executebuiltin('UpdateLibrary(video)')
|
|
elapsedtotal = datetime.now() - starttotal
|
|
|
|
utils.window('emby_initialScan', clear=True)
|
|
xbmcgui.Dialog().notification(
|
|
heading=self.addonName,
|
|
message="%s completed in: %s"
|
|
% (message, str(elapsedtotal).split('.')[0]),
|
|
icon="special://home/addons/plugin.video.plexkodiconnect/icon.png",
|
|
sound=False)
|
|
return True
|
|
|
|
def maintainViews(self):
|
|
"""
|
|
Compare the views to Plex
|
|
"""
|
|
# Open DB links
|
|
embyconn = utils.kodiSQL('emby')
|
|
embycursor = embyconn.cursor()
|
|
kodiconn = utils.kodiSQL('video')
|
|
kodicursor = kodiconn.cursor()
|
|
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
kodi_db = kodidb.Kodidb_Functions(kodicursor)
|
|
doUtils = self.doUtils
|
|
vnodes = self.vnodes
|
|
|
|
# Get views
|
|
result = doUtils.downloadUrl("{server}/library/sections")
|
|
if not result:
|
|
self.logMsg("Error download PMS views, abort maintainViews", -1)
|
|
return False
|
|
|
|
# total nodes for window properties
|
|
vnodes.clearProperties()
|
|
totalnodes = 0
|
|
|
|
# Set views for supported media type
|
|
mediatypes = [
|
|
'movie',
|
|
'show'
|
|
]
|
|
for folderItem in result:
|
|
folder = folderItem.attrib
|
|
mediatype = folder['type']
|
|
if mediatype in mediatypes:
|
|
folderid = folder['key']
|
|
foldername = folder['title']
|
|
viewtype = folder['type']
|
|
|
|
# Get current media folders from emby database
|
|
view = emby_db.getView_byId(folderid)
|
|
try:
|
|
current_viewname = view[0]
|
|
current_viewtype = view[1]
|
|
current_tagid = view[2]
|
|
|
|
except TypeError:
|
|
self.logMsg("Creating viewid: %s in Emby database." % folderid, 1)
|
|
tagid = kodi_db.createTag(foldername)
|
|
# Create playlist for the video library
|
|
if mediatype in ['movies', 'tvshows', 'musicvideos']:
|
|
utils.playlistXSP(mediatype, foldername, viewtype)
|
|
# Create the video node
|
|
if mediatype in ['movies', 'tvshows', 'musicvideos', 'homevideos']:
|
|
vnodes.viewNode(totalnodes, foldername, mediatype, viewtype)
|
|
totalnodes += 1
|
|
# Add view to emby database
|
|
emby_db.addView(folderid, foldername, viewtype, tagid)
|
|
|
|
else:
|
|
self.logMsg(' '.join((
|
|
|
|
"Found viewid: %s" % folderid,
|
|
"viewname: %s" % current_viewname,
|
|
"viewtype: %s" % current_viewtype,
|
|
"tagid: %s" % current_tagid)), 2)
|
|
|
|
# View was modified, update with latest info
|
|
if current_viewname != foldername:
|
|
self.logMsg("viewid: %s new viewname: %s" % (folderid, foldername), 1)
|
|
tagid = kodi_db.createTag(foldername)
|
|
|
|
# Update view with new info
|
|
emby_db.updateView(foldername, tagid, folderid)
|
|
|
|
if mediatype != "music":
|
|
if emby_db.getView_byName(current_viewname) is None:
|
|
# The tag could be a combined view. Ensure there's no other tags
|
|
# with the same name before deleting playlist.
|
|
utils.playlistXSP(
|
|
mediatype, current_viewname, current_viewtype, True)
|
|
# Delete video node
|
|
if mediatype != "musicvideos":
|
|
vnodes.viewNode(
|
|
totalnodes,
|
|
current_viewname,
|
|
mediatype,
|
|
current_viewtype,
|
|
delete=True)
|
|
# Added new playlist
|
|
if mediatype in ['movies', 'tvshows', 'musicvideos']:
|
|
utils.playlistXSP(mediatype, foldername, viewtype)
|
|
# Add new video node
|
|
if mediatype != "musicvideos":
|
|
vnodes.viewNode(totalnodes, foldername, mediatype, viewtype)
|
|
totalnodes += 1
|
|
|
|
# Update items with new tag
|
|
items = emby_db.getItem_byView(folderid)
|
|
for item in items:
|
|
# Remove the "s" from viewtype for tags
|
|
kodi_db.updateTag(
|
|
current_tagid, tagid, item[0], current_viewtype[:-1])
|
|
else:
|
|
if mediatype != "music":
|
|
# Validate the playlist exists or recreate it
|
|
if mediatype in ['movies', 'tvshows', 'musicvideos']:
|
|
utils.playlistXSP(mediatype, foldername, viewtype)
|
|
# Create the video node if not already exists
|
|
if mediatype != "musicvideos":
|
|
vnodes.viewNode(totalnodes, foldername, mediatype, viewtype)
|
|
totalnodes += 1
|
|
else:
|
|
# Add video nodes listings
|
|
# vnodes.singleNode(totalnodes, "Favorite movies", "movies", "favourites")
|
|
# totalnodes += 1
|
|
# vnodes.singleNode(totalnodes, "Favorite tvshows", "tvshows", "favourites")
|
|
# totalnodes += 1
|
|
# vnodes.singleNode(totalnodes, "channels", "movies", "channels")
|
|
# totalnodes += 1
|
|
# Save total
|
|
utils.window('Emby.nodes.total', str(totalnodes))
|
|
|
|
# update views for all:
|
|
self.views = emby_db.getAllViewInfo()
|
|
self.logMsg("views saved: %s" % self.views, 1)
|
|
# commit changes to DB
|
|
embyconn.commit()
|
|
kodiconn.commit()
|
|
embyconn.close()
|
|
kodiconn.close()
|
|
|
|
def GetUpdatelist(self, xml, itemType, method, viewName, viewId,
|
|
dontCheck=False):
|
|
"""
|
|
Adds items to self.updatelist as well as self.allPlexElementsId dict
|
|
|
|
Input:
|
|
xml: PMS answer for section items
|
|
itemType: 'Movies', 'TVShows', ...
|
|
method: Method name to be called with this itemtype
|
|
see itemtypes.py
|
|
viewName: Name of the Plex view (e.g. 'My TV shows')
|
|
viewId: Id/Key of Plex library (e.g. '1')
|
|
dontCheck: If True, skips checksum check but assumes
|
|
that all items in xml must be processed
|
|
|
|
Output: self.updatelist, self.allPlexElementsId
|
|
self.updatelist APPENDED(!!) list itemids (Plex Keys as
|
|
as received from API.getRatingKey())
|
|
One item in this list is of the form:
|
|
'itemId': xxx,
|
|
'itemType': 'Movies','TVShows', ...
|
|
'method': 'add_update', 'add_updateSeason', ...
|
|
'viewName': xxx,
|
|
'viewId': xxx,
|
|
'title': xxx
|
|
|
|
self.allPlexElementsId APPENDED(!!) dict
|
|
= {itemid: checksum}
|
|
"""
|
|
if self.compare or not dontCheck:
|
|
# Manual sync
|
|
for item in xml:
|
|
# Skipping items 'title=All episodes' without a 'ratingKey'
|
|
if not item.attrib.get('ratingKey', False):
|
|
continue
|
|
API = PlexAPI.API(item)
|
|
itemId = API.getRatingKey()
|
|
title, sorttitle = API.getTitle()
|
|
plex_checksum = API.getChecksum()
|
|
self.allPlexElementsId[itemId] = plex_checksum
|
|
kodi_checksum = self.allKodiElementsId.get(itemId)
|
|
if kodi_checksum != plex_checksum:
|
|
# Only update if movie is not in Kodi or checksum is
|
|
# different
|
|
self.updatelist.append({'itemId': itemId,
|
|
'itemType': itemType,
|
|
'method': method,
|
|
'viewName': viewName,
|
|
'viewId': viewId,
|
|
'title': title})
|
|
else:
|
|
# Initial or repair sync: get all Plex movies
|
|
for item in xml:
|
|
# Only look at valid items = Plex library items
|
|
if not item.attrib.get('ratingKey', False):
|
|
continue
|
|
API = PlexAPI.API(item)
|
|
itemId = API.getRatingKey()
|
|
title, sorttitle = API.getTitle()
|
|
plex_checksum = API.getChecksum()
|
|
self.allPlexElementsId[itemId] = plex_checksum
|
|
self.updatelist.append({'itemId': itemId,
|
|
'itemType': itemType,
|
|
'method': method,
|
|
'viewName': viewName,
|
|
'viewId': viewId,
|
|
'title': title})
|
|
|
|
def GetAndProcessXMLs(self, itemType):
|
|
"""
|
|
Downloads all XMLs for itemType (e.g. Movies, TV-Shows). Processes them
|
|
by then calling itemtypes.<itemType>()
|
|
|
|
Input:
|
|
itemType: 'Movies', 'TVShows', ...
|
|
self.updatelist
|
|
"""
|
|
# Some logging, just in case.
|
|
self.logMsg("self.updatelist: %s" % self.updatelist, 2)
|
|
itemNumber = len(self.updatelist)
|
|
if itemNumber == 0:
|
|
return True
|
|
|
|
# Run through self.updatelist, get XML metadata per item
|
|
# Initiate threads
|
|
self.logMsg("Starting sync threads", 1)
|
|
getMetadataQueue = Queue.Queue()
|
|
processMetadataQueue = Queue.Queue(maxsize=100)
|
|
getMetadataLock = threading.Lock()
|
|
processMetadataLock = threading.Lock()
|
|
# To keep track
|
|
global getMetadataCount
|
|
getMetadataCount = 0
|
|
global processMetadataCount
|
|
processMetadataCount = 0
|
|
global processingViewName
|
|
processingViewName = ''
|
|
# Populate queue: GetMetadata
|
|
for updateItem in self.updatelist:
|
|
getMetadataQueue.put(updateItem)
|
|
# Spawn GetMetadata threads for downloading
|
|
threads = []
|
|
for i in range(min(self.syncThreadNumber, itemNumber)):
|
|
thread = ThreadedGetMetadata(getMetadataQueue,
|
|
processMetadataQueue,
|
|
getMetadataLock)
|
|
thread.setDaemon(True)
|
|
thread.start()
|
|
threads.append(thread)
|
|
self.logMsg("%s download threads spawned" % len(threads), 1)
|
|
# Spawn one more thread to process Metadata, once downloaded
|
|
thread = ThreadedProcessMetadata(processMetadataQueue,
|
|
itemType,
|
|
processMetadataLock)
|
|
thread.setDaemon(True)
|
|
thread.start()
|
|
threads.append(thread)
|
|
self.logMsg("Processing thread spawned", 1)
|
|
# Start one thread to show sync progress
|
|
dialog = xbmcgui.DialogProgressBG()
|
|
thread = ThreadedShowSyncInfo(dialog,
|
|
[getMetadataLock, processMetadataLock],
|
|
itemNumber,
|
|
itemType)
|
|
thread.setDaemon(True)
|
|
thread.start()
|
|
threads.append(thread)
|
|
self.logMsg("Kodi Infobox thread spawned", 1)
|
|
|
|
# Wait until finished
|
|
getMetadataQueue.join()
|
|
processMetadataQueue.join()
|
|
# Kill threads
|
|
self.logMsg("Waiting to kill threads", 1)
|
|
for thread in threads:
|
|
thread.stopThread()
|
|
self.logMsg("Stop sent to all threads", 1)
|
|
# Wait till threads are indeed dead
|
|
for thread in threads:
|
|
thread.join(5.0)
|
|
if thread.isAlive():
|
|
self.logMsg("Could not terminate thread", -1)
|
|
try:
|
|
del threads
|
|
except:
|
|
self.logMsg("Could not delete threads", -1)
|
|
# Make sure dialog window is closed
|
|
if dialog:
|
|
dialog.close()
|
|
self.logMsg("Sync threads finished", 1)
|
|
self.updatelist = []
|
|
return True
|
|
|
|
def PlexMovies(self):
|
|
# Initialize
|
|
self.allPlexElementsId = {}
|
|
|
|
itemType = 'Movies'
|
|
|
|
views = [x for x in self.views if x['itemtype'] == 'movie']
|
|
self.logMsg("Processing Plex %s. Libraries: %s" % (itemType, views), 1)
|
|
|
|
self.allKodiElementsId = {}
|
|
if self.compare:
|
|
with embydb.GetEmbyDB() as emby_db:
|
|
# Get movies from Plex server
|
|
# Pull the list of movies and boxsets in Kodi
|
|
try:
|
|
self.allKodiElementsId = dict(emby_db.getChecksum('Movie'))
|
|
except ValueError:
|
|
self.allKodiElementsId = {}
|
|
|
|
##### PROCESS MOVIES #####
|
|
self.updatelist = []
|
|
for view in views:
|
|
if self.threadStopped():
|
|
return False
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
all_plexmovies = PlexFunctions.GetPlexSectionResults(viewId)
|
|
if not all_plexmovies:
|
|
self.logMsg("Couldnt get section items, aborting for view.", 1)
|
|
continue
|
|
# Populate self.updatelist and self.allPlexElementsId
|
|
self.GetUpdatelist(all_plexmovies,
|
|
itemType,
|
|
'add_update',
|
|
viewName,
|
|
viewId)
|
|
self.GetAndProcessXMLs(itemType)
|
|
self.logMsg("Processed view %s with ID %s" % (viewName, viewId), 1)
|
|
# Update viewstate
|
|
for view in views:
|
|
if self.threadStopped():
|
|
return False
|
|
self.PlexUpdateWatched(view['id'], itemType)
|
|
|
|
##### PROCESS DELETES #####
|
|
if self.compare:
|
|
# Manual sync, process deletes
|
|
with itemtypes.Movies() as Movie:
|
|
for kodimovie in self.allKodiElementsId:
|
|
if kodimovie not in self.allPlexElementsId:
|
|
Movie.remove(kodimovie)
|
|
self.logMsg("%s sync is finished." % itemType, 1)
|
|
return True
|
|
|
|
def PlexUpdateWatched(self, viewId, itemType,
|
|
lastViewedAt=None, updatedAt=None):
|
|
"""
|
|
Updates plex elements' view status ('watched' or 'unwatched') and
|
|
also updates resume times.
|
|
This is done by downloading one XML for ALL elements with viewId
|
|
"""
|
|
xml = PlexFunctions.GetAllPlexLeaves(viewId,
|
|
lastViewedAt=lastViewedAt,
|
|
updatedAt=updatedAt)
|
|
if xml:
|
|
itemMth = getattr(itemtypes, itemType)
|
|
with itemMth() as method:
|
|
method.updateUserdata(xml)
|
|
|
|
def musicvideos(self, embycursor, kodicursor, pdialog):
|
|
# Get musicvideos from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
mvideos = itemtypes.MusicVideos(embycursor, kodicursor)
|
|
|
|
views = emby_db.getView_byType('musicvideos')
|
|
self.logMsg("Media folders: %s" % views, 1)
|
|
|
|
for view in views:
|
|
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Gathering musicvideos from view: %s..." % viewName)
|
|
|
|
# Initial or repair sync
|
|
all_embymvideos = emby.getMusicVideos(viewId, dialog=pdialog)
|
|
total = all_embymvideos['TotalRecordCount']
|
|
embymvideos = all_embymvideos['Items']
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (viewName, total))
|
|
|
|
count = 0
|
|
for embymvideo in embymvideos:
|
|
# Process individual musicvideo
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
title = embymvideo['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
mvideos.add_update(embymvideo, viewName, viewId)
|
|
else:
|
|
self.logMsg("MusicVideos finished.", 2)
|
|
|
|
return True
|
|
|
|
def PlexTVShows(self):
|
|
# Initialize
|
|
self.allPlexElementsId = {}
|
|
itemType = 'TVShows'
|
|
|
|
views = [x for x in self.views if x['itemtype'] == 'show']
|
|
self.logMsg("Media folders for %s: %s" % (itemType, views), 1)
|
|
|
|
self.allKodiElementsId = {}
|
|
if self.compare:
|
|
with embydb.GetEmbyDB() as emby_db:
|
|
# Get movies from Plex server
|
|
# Pull the list of TV shows already in Kodi
|
|
try:
|
|
all_koditvshows = dict(emby_db.getChecksum('Series'))
|
|
self.allKodiElementsId.update(all_koditvshows)
|
|
except ValueError:
|
|
pass
|
|
# Same for seasons
|
|
try:
|
|
all_kodiseasons = dict(emby_db.getChecksum('Season'))
|
|
self.allKodiElementsId.update(all_kodiseasons)
|
|
except ValueError:
|
|
pass
|
|
# Same for the episodes (sub-element of shows/series)
|
|
try:
|
|
all_kodiepisodes = dict(emby_db.getChecksum('Episode'))
|
|
self.allKodiElementsId.update(all_kodiepisodes)
|
|
except ValueError:
|
|
pass
|
|
|
|
##### PROCESS TV Shows #####
|
|
self.updatelist = []
|
|
for view in views:
|
|
if self.threadStopped():
|
|
return False
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
allPlexTvShows = PlexFunctions.GetPlexSectionResults(viewId)
|
|
if not allPlexTvShows:
|
|
self.logMsg(
|
|
"Error downloading show view xml for view %s" % viewId, -1)
|
|
continue
|
|
# Populate self.updatelist and self.allPlexElementsId
|
|
self.GetUpdatelist(allPlexTvShows,
|
|
itemType,
|
|
'add_update',
|
|
viewName,
|
|
viewId)
|
|
self.logMsg("Analyzed view %s with ID %s" % (viewName, viewId), 1)
|
|
|
|
# COPY for later use
|
|
allPlexTvShowsId = self.allPlexElementsId.copy()
|
|
|
|
##### PROCESS TV Seasons #####
|
|
# Cycle through tv shows
|
|
for tvShowId in allPlexTvShowsId:
|
|
if self.threadStopped():
|
|
return False
|
|
# Grab all seasons to tvshow from PMS
|
|
seasons = PlexFunctions.GetAllPlexChildren(tvShowId)
|
|
if not seasons:
|
|
self.logMsg(
|
|
"Error downloading season xml for show %s" % tvShowId, -1)
|
|
continue
|
|
# Populate self.updatelist and self.allPlexElementsId
|
|
self.GetUpdatelist(seasons,
|
|
itemType,
|
|
'add_updateSeason',
|
|
None,
|
|
tvShowId) # send showId instead of viewid
|
|
self.logMsg("Analyzed all seasons of TV show with Plex Id %s"
|
|
% tvShowId, 1)
|
|
|
|
##### PROCESS TV Episodes #####
|
|
# Cycle through tv shows
|
|
for view in views:
|
|
if self.threadStopped():
|
|
return False
|
|
# Grab all episodes to tvshow from PMS
|
|
episodes = PlexFunctions.GetAllPlexLeaves(view['id'])
|
|
if not episodes:
|
|
self.logMsg(
|
|
"Error downloading episod xml for view %s"
|
|
% view.get('name'), -1)
|
|
continue
|
|
# Populate self.updatelist and self.allPlexElementsId
|
|
self.GetUpdatelist(episodes,
|
|
itemType,
|
|
'add_updateEpisode',
|
|
None,
|
|
None)
|
|
self.logMsg("Analyzed all episodes of TV show with Plex Id %s"
|
|
% tvShowId, 1)
|
|
|
|
# Process self.updatelist
|
|
self.GetAndProcessXMLs(itemType)
|
|
self.logMsg("GetAndProcessXMLs completed", 1)
|
|
# Refresh season info
|
|
# Cycle through tv shows
|
|
with itemtypes.TVShows() as TVshow:
|
|
for tvShowId in allPlexTvShowsId:
|
|
XMLtvshow = PlexFunctions.GetPlexMetadata(tvShowId)
|
|
TVshow.refreshSeasonEntry(XMLtvshow, tvShowId)
|
|
self.logMsg("Season info refreshed", 1)
|
|
|
|
# Update viewstate:
|
|
for view in views:
|
|
self.PlexUpdateWatched(view['id'], itemType)
|
|
|
|
if self.compare:
|
|
# Manual sync, process deletes
|
|
with itemtypes.TVShows() as TVShow:
|
|
for kodiTvElement in self.allKodiElementsId:
|
|
if kodiTvElement not in self.allPlexElementsId:
|
|
TVShow.remove(kodiTvElement)
|
|
self.logMsg("%s sync is finished." % itemType, 1)
|
|
return True
|
|
|
|
def music(self, embycursor, kodicursor, pdialog):
|
|
# Get music from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
music = itemtypes.Music(embycursor, kodicursor)
|
|
|
|
process = {
|
|
|
|
'artists': [emby.getArtists, music.add_updateArtist],
|
|
'albums': [emby.getAlbums, music.add_updateAlbum],
|
|
'songs': [emby.getSongs, music.add_updateSong]
|
|
}
|
|
types = ['artists', 'albums', 'songs']
|
|
for type in types:
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Gathering %s..." % type)
|
|
|
|
all_embyitems = process[type][0](dialog=pdialog)
|
|
total = all_embyitems['TotalRecordCount']
|
|
embyitems = all_embyitems['Items']
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (type, total))
|
|
|
|
count = 0
|
|
for embyitem in embyitems:
|
|
# Process individual item
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
title = embyitem['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
|
|
process[type][1](embyitem)
|
|
else:
|
|
self.logMsg("%s finished." % type, 2)
|
|
|
|
return True
|
|
|
|
# Reserved for websocket_client.py and fast start
|
|
def triage_items(self, process, items):
|
|
|
|
processlist = {
|
|
|
|
'added': self.addedItems,
|
|
'update': self.updateItems,
|
|
'userdata': self.userdataItems,
|
|
'remove': self.removeItems
|
|
}
|
|
if items:
|
|
if process == "userdata":
|
|
itemids = []
|
|
for item in items:
|
|
itemids.append(item['ItemId'])
|
|
items = itemids
|
|
|
|
self.logMsg("Queue %s: %s" % (process, items), 1)
|
|
processlist[process].extend(items)
|
|
|
|
def incrementalSync(self):
|
|
|
|
embyconn = utils.kodiSQL('emby')
|
|
embycursor = embyconn.cursor()
|
|
kodiconn = utils.kodiSQL('video')
|
|
kodicursor = kodiconn.cursor()
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
pDialog = None
|
|
update_embydb = False
|
|
|
|
if self.refresh_views:
|
|
# Received userconfig update
|
|
self.refresh_views = False
|
|
self.maintainViews(embycursor, kodicursor)
|
|
self.forceLibraryUpdate = True
|
|
update_embydb = True
|
|
|
|
if self.addedItems or self.updateItems or self.userdataItems or self.removeItems:
|
|
# Only present dialog if we are going to process items
|
|
pDialog = self.progressDialog('Incremental sync')
|
|
|
|
|
|
process = {
|
|
|
|
'added': self.addedItems,
|
|
'update': self.updateItems,
|
|
'userdata': self.userdataItems,
|
|
'remove': self.removeItems
|
|
}
|
|
types = ['added', 'update', 'userdata', 'remove']
|
|
for type in types:
|
|
|
|
if process[type] and utils.window('emby_kodiScan') != "true":
|
|
|
|
listItems = list(process[type])
|
|
del process[type][:] # Reset class list
|
|
|
|
items_process = itemtypes.Items(embycursor, kodicursor)
|
|
update = False
|
|
|
|
# Prepare items according to process type
|
|
if type == "added":
|
|
items = emby.sortby_mediatype(listItems)
|
|
|
|
elif type in ("userdata", "remove"):
|
|
items = emby_db.sortby_mediaType(listItems, unsorted=False)
|
|
|
|
else:
|
|
items = emby_db.sortby_mediaType(listItems)
|
|
if items.get('Unsorted'):
|
|
sorted_items = emby.sortby_mediatype(items['Unsorted'])
|
|
doupdate = items_process.itemsbyId(sorted_items, "added", pDialog)
|
|
if doupdate:
|
|
embyupdate, kodiupdate_video = doupdate
|
|
if embyupdate:
|
|
update_embydb = True
|
|
if kodiupdate_video:
|
|
self.forceLibraryUpdate = True
|
|
del items['Unsorted']
|
|
|
|
doupdate = items_process.itemsbyId(items, type, pDialog)
|
|
if doupdate:
|
|
embyupdate, kodiupdate_video = doupdate
|
|
if embyupdate:
|
|
update_embydb = True
|
|
if kodiupdate_video:
|
|
self.forceLibraryUpdate = True
|
|
|
|
if update_embydb:
|
|
update_embydb = False
|
|
self.logMsg("Updating emby database.", 1)
|
|
embyconn.commit()
|
|
self.saveLastSync()
|
|
|
|
if self.forceLibraryUpdate:
|
|
# Force update the Kodi library
|
|
self.forceLibraryUpdate = False
|
|
self.dbCommit(kodiconn)
|
|
|
|
self.logMsg("Updating video library.", 1)
|
|
utils.window('emby_kodiScan', value="true")
|
|
xbmc.executebuiltin('UpdateLibrary(video)')
|
|
|
|
if pDialog:
|
|
pDialog.close()
|
|
|
|
kodicursor.close()
|
|
embycursor.close()
|
|
|
|
|
|
def compareDBVersion(self, current, minimum):
|
|
# It returns True is database is up to date. False otherwise.
|
|
self.logMsg("current: %s minimum: %s" % (current, minimum), 1)
|
|
currMajor, currMinor, currPatch = current.split(".")
|
|
minMajor, minMinor, minPatch = minimum.split(".")
|
|
|
|
if currMajor > minMajor:
|
|
return True
|
|
elif currMajor == minMajor and (currMinor > minMinor or
|
|
(currMinor == minMinor and currPatch >= minPatch)):
|
|
return True
|
|
else:
|
|
# Database out of date.
|
|
return False
|
|
|
|
def run(self):
|
|
|
|
try:
|
|
self.run_internal()
|
|
except Exception as e:
|
|
utils.window('emby_dbScan', clear=True)
|
|
xbmcgui.Dialog().ok(
|
|
heading=self.addonName,
|
|
line1=("Library sync thread has exited! "
|
|
"You should restart Kodi now. "
|
|
"Please report this on the forum."))
|
|
raise
|
|
|
|
def run_internal(self):
|
|
|
|
startupComplete = False
|
|
self.views = []
|
|
count = 0
|
|
|
|
self.logMsg("---===### Starting LibrarySync ###===---", 0)
|
|
while not self.threadStopped():
|
|
|
|
# In the event the server goes offline
|
|
while self.threadSuspended():
|
|
# Set in service.py
|
|
if self.threadStopped():
|
|
# Abort was requested while waiting. We should exit
|
|
return
|
|
xbmc.sleep(3000)
|
|
|
|
if (utils.window('emby_dbCheck') != "true" and
|
|
self.installSyncDone):
|
|
# Verify the validity of the database
|
|
currentVersion = utils.settings('dbCreatedWithVersion')
|
|
minVersion = utils.window('emby_minDBVersion')
|
|
uptoDate = self.compareDBVersion(currentVersion, minVersion)
|
|
|
|
if not uptoDate:
|
|
self.logMsg(
|
|
"Db version out of date: %s minimum version required: %s"
|
|
% (currentVersion, minVersion), 0)
|
|
|
|
resp = xbmcgui.Dialog().yesno(
|
|
heading="Db Version",
|
|
line1=("Detected the database needs to be recreated "
|
|
"for this version of " + self.addonName +
|
|
"Proceed?"))
|
|
if not resp:
|
|
self.logMsg("Db version out of date! USER IGNORED!", 0)
|
|
xbmcgui.Dialog().ok(
|
|
heading=self.addonName,
|
|
line1=(self.addonName + " may not work correctly "
|
|
"until the database is reset."))
|
|
else:
|
|
utils.reset()
|
|
|
|
utils.window('emby_dbCheck', value="true")
|
|
|
|
if not startupComplete:
|
|
# Also runs when installed first
|
|
# Verify the video database can be found
|
|
videoDb = utils.getKodiVideoDBPath()
|
|
if not xbmcvfs.exists(videoDb):
|
|
# Database does not exists
|
|
self.logMsg(
|
|
"The current Kodi version is incompatible "
|
|
"with the" + self.addonName + " add-on. Please visit "
|
|
"https://github.com/croneter/PlexKodiConnect "
|
|
"to know which Kodi versions are supported.", 0)
|
|
|
|
xbmcgui.Dialog().ok(
|
|
heading=self.addonName,
|
|
line1=("Cancelling the database syncing process. "
|
|
"Current Kodi version: %s is unsupported. "
|
|
"Please verify your logs for more info."
|
|
% xbmc.getInfoLabel('System.BuildVersion')))
|
|
break
|
|
|
|
# Run start up sync
|
|
utils.window('emby_dbScan', value="true")
|
|
self.logMsg("Db version: %s" % utils.settings('dbCreatedWithVersion'), 0)
|
|
self.logMsg("SyncDatabase (started)", 1)
|
|
startTime = datetime.now()
|
|
librarySync = self.fullSync(manualrun=True)
|
|
elapsedTime = datetime.now() - startTime
|
|
self.logMsg(
|
|
"SyncDatabase (finished in: %s) %s"
|
|
% (str(elapsedTime).split('.')[0], librarySync), 1)
|
|
# Only try the initial sync once per kodi session regardless
|
|
# This will prevent an infinite loop in case something goes wrong.
|
|
startupComplete = True
|
|
utils.settings(
|
|
'SyncInstallRunDone', value="true")
|
|
utils.settings(
|
|
"dbCreatedWithVersion", self.clientInfo.getVersion())
|
|
self.installSyncDone = True
|
|
utils.window('emby_dbScan', value="false")
|
|
|
|
# Currently no db scan, so we can start a new scan
|
|
elif utils.window('emby_dbScan') != "true":
|
|
# Full scan was requested from somewhere else, e.g. userclient
|
|
if utils.window('plex_runLibScan') == "true":
|
|
self.logMsg('Full library scan requested, starting', 1)
|
|
utils.window('emby_dbScan', value="true")
|
|
utils.window('plex_runLibScan', value='false')
|
|
self.fullSync(manualrun=True)
|
|
utils.window('emby_dbScan', value="false")
|
|
count = 0
|
|
else:
|
|
# Run full lib scan approx every 10min
|
|
if count % 600 == 0:
|
|
utils.window('emby_dbScan', value="true")
|
|
self.logMsg('Running maintainViews() scan', 1)
|
|
self.fullSync(manualrun=True)
|
|
count = 0
|
|
utils.window('emby_dbScan', value="false")
|
|
# Update views / PMS libraries approx. every 5
|
|
elif count % 120 == 0:
|
|
self.logMsg('Running maintainViews() scan', 1)
|
|
utils.window('emby_dbScan', value="true")
|
|
self.maintainViews()
|
|
self.startSync()
|
|
# Run fast sync otherwise (ever second or so)
|
|
else:
|
|
self.startSync()
|
|
|
|
xbmc.sleep(1000)
|
|
count += 1
|
|
|
|
self.logMsg("###===--- LibrarySync Stopped ---===###", 0)
|
|
|
|
|
|
class ManualSync(LibrarySync):
|
|
|
|
|
|
def __init__(self):
|
|
|
|
LibrarySync.__init__(self)
|
|
self.fullSync(manualrun=True)
|
|
|
|
|
|
def movies(self, embycursor, kodicursor, pdialog):
|
|
# Get movies from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
movies = itemtypes.Movies(embycursor, kodicursor)
|
|
|
|
views = emby_db.getView_byType('movies')
|
|
views += emby_db.getView_byType('mixed')
|
|
self.logMsg("Media folders: %s" % views, 1)
|
|
|
|
# Pull the list of movies and boxsets in Kodi
|
|
try:
|
|
all_kodimovies = dict(emby_db.getChecksum('Movie'))
|
|
except ValueError:
|
|
all_kodimovies = {}
|
|
|
|
try:
|
|
all_kodisets = dict(emby_db.getChecksum('BoxSet'))
|
|
except ValueError:
|
|
all_kodisets = {}
|
|
|
|
all_embymoviesIds = set()
|
|
all_embyboxsetsIds = set()
|
|
updatelist = []
|
|
|
|
##### PROCESS MOVIES #####
|
|
for view in views:
|
|
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing movies from view: %s..." % viewName)
|
|
|
|
all_embymovies = emby.getMovies(viewId, basic=True, dialog=pdialog)
|
|
for embymovie in all_embymovies['Items']:
|
|
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
API = api.API(embymovie)
|
|
itemid = embymovie['Id']
|
|
all_embymoviesIds.add(itemid)
|
|
|
|
|
|
if all_kodimovies.get(itemid) != API.getChecksum():
|
|
# Only update if movie is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
|
|
self.logMsg("Movies to update for %s: %s" % (viewName, updatelist), 1)
|
|
embymovies = emby.getFullItems(updatelist)
|
|
total = len(updatelist)
|
|
del updatelist[:]
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (viewName, total))
|
|
|
|
count = 0
|
|
for embymovie in embymovies:
|
|
# Process individual movies
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
title = embymovie['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
movies.add_update(embymovie, viewName, viewId)
|
|
|
|
##### PROCESS BOXSETS #####
|
|
|
|
boxsets = emby.getBoxset(dialog=pdialog)
|
|
embyboxsets = []
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing boxsets...")
|
|
|
|
for boxset in boxsets['Items']:
|
|
|
|
if self.threadStopped():
|
|
return False
|
|
|
|
# Boxset has no real userdata, so using etag to compare
|
|
checksum = boxset['Etag']
|
|
itemid = boxset['Id']
|
|
all_embyboxsetsIds.add(itemid)
|
|
|
|
if all_kodisets.get(itemid) != checksum:
|
|
# Only update if boxset is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
embyboxsets.append(boxset)
|
|
|
|
self.logMsg("Boxsets to update: %s" % updatelist, 1)
|
|
total = len(updatelist)
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing Boxsets / %s items" % total)
|
|
|
|
count = 0
|
|
for boxset in embyboxsets:
|
|
# Process individual boxset
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
title = boxset['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
movies.add_updateBoxset(boxset)
|
|
|
|
##### PROCESS DELETES #####
|
|
|
|
for kodimovie in all_kodimovies:
|
|
if kodimovie not in all_embymoviesIds:
|
|
movies.remove(kodimovie)
|
|
else:
|
|
self.logMsg("Movies compare finished.", 1)
|
|
|
|
for boxset in all_kodisets:
|
|
if boxset not in all_embyboxsetsIds:
|
|
movies.remove(boxset)
|
|
else:
|
|
self.logMsg("Boxsets compare finished.", 1)
|
|
|
|
return True
|
|
|
|
def musicvideos(self, embycursor, kodicursor, pdialog):
|
|
# Get musicvideos from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
mvideos = itemtypes.MusicVideos(embycursor, kodicursor)
|
|
|
|
views = emby_db.getView_byType('musicvideos')
|
|
self.logMsg("Media folders: %s" % views, 1)
|
|
|
|
# Pull the list of musicvideos in Kodi
|
|
try:
|
|
all_kodimvideos = dict(emby_db.getChecksum('MusicVideo'))
|
|
except ValueError:
|
|
all_kodimvideos = {}
|
|
|
|
all_embymvideosIds = set()
|
|
updatelist = []
|
|
|
|
for view in views:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing musicvideos from view: %s..." % viewName)
|
|
|
|
all_embymvideos = emby.getMusicVideos(viewId, basic=True, dialog=pdialog)
|
|
for embymvideo in all_embymvideos['Items']:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
API = api.API(embymvideo)
|
|
itemid = embymvideo['Id']
|
|
all_embymvideosIds.add(itemid)
|
|
|
|
|
|
if all_kodimvideos.get(itemid) != API.getChecksum():
|
|
# Only update if musicvideo is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
|
|
self.logMsg("MusicVideos to update for %s: %s" % (viewName, updatelist), 1)
|
|
embymvideos = emby.getFullItems(updatelist)
|
|
total = len(updatelist)
|
|
del updatelist[:]
|
|
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (viewName, total))
|
|
|
|
count = 0
|
|
for embymvideo in embymvideos:
|
|
# Process individual musicvideo
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
title = embymvideo['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
mvideos.add_update(embymvideo, viewName, viewId)
|
|
|
|
##### PROCESS DELETES #####
|
|
|
|
for kodimvideo in all_kodimvideos:
|
|
if kodimvideo not in all_embymvideosIds:
|
|
mvideos.remove(kodimvideo)
|
|
else:
|
|
self.logMsg("MusicVideos compare finished.", 1)
|
|
|
|
return True
|
|
|
|
def tvshows(self, embycursor, kodicursor, pdialog):
|
|
# Get shows from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
tvshows = itemtypes.TVShows(embycursor, kodicursor)
|
|
|
|
views = emby_db.getView_byType('tvshows')
|
|
views += emby_db.getView_byType('mixed')
|
|
self.logMsg("Media folders: %s" % views, 1)
|
|
|
|
# Pull the list of tvshows and episodes in Kodi
|
|
try:
|
|
all_koditvshows = dict(emby_db.getChecksum('Series'))
|
|
except ValueError:
|
|
all_koditvshows = {}
|
|
|
|
try:
|
|
all_kodiepisodes = dict(emby_db.getChecksum('Episode'))
|
|
except ValueError:
|
|
all_kodiepisodes = {}
|
|
|
|
all_embytvshowsIds = set()
|
|
all_embyepisodesIds = set()
|
|
updatelist = []
|
|
|
|
|
|
for view in views:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
# Get items per view
|
|
viewId = view['id']
|
|
viewName = view['name']
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing tvshows from view: %s..." % viewName)
|
|
|
|
all_embytvshows = emby.getShows(viewId, basic=True, dialog=pdialog)
|
|
for embytvshow in all_embytvshows['Items']:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
API = api.API(embytvshow)
|
|
itemid = embytvshow['Id']
|
|
all_embytvshowsIds.add(itemid)
|
|
|
|
|
|
if all_koditvshows.get(itemid) != API.getChecksum():
|
|
# Only update if movie is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
|
|
self.logMsg("TVShows to update for %s: %s" % (viewName, updatelist), 1)
|
|
embytvshows = emby.getFullItems(updatelist)
|
|
total = len(updatelist)
|
|
del updatelist[:]
|
|
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (viewName, total))
|
|
|
|
count = 0
|
|
for embytvshow in embytvshows:
|
|
# Process individual show
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
itemid = embytvshow['Id']
|
|
title = embytvshow['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
tvshows.add_update(embytvshow, viewName, viewId)
|
|
|
|
else:
|
|
# Get all episodes in view
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing episodes from view: %s..." % viewName)
|
|
|
|
all_embyepisodes = emby.getEpisodes(viewId, basic=True, dialog=pdialog)
|
|
for embyepisode in all_embyepisodes['Items']:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
API = api.API(embyepisode)
|
|
itemid = embyepisode['Id']
|
|
all_embyepisodesIds.add(itemid)
|
|
|
|
if all_kodiepisodes.get(itemid) != API.getChecksum():
|
|
# Only update if movie is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
|
|
self.logMsg("Episodes to update for %s: %s" % (viewName, updatelist), 1)
|
|
embyepisodes = emby.getFullItems(updatelist)
|
|
total = len(updatelist)
|
|
del updatelist[:]
|
|
|
|
count = 0
|
|
for episode in embyepisodes:
|
|
|
|
# Process individual episode
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
title = episode['SeriesName']
|
|
episodetitle = episode['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message="%s - %s" % (title, episodetitle))
|
|
count += 1
|
|
tvshows.add_updateEpisode(episode)
|
|
|
|
##### PROCESS DELETES #####
|
|
|
|
for koditvshow in all_koditvshows:
|
|
if koditvshow not in all_embytvshowsIds:
|
|
tvshows.remove(koditvshow)
|
|
else:
|
|
self.logMsg("TVShows compare finished.", 1)
|
|
|
|
for kodiepisode in all_kodiepisodes:
|
|
if kodiepisode not in all_embyepisodesIds:
|
|
tvshows.remove(kodiepisode)
|
|
else:
|
|
self.logMsg("Episodes compare finished.", 1)
|
|
|
|
return True
|
|
|
|
def music(self, embycursor, kodicursor, pdialog):
|
|
# Get music from emby
|
|
emby = self.emby
|
|
emby_db = embydb.Embydb_Functions(embycursor)
|
|
music = itemtypes.Music(embycursor, kodicursor)
|
|
|
|
# Pull the list of artists, albums, songs
|
|
try:
|
|
all_kodiartists = dict(emby_db.getChecksum('MusicArtist'))
|
|
except ValueError:
|
|
all_kodiartists = {}
|
|
|
|
try:
|
|
all_kodialbums = dict(emby_db.getChecksum('MusicAlbum'))
|
|
except ValueError:
|
|
all_kodialbums = {}
|
|
|
|
try:
|
|
all_kodisongs = dict(emby_db.getChecksum('Audio'))
|
|
except ValueError:
|
|
all_kodisongs = {}
|
|
|
|
all_embyartistsIds = set()
|
|
all_embyalbumsIds = set()
|
|
all_embysongsIds = set()
|
|
updatelist = []
|
|
|
|
process = {
|
|
|
|
'artists': [emby.getArtists, music.add_updateArtist],
|
|
'albums': [emby.getAlbums, music.add_updateAlbum],
|
|
'songs': [emby.getSongs, music.add_updateSong]
|
|
}
|
|
types = ['artists', 'albums', 'songs']
|
|
for type in types:
|
|
|
|
if pdialog:
|
|
pdialog.update(
|
|
heading="Emby for Kodi",
|
|
message="Comparing %s..." % type)
|
|
|
|
if type != "artists":
|
|
all_embyitems = process[type][0](basic=True, dialog=pdialog)
|
|
else:
|
|
all_embyitems = process[type][0](dialog=pdialog)
|
|
for embyitem in all_embyitems['Items']:
|
|
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
API = api.API(embyitem)
|
|
itemid = embyitem['Id']
|
|
if type == "artists":
|
|
all_embyartistsIds.add(itemid)
|
|
if all_kodiartists.get(itemid) != API.getChecksum():
|
|
# Only update if artist is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
elif type == "albums":
|
|
all_embyalbumsIds.add(itemid)
|
|
if all_kodialbums.get(itemid) != API.getChecksum():
|
|
# Only update if album is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
else:
|
|
all_embysongsIds.add(itemid)
|
|
if all_kodisongs.get(itemid) != API.getChecksum():
|
|
# Only update if songs is not in Kodi or checksum is different
|
|
updatelist.append(itemid)
|
|
|
|
self.logMsg("%s to update: %s" % (type, updatelist), 1)
|
|
embyitems = emby.getFullItems(updatelist)
|
|
total = len(updatelist)
|
|
del updatelist[:]
|
|
|
|
if pdialog:
|
|
pdialog.update(heading="Processing %s / %s items" % (type, total))
|
|
|
|
count = 0
|
|
for embyitem in embyitems:
|
|
# Process individual item
|
|
if self.shouldStop():
|
|
return False
|
|
|
|
title = embyitem['Name']
|
|
if pdialog:
|
|
percentage = int((float(count) / float(total))*100)
|
|
pdialog.update(percentage, message=title)
|
|
count += 1
|
|
|
|
process[type][1](embyitem)
|
|
|
|
##### PROCESS DELETES #####
|
|
|
|
for kodiartist in all_kodiartists:
|
|
if kodiartist not in all_embyartistsIds and all_kodiartists[kodiartist] is not None:
|
|
music.remove(kodiartist)
|
|
else:
|
|
self.logMsg("Artist compare finished.", 1)
|
|
|
|
for kodialbum in all_kodialbums:
|
|
if kodialbum not in all_embyalbumsIds:
|
|
music.remove(kodialbum)
|
|
else:
|
|
self.logMsg("Albums compare finished.", 1)
|
|
|
|
for kodisong in all_kodisongs:
|
|
if kodisong not in all_embysongsIds:
|
|
music.remove(kodisong)
|
|
else:
|
|
self.logMsg("Songs compare finished.", 1)
|
|
|
|
return True |