mirror of
https://github.com/jellyfin/jellyfin-kodi.git
synced 2025-10-16 21:56:48 +00:00
394a67fe33
Also update checksum for boxsets to regular checksum. Seems userdata was added for them. This will cause all boxsets to update once.
483 lines
17 KiB
Python
483 lines
17 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
##################################################################################################
|
|
|
|
import logging
|
|
import urllib
|
|
|
|
import api
|
|
import embydb_functions as embydb
|
|
import _kodi_movies
|
|
from _common import Items
|
|
from utils import window, settings, language as lang, catch_except
|
|
|
|
##################################################################################################
|
|
|
|
log = logging.getLogger("EMBY."+__name__)
|
|
|
|
##################################################################################################
|
|
|
|
|
|
class Movies(Items):
|
|
|
|
|
|
def __init__(self, embycursor, kodicursor, pdialog=None):
|
|
|
|
self.embycursor = embycursor
|
|
self.emby_db = embydb.Embydb_Functions(self.embycursor)
|
|
self.kodicursor = kodicursor
|
|
self.kodi_db = _kodi_movies.KodiMovies(self.kodicursor)
|
|
self.pdialog = pdialog
|
|
|
|
self.new_time = int(settings('newvideotime'))*1000
|
|
|
|
Items.__init__(self)
|
|
|
|
def _get_func(self, item_type, action):
|
|
|
|
if item_type == "Movie":
|
|
actions = {
|
|
'added': self.added,
|
|
'update': self.add_update,
|
|
'userdata': self.updateUserdata,
|
|
'remove': self.remove
|
|
}
|
|
elif item_type == "BoxSet":
|
|
actions = {
|
|
'added': self.add_updateBoxset,
|
|
'update': self.add_updateBoxset,
|
|
'remove': self.remove
|
|
}
|
|
else:
|
|
log.info("Unsupported item_type: %s", item_type)
|
|
actions = {}
|
|
|
|
return actions.get(action)
|
|
|
|
def compare_all(self):
|
|
# Pull the list of movies and boxsets in Kodi
|
|
views = self.emby_db.getView_byType('movies')
|
|
views += self.emby_db.getView_byType('mixed')
|
|
log.info("Media folders: %s", views)
|
|
|
|
# Process movies
|
|
for view in views:
|
|
|
|
if self.should_stop():
|
|
return False
|
|
|
|
if not self.compare_movies(view):
|
|
return False
|
|
|
|
# Process boxsets
|
|
if not self.compare_boxsets():
|
|
return False
|
|
|
|
return True
|
|
|
|
def compare_movies(self, view):
|
|
|
|
view_id = view['id']
|
|
view_name = view['name']
|
|
|
|
if self.pdialog:
|
|
self.pdialog.update(heading=lang(29999), message="%s %s..." % (lang(33026), view_name))
|
|
|
|
movies = dict(self.emby_db.get_checksum_by_view('Movie', view_id))
|
|
emby_movies = self.emby.getMovies(view_id, basic=True, dialog=self.pdialog)
|
|
|
|
update_list = self.compare_checksum(emby_movies['Items'], movies)
|
|
log.info("Movies to update for %s: %s", view_name, update_list)
|
|
emby_movies = self.emby.getFullItems(update_list)
|
|
total = len(update_list)
|
|
|
|
if self.pdialog:
|
|
self.pdialog.update(heading="Processing %s / %s items" % (view_name, total))
|
|
|
|
# Process additions and updates
|
|
if emby_movies:
|
|
self.added(emby_movies, total, view)
|
|
# Process deletes
|
|
if movies:
|
|
self.remove_all("Movie", movies.items())
|
|
|
|
return True
|
|
|
|
def compare_boxsets(self):
|
|
|
|
if self.pdialog:
|
|
self.pdialog.update(heading=lang(29999), message=lang(33027))
|
|
|
|
boxsets = dict(self.emby_db.get_checksum('BoxSet'))
|
|
emby_boxsets = self.emby.getBoxset(dialog=self.pdialog)
|
|
|
|
update_list = self.compare_checksum(emby_boxsets['Items'], boxsets)
|
|
log.info("Boxsets to update: %s", update_list)
|
|
emby_boxsets = self.emby.getFullItems(update_list)
|
|
total = len(update_list)
|
|
|
|
if self.pdialog:
|
|
self.pdialog.update(heading="Processing Boxsets / %s items" % total)
|
|
|
|
# Processing additions and updates
|
|
if emby_boxsets:
|
|
self.added_boxset(emby_boxsets, total)
|
|
# Processing removals
|
|
if boxsets:
|
|
self.remove_all("BoxSet", boxsets.items())
|
|
|
|
return True
|
|
|
|
def added(self, items, total=None, view=None):
|
|
|
|
for item in super(Movies, self).added(items, total):
|
|
if self.add_update(item, view):
|
|
self.content_pop(item.get('Name', "unknown"))
|
|
|
|
def added_boxset(self, items, total=None):
|
|
|
|
for item in super(Movies, self).added(items, total):
|
|
self.add_updateBoxset(item)
|
|
|
|
@catch_except()
|
|
def add_update(self, item, view=None):
|
|
# Process single movie
|
|
emby_db = self.emby_db
|
|
artwork = self.artwork
|
|
API = api.API(item)
|
|
|
|
# If the item already exist in the local Kodi DB we'll perform a full item update
|
|
# If the item doesn't exist, we'll add it to the database
|
|
update_item = True
|
|
itemid = item['Id']
|
|
emby_dbitem = emby_db.getItem_byId(itemid)
|
|
try:
|
|
movieid = emby_dbitem[0]
|
|
fileid = emby_dbitem[1]
|
|
pathid = emby_dbitem[2]
|
|
log.info("movieid: %s fileid: %s pathid: %s", movieid, fileid, pathid)
|
|
|
|
except TypeError:
|
|
update_item = False
|
|
log.debug("movieid: %s not found", itemid)
|
|
# movieid
|
|
movieid = self.kodi_db.create_entry()
|
|
|
|
else:
|
|
if self.kodi_db.get_movie(movieid) is None:
|
|
# item is not found, let's recreate it.
|
|
update_item = False
|
|
log.info("movieid: %s missing from Kodi, repairing the entry", movieid)
|
|
|
|
if not view:
|
|
# Get view tag from emby
|
|
viewtag, viewid, mediatype = self.emby.getView_embyId(itemid)
|
|
log.debug("View tag found: %s", viewtag)
|
|
else:
|
|
viewtag = view['name']
|
|
viewid = view['id']
|
|
|
|
# fileId information
|
|
checksum = API.get_checksum()
|
|
dateadded = API.get_date_created()
|
|
userdata = API.get_userdata()
|
|
playcount = userdata['PlayCount']
|
|
dateplayed = userdata['LastPlayedDate']
|
|
|
|
# item details
|
|
people = API.get_people()
|
|
writer = " / ".join(people['Writer'])
|
|
director = " / ".join(people['Director'])
|
|
genres = item['Genres']
|
|
title = item['Name']
|
|
plot = API.get_overview()
|
|
shortplot = item.get('ShortOverview')
|
|
tagline = API.get_tagline()
|
|
votecount = item.get('VoteCount')
|
|
rating = item.get('CommunityRating')
|
|
year = item.get('ProductionYear')
|
|
imdb = API.get_provider('Imdb')
|
|
sorttitle = item['SortName']
|
|
runtime = API.get_runtime()
|
|
mpaa = API.get_mpaa()
|
|
genre = " / ".join(genres)
|
|
country = API.get_country()
|
|
studios = API.get_studios()
|
|
try:
|
|
studio = studios[0]
|
|
except IndexError:
|
|
studio = None
|
|
|
|
if item.get('LocalTrailerCount'):
|
|
# There's a local trailer
|
|
url = (
|
|
"{server}/emby/Users/{UserId}/Items/%s/LocalTrailers?format=json"
|
|
% itemid
|
|
)
|
|
result = self.do_url(url)
|
|
try:
|
|
trailer = "plugin://plugin.video.emby/trailer/?id=%s&mode=play" % result[0]['Id']
|
|
except IndexError:
|
|
log.info("Failed to process local trailer.")
|
|
trailer = None
|
|
else:
|
|
# Try to get the youtube trailer
|
|
try:
|
|
trailer = item['RemoteTrailers'][0]['Url']
|
|
except (KeyError, IndexError):
|
|
trailer = None
|
|
else:
|
|
try:
|
|
trailer_id = trailer.rsplit('=', 1)[1]
|
|
except IndexError:
|
|
log.info("Failed to process trailer: %s", trailer)
|
|
trailer = None
|
|
else:
|
|
trailer = "plugin://plugin.video.youtube/play/?video_id=%s" % trailer_id
|
|
|
|
|
|
##### GET THE FILE AND PATH #####
|
|
playurl = API.get_file_path()
|
|
|
|
if "\\" in playurl:
|
|
# Local path
|
|
filename = playurl.rsplit("\\", 1)[1]
|
|
else: # Network share
|
|
filename = playurl.rsplit("/", 1)[1]
|
|
|
|
if self.direct_path:
|
|
# Direct paths is set the Kodi way
|
|
if not self.path_validation(playurl):
|
|
return False
|
|
|
|
path = playurl.replace(filename, "")
|
|
window('emby_pathverified', value="true")
|
|
else:
|
|
# Set plugin path and media flags using real filename
|
|
path = "plugin://plugin.video.emby.movies/"
|
|
params = {
|
|
|
|
'filename': filename.encode('utf-8'),
|
|
'id': itemid,
|
|
'dbid': movieid,
|
|
'mode': "play"
|
|
}
|
|
filename = "%s?%s" % (path, urllib.urlencode(params))
|
|
|
|
|
|
##### UPDATE THE MOVIE #####
|
|
if update_item:
|
|
log.info("UPDATE movie itemid: %s - Title: %s", itemid, title)
|
|
|
|
# Update the movie entry
|
|
if self.kodi_version > 16:
|
|
self.kodi_db.update_movie_17(title, plot, shortplot, tagline, votecount, rating,
|
|
writer, year, imdb, sorttitle, runtime, mpaa, genre,
|
|
director, title, studio, trailer, country, year,
|
|
movieid)
|
|
else:
|
|
self.kodi_db.update_movie(title, plot, shortplot, tagline, votecount, rating,
|
|
writer, year, imdb, sorttitle, runtime, mpaa, genre,
|
|
director, title, studio, trailer, country, movieid)
|
|
|
|
# Update the checksum in emby table
|
|
emby_db.updateReference(itemid, checksum)
|
|
|
|
##### OR ADD THE MOVIE #####
|
|
else:
|
|
log.info("ADD movie itemid: %s - Title: %s", itemid, title)
|
|
|
|
# Add path
|
|
pathid = self.kodi_db.add_path(path)
|
|
# Add the file
|
|
fileid = self.kodi_db.add_file(filename, pathid)
|
|
|
|
# Create the movie entry
|
|
if self.kodi_version > 16:
|
|
self.kodi_db.add_movie_17(movieid, fileid, title, plot, shortplot, tagline,
|
|
votecount, rating, writer, year, imdb, sorttitle,
|
|
runtime, mpaa, genre, director, title, studio, trailer,
|
|
country, year)
|
|
else:
|
|
self.kodi_db.add_movie(movieid, fileid, title, plot, shortplot, tagline,
|
|
votecount, rating, writer, year, imdb, sorttitle,
|
|
runtime, mpaa, genre, director, title, studio, trailer,
|
|
country)
|
|
|
|
# Create the reference in emby table
|
|
emby_db.addReference(itemid, movieid, "Movie", "movie", fileid, pathid, None,
|
|
checksum, viewid)
|
|
|
|
# Update the path
|
|
self.kodi_db.update_path(pathid, path, "movies", "metadata.local")
|
|
# Update the file
|
|
self.kodi_db.update_file(fileid, filename, pathid, dateadded)
|
|
|
|
# Process countries
|
|
if 'ProductionLocations' in item:
|
|
self.kodi_db.add_countries(movieid, item['ProductionLocations'])
|
|
# Process cast
|
|
people = artwork.get_people_artwork(item['People'])
|
|
self.kodi_db.add_people(movieid, people, "movie")
|
|
# Process genres
|
|
self.kodi_db.add_genres(movieid, genres, "movie")
|
|
# Process artwork
|
|
artwork.add_artwork(artwork.get_all_artwork(item), movieid, "movie", self.kodicursor)
|
|
# Process stream details
|
|
streams = API.get_media_streams()
|
|
self.kodi_db.add_streams(fileid, streams, runtime)
|
|
# Process studios
|
|
self.kodi_db.add_studios(movieid, studios, "movie")
|
|
# Process tags: view, emby tags
|
|
tags = [viewtag]
|
|
tags.extend(item['Tags'])
|
|
if userdata['Favorite']:
|
|
tags.append("Favorite movies")
|
|
log.info("Applied tags: %s", tags)
|
|
self.kodi_db.add_tags(movieid, tags, "movie")
|
|
# Process playstates
|
|
resume = API.adjust_resume(userdata['Resume'])
|
|
total = round(float(runtime), 6)
|
|
self.kodi_db.add_playstate(fileid, resume, total, playcount, dateplayed)
|
|
|
|
return True
|
|
|
|
def add_updateBoxset(self, boxset):
|
|
|
|
emby = self.emby
|
|
emby_db = self.emby_db
|
|
artwork = self.artwork
|
|
API = api.API(boxset)
|
|
|
|
boxsetid = boxset['Id']
|
|
title = boxset['Name']
|
|
checksum = API.get_checksum()
|
|
emby_dbitem = emby_db.getItem_byId(boxsetid)
|
|
try:
|
|
setid = emby_dbitem[0]
|
|
|
|
except TypeError:
|
|
setid = self.kodi_db.add_boxset(title)
|
|
|
|
# Process artwork
|
|
artwork.add_artwork(artwork.get_all_artwork(boxset), setid, "set", self.kodicursor)
|
|
|
|
# Process movies inside boxset
|
|
current_movies = emby_db.getItemId_byParentId(setid, "movie")
|
|
process = []
|
|
try:
|
|
# Try to convert tuple to dictionary
|
|
current = dict(current_movies)
|
|
except ValueError:
|
|
current = {}
|
|
|
|
# Sort current titles
|
|
for current_movie in current:
|
|
process.append(current_movie)
|
|
|
|
# New list to compare
|
|
for movie in emby.getMovies_byBoxset(boxsetid)['Items']:
|
|
|
|
itemid = movie['Id']
|
|
|
|
if not current.get(itemid):
|
|
# Assign boxset to movie
|
|
emby_dbitem = emby_db.getItem_byId(itemid)
|
|
try:
|
|
movieid = emby_dbitem[0]
|
|
except TypeError:
|
|
log.info("Failed to add: %s to boxset", movie['Name'])
|
|
continue
|
|
|
|
log.info("New addition to boxset %s: %s", title, movie['Name'])
|
|
self.kodi_db.set_boxset(setid, movieid)
|
|
# Update emby reference
|
|
emby_db.updateParentId(itemid, setid)
|
|
else:
|
|
# Remove from process, because the item still belongs
|
|
process.remove(itemid)
|
|
|
|
# Process removals from boxset
|
|
for movie in process:
|
|
movieid = current[movie]
|
|
log.info("Remove from boxset %s: %s", title, movieid)
|
|
self.kodi_db.remove_from_boxset(movieid)
|
|
# Update emby reference
|
|
emby_db.updateParentId(movie, None)
|
|
|
|
# Update the reference in the emby table
|
|
emby_db.addReference(boxsetid, setid, "BoxSet", mediatype="set", checksum=checksum)
|
|
|
|
def updateUserdata(self, item):
|
|
# This updates: Favorite, LastPlayedDate, Playcount, PlaybackPositionTicks
|
|
# Poster with progress bar
|
|
emby_db = self.emby_db
|
|
API = api.API(item)
|
|
|
|
# Get emby information
|
|
itemid = item['Id']
|
|
checksum = API.get_checksum()
|
|
userdata = API.get_userdata()
|
|
runtime = API.get_runtime()
|
|
|
|
# Get Kodi information
|
|
emby_dbitem = emby_db.getItem_byId(itemid)
|
|
try:
|
|
movieid = emby_dbitem[0]
|
|
fileid = emby_dbitem[1]
|
|
log.info("Update playstate for movie: %s fileid: %s", item['Name'], fileid)
|
|
except TypeError:
|
|
return
|
|
|
|
# Process favorite tags
|
|
if userdata['Favorite']:
|
|
self.kodi_db.get_tag(movieid, "Favorite movies", "movie")
|
|
else:
|
|
self.kodi_db.remove_tag(movieid, "Favorite movies", "movie")
|
|
|
|
# Process playstates
|
|
playcount = userdata['PlayCount']
|
|
dateplayed = userdata['LastPlayedDate']
|
|
resume = API.adjust_resume(userdata['Resume'])
|
|
total = round(float(runtime), 6)
|
|
|
|
log.debug("%s New resume point: %s", itemid, resume)
|
|
|
|
self.kodi_db.add_playstate(fileid, resume, total, playcount, dateplayed)
|
|
emby_db.updateReference(itemid, checksum)
|
|
|
|
def remove(self, itemid):
|
|
# Remove movieid, fileid, emby reference
|
|
emby_db = self.emby_db
|
|
artwork = self.artwork
|
|
|
|
emby_dbitem = emby_db.getItem_byId(itemid)
|
|
try:
|
|
kodiid = emby_dbitem[0]
|
|
fileid = emby_dbitem[1]
|
|
mediatype = emby_dbitem[4]
|
|
log.info("Removing %sid: %s fileid: %s", mediatype, kodiid, fileid)
|
|
except TypeError:
|
|
return
|
|
|
|
# Remove the emby reference
|
|
emby_db.removeItem(itemid)
|
|
# Remove artwork
|
|
artwork.delete_artwork(kodiid, mediatype, self.kodicursor)
|
|
|
|
if mediatype == "movie":
|
|
self.kodi_db.remove_movie(kodiid, fileid)
|
|
|
|
elif mediatype == "set":
|
|
# Delete kodi boxset
|
|
boxset_movies = emby_db.getItem_byParentId(kodiid, "movie")
|
|
for movie in boxset_movies:
|
|
embyid = movie[0]
|
|
movieid = movie[1]
|
|
self.kodi_db.remove_from_boxset(movieid)
|
|
# Update emby reference
|
|
emby_db.updateParentId(embyid, None)
|
|
|
|
self.kodi_db.remove_boxset(kodiid)
|
|
|
|
log.info("Deleted %s %s from kodi database", mediatype, itemid)
|