PlexKodiConnect/resources/lib/WriteKodiVideoDB.py

1338 lines
64 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
2015-03-18 06:02:42 +11:00
#################################################################################################
# WriteKodiVideoDB
2015-03-18 06:02:42 +11:00
#################################################################################################
import sqlite3
2015-07-24 17:52:36 +10:00
from ntpath import dirname, split as ntsplit
2015-03-18 06:02:42 +11:00
import xbmc
import xbmcgui
import xbmcaddon
from ClientInformation import ClientInformation
import Utils as utils
from API import API
2015-03-18 06:02:42 +11:00
from DownloadUtils import DownloadUtils
from PlayUtils import PlayUtils
2015-03-18 07:18:27 +11:00
from ReadKodiDB import ReadKodiDB
2015-03-21 02:58:46 +11:00
from ReadEmbyDB import ReadEmbyDB
from TextureCache import TextureCache
2015-03-18 06:02:42 +11:00
class WriteKodiVideoDB():
textureCache = TextureCache()
doUtils = DownloadUtils()
kodiversion = int(xbmc.getInfoLabel("System.BuildVersion")[:2])
addonName = ClientInformation().getAddonName()
def __init__(self):
username = utils.window('currUser')
self.userid = utils.window('userId%s' % username)
self.server = utils.window('server%s' % username)
self.directpath = utils.settings('useDirectPaths') == "true"
def logMsg(self, msg, lvl = 1):
className = self.__class__.__name__
utils.logMsg("%s %s" % (self.addonName, className), msg, int(lvl))
def updatePlayCountFromKodi(self, id, type, playcount = 0):
# When user marks item watched from kodi interface update this in Emby
# Erase resume point when user marks watched/unwatched to follow Emby behavior
doUtils = self.doUtils
self.logMsg("updatePlayCountFromKodi Called", 2)
2015-05-02 11:47:05 +10:00
connection = utils.KodiSQL()
cursor = connection.cursor()
cursor.execute("SELECT emby_id FROM emby WHERE media_type = ? AND kodi_id = ?", (type, id,))
2015-03-18 06:02:42 +11:00
try: # Find associated Kodi Id to Emby Id
emby_id = cursor.fetchone()[0]
except:
# Could not find the Emby Id
self.logMsg("Emby Id not found.", 2)
else:
# Stop from manually marking as watched unwatched, with actual playback.
# Window property is set in Player.py
if utils.window('SkipWatched%s' % emby_id) == "true":
utils.window('SkipWatched%s' % emby_id, clear=True)
else:
# Found the Emby Id, let Emby server know of new playcount
watchedurl = "{server}/mediabrowser/Users/{UserId}/PlayedItems/%s" % emby_id
if playcount != 0:
doUtils.downloadUrl(watchedurl, type = "POST")
self.logMsg("Mark as watched for Id: %s, playcount: %s." % (emby_id, playcount), 1)
else:
doUtils.downloadUrl(watchedurl, type = "DELETE")
self.logMsg("Mark as unwatched for Id: %s, playcount: %s." % (emby_id, playcount), 1)
# Erase any resume point associated
self.setKodiResumePoint(id, 0, 0, cursor, playcount)
finally:
cursor.close
2015-03-18 06:02:42 +11:00
def addOrUpdateMovieToKodiLibrary(self, embyId, connection, cursor, viewTag):
2015-07-26 22:09:09 +10:00
MBitem = ReadEmbyDB().getFullItem(embyId)
2015-05-13 02:16:30 +10:00
if not MBitem:
2015-07-26 22:09:09 +10:00
self.logMsg("ADD movie to Kodi library FAILED, Item %s not found on server!" % embyId, 1)
2015-05-13 02:16:30 +10:00
return
# 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
cursor.execute("SELECT kodi_id FROM emby WHERE emby_id = ?", (embyId,))
try:
movieid = cursor.fetchone()[0]
except:
movieid = None
2015-07-26 22:09:09 +10:00
self.logMsg("Movie Id: %s not found." % embyId, 1)
2015-07-26 22:09:09 +10:00
timeInfo = API().getTimeInfo(MBitem)
userData = API().getUserData(MBitem)
people = API().getPeople(MBitem)
genres = MBitem.get('Genres')
studios = API().getStudios(MBitem)
#### The movie details ####
playcount = userData.get('PlayCount')
dateplayed = userData.get("LastPlayedDate")
dateadded = API().getDateCreated(MBitem)
checksum = API().getChecksum(MBitem)
title = MBitem['Name']
plot = API().getOverview(MBitem)
shortplot = MBitem.get('ShortOverview')
tagline = API().getTagline(MBitem)
votecount = MBitem.get('VoteCount')
rating = MBitem.get('CommunityRating')
writer = " / ".join(people.get('Writer'))
year = MBitem.get('ProductionYear')
imdb = API().getProvider(MBitem, "imdb")
sorttitle = MBitem['SortName']
runtime = timeInfo.get('TotalTime')
mpaa = API().getMpaa(MBitem)
genre = " / ".join(genres)
director = " / ".join(people.get('Director'))
try:
studio = studios[0]
except IndexError:
studio = None
country = API().getCountry(MBitem)
try: # Verify if there's a local trailer
if MBitem.get('LocalTrailerCount'):
itemTrailerUrl = "{server}/mediabrowser/Users/{UserId}/Items/%s/LocalTrailers?format=json" % embyId
result = self.doUtils.downloadUrl(itemTrailerUrl)
trailerUrl = "plugin://plugin.video.emby/trailer/?id=%s&mode=play" % result[0]['Id']
# Or get youtube trailer
else:
trailerUrl = MBitem['RemoteTrailers'][0]['Url']
2015-06-17 15:35:07 +10:00
trailerId = trailerUrl.split('=')[1]
trailerUrl = "plugin://plugin.video.youtube/play/?video_id=%s" % trailerId
except:
trailerUrl = None
2015-07-26 22:09:09 +10:00
##### ADD OR UPDATE THE FILE AND PATH #####
##### NOTE THAT LASTPLAYED AND PLAYCOUNT ARE STORED AT THE FILE ENTRY #####
playurl = PlayUtils().directPlay(MBitem)
realfile = ""
realpath = ""
2015-07-26 22:09:09 +10:00
if self.directpath:
if playurl == False:
return
elif "\\" in playurl:
filename = playurl.rsplit("\\",1)[-1]
path = playurl.replace(filename, "")
elif "/" in playurl:
filename = playurl.rsplit("/",1)[-1]
path = playurl.replace(filename, "")
else:
self.logMsg("Invalid path: %s" % playurl, 1)
return
2015-07-26 22:09:09 +10:00
else: # Set plugin path and media flags using real filename
try:
if not "plugin://" in playurl:
realpath, realfile = ntsplit(playurl)
if "/" in playurl:
realpath = realpath + "/"
else:
realpath = realpath + "\\"
except:
pass
2015-07-26 22:09:09 +10:00
filename = "plugin://plugin.video.emby/movies/%s/?filename=%s&id=%s&mode=play" % (embyId, realfile, embyId)
2015-07-26 22:09:09 +10:00
path = "plugin://plugin.video.emby/movies/%s/" % embyId
##### UPDATE THE MOVIE #####
if movieid:
self.logMsg("UPDATE movie to Kodi Library, Id: %s - Title: %s" % (embyId, title), 1)
#get the file ID
cursor.execute("SELECT idFile as fileid FROM movie WHERE idMovie = ?", (movieid,))
fileid = cursor.fetchone()[0]
#always update the filepath (fix for path change)
query = "UPDATE files SET strFilename = ?, dateAdded = ? WHERE idFile = ?"
cursor.execute(query, (filename, dateadded, fileid))
query = "UPDATE movie SET c00 = ?, c01 = ?, c02 = ?, c03 = ?, c04 = ?, c05 = ?, c06 = ?, c07 = ?, c09 = ?, c10 = ?, c11 = ?, c12 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, c21 = ? WHERE idMovie = ?"
cursor.execute(query, (title, plot, shortplot, tagline, votecount, rating, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailerUrl, country, movieid))
# Update the checksum in emby table and critic ratings
2015-09-28 07:06:22 +10:00
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
cursor.execute(query, (checksum, embyId))
##### OR ADD THE MOVIE #####
2015-05-02 22:02:31 +10:00
else:
self.logMsg("ADD movie to Kodi Library, Id: %s - Title: %s" % (embyId, title), 1)
# Validate the path in database
cursor.execute("SELECT idPath as pathid FROM path WHERE strPath = ?", (path,))
try:
pathid = cursor.fetchone()[0]
except:
# Path does not exist yet
cursor.execute("select coalesce(max(idPath),0) as pathid from path")
pathid = cursor.fetchone()[0] + 1
query = "INSERT into path(idPath, strPath, strContent, strScraper, noUpdate) values(?, ?, ?, ?, ?)"
cursor.execute(query, (pathid, path, "movies", "metadata.local", 1))
# Validate the file in database
if self.directpath:
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename = ? and idPath = ?", (filename, pathid,))
else:
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename LIKE ? and idPath = ?", (filename, embyId,))
try:
fileid = cursor.fetchone()[0]
except:
# File does not exist yet
cursor.execute("select coalesce(max(idFile),0) as fileid from files")
fileid = cursor.fetchone()[0] + 1
query = "INSERT INTO files(idFile, idPath, strFilename, playCount, lastPlayed, dateAdded) values(?, ?, ?, ?, ?, ?)"
cursor.execute(query, (fileid, pathid, filename, playcount, dateplayed, dateadded))
# Create the movie
cursor.execute("select coalesce(max(idMovie),0) as movieid from movie")
movieid = cursor.fetchone()[0] + 1
query = "INSERT INTO movie(idMovie, idFile, c00, c01, c02, c03, c04, c05, c06, c07, c09, c10, c11, c12, c14, c15, c16, c18, c19, c21) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (movieid, fileid, title, plot, shortplot, tagline, votecount, rating, writer, year, imdb, sorttitle, runtime, mpaa, genre, director, title, studio, trailerUrl, country))
# Create the reference in emby table
2015-09-28 07:06:22 +10:00
query = "INSERT INTO emby(emby_id, kodi_id, kodi_file_id, media_type, checksum) values(?, ?, ?, ?, ?)"
cursor.execute(query, (embyId, movieid, fileid, "movie", checksum))
2015-10-02 13:30:11 +10:00
# Add tags to item, view tag and emby tags
tags = [viewTag]
tags.extend(MBitem['Tags'])
if userData['Favorite']:
tags.append("Favorite movies")
self.AddTagsToMedia(movieid, tags, "movie", cursor)
# Update artwork
self.textureCache.addArtwork(API().getAllArtwork(MBitem), movieid, "movie", cursor)
# Update or insert actors
2015-07-26 22:09:09 +10:00
self.AddPeopleToMedia(movieid, MBitem.get('People'), "movie", connection, cursor)
# Update genres
self.AddGenresToMedia(movieid, genres, "movie", cursor)
# Update countries
self.AddCountriesToMedia(movieid, MBitem.get('ProductionLocations'), "movie", cursor)
2015-05-06 00:20:09 +10:00
# Update studios
2015-05-06 00:20:09 +10:00
self.AddStudiosToMedia(movieid, studios, "movie", cursor)
# Add streamdetails
2015-08-01 06:17:05 +10:00
self.AddStreamDetailsToMedia(API().getMediaStreams(MBitem), runtime ,fileid, cursor)
# Set resume point and round to 6th decimal
resume = round(float(timeInfo.get('ResumeTime')), 6)
total = round(float(timeInfo.get('TotalTime')), 6)
jumpback = int(utils.settings('resumeJumpBack'))
if resume > jumpback:
# To avoid negative bookmark
resume = resume - jumpback
self.setKodiResumePoint(fileid, resume, total, cursor, playcount, dateplayed, realpath, realfile)
2015-05-02 10:26:06 +10:00
2015-07-26 22:09:09 +10:00
def addOrUpdateMusicVideoToKodiLibrary( self, embyId ,connection, cursor):
WINDOW = xbmcgui.Window(10000)
username = WINDOW.getProperty('currUser')
userid = WINDOW.getProperty('userId%s' % username)
server = WINDOW.getProperty('server%s' % username)
downloadUtils = DownloadUtils()
2015-05-03 06:02:06 +10:00
MBitem = ReadEmbyDB().getFullItem(embyId)
2015-05-13 02:16:30 +10:00
if not MBitem:
2015-07-26 22:09:09 +10:00
utils.logMsg("ADD musicvideo to Kodi library FAILED", "Item %s not found on server!" %embyId)
2015-05-13 02:16:30 +10:00
return
2015-07-26 22:09:09 +10:00
2015-05-03 06:02:06 +10:00
# 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
2015-07-26 22:09:09 +10:00
cursor.execute("SELECT kodi_id FROM emby WHERE emby_id = ?",(MBitem["Id"],))
result = cursor.fetchone()
if result != None:
idMVideo = result[0]
else:
2015-05-03 06:02:06 +10:00
idMVideo = None
timeInfo = API().getTimeInfo(MBitem)
2015-07-26 22:09:09 +10:00
userData=API().getUserData(MBitem)
2015-05-03 06:02:06 +10:00
people = API().getPeople(MBitem)
2015-07-26 22:09:09 +10:00
#### The video details #########
runtime = timeInfo.get('TotalTime')
2015-07-26 22:09:09 +10:00
plot = utils.convertEncoding(API().getOverview(MBitem))
title = utils.convertEncoding(MBitem["Name"])
year = MBitem.get("ProductionYear")
genres = MBitem.get("Genres")
genre = " / ".join(genres)
2015-07-26 22:09:09 +10:00
studios = API().getStudios(MBitem)
studio = " / ".join(studios)
director = " / ".join(people.get("Director"))
artist = " / ".join(MBitem.get("Artists"))
album = MBitem.get("Album")
track = MBitem.get("Track")
dateplayed = userData.get("LastPlayedDate")
playcount = userData.get('PlayCount')
dateadded = API().getDateCreated(MBitem)
2015-05-03 06:02:06 +10:00
##### ADD OR UPDATE THE FILE AND PATH #####
##### NOTE THAT LASTPLAYED AND PLAYCOUNT ARE STORED AT THE FILE ENTRY #####
playurl = PlayUtils().directPlay(MBitem)
realfile = ""
realpath = ""
2015-05-03 06:02:06 +10:00
if self.directpath:
if playurl == False:
return
elif "\\" in playurl:
filename = playurl.rsplit("\\",1)[-1]
path = playurl.replace(filename, "")
elif "/" in playurl:
filename = playurl.rsplit("/",1)[-1]
path = playurl.replace(filename, "")
else:
self.logMsg("Invalid path: %s" % playurl, 1)
return
else: # Set plugin path and media flags using real filename
try:
if not "plugin://" in playurl:
realpath, realfile = ntsplit(playurl)
if "/" in playurl:
realpath = realpath + "/"
else:
realpath = realpath + "\\"
except:
pass
filename = "plugin://plugin.video.emby/musicvideos/%s/?filename=%s&id=%s&mode=play" % (MBitem["Id"], realfile, MBitem["Id"])
path = "plugin://plugin.video.emby/movies/%s/" % embyId
2015-07-26 22:09:09 +10:00
##### ADD THE VIDEO ############
if idMVideo == None:
utils.logMsg("ADD musicvideo to Kodi library","Id: %s - Title: %s" % (embyId, title))
#create the path
cursor.execute("SELECT idPath as pathid FROM path WHERE strPath = ?",(path,))
result = cursor.fetchone()
if result != None:
pathid = result[0]
else:
cursor.execute("select coalesce(max(idPath),0) as pathid from path")
pathid = cursor.fetchone()[0]
pathid = pathid + 1
pathsql = "insert into path(idPath, strPath, strContent, strScraper, noUpdate) values(?, ?, ?, ?, ?)"
cursor.execute(pathsql, (pathid,path,"musicvideos","metadata.local",1))
#create the file if not exists
if self.directpath:
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename = ? and idPath = ?", (filename, pathid,))
else:
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename LIKE ? and idPath = ?", (filename, embyId,))
result = cursor.fetchone()
if result != None:
fileid = result[0]
if result == None:
cursor.execute("select coalesce(max(idFile),0) as fileid from files")
fileid = cursor.fetchone()[0]
fileid = fileid + 1
pathsql="insert into files(idFile, idPath, strFilename, playCount, lastPlayed, dateAdded) values(?, ?, ?, ?, ?, ?)"
cursor.execute(pathsql, (fileid,pathid,filename,playcount,userData.get("LastPlayedDate"),dateadded))
2015-07-26 22:09:09 +10:00
#create the video
cursor.execute("select coalesce(max(idMVideo),0) as idMVideo from musicvideo")
2015-07-26 22:09:09 +10:00
idMVideo = cursor.fetchone()[0]
idMVideo = idMVideo + 1
pathsql="insert into musicvideo(idMVideo, idFile, c00, c04, c05, c06, c07, c08, c09, c10, c11, c12) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(pathsql, (idMVideo, fileid, title, runtime, director, studio, year, plot, album, artist, genre, track))
#create the reference in emby table
pathsql = "INSERT into emby(emby_id, kodi_id, media_type, checksum) values(?, ?, ?, ?)"
cursor.execute(pathsql, (MBitem["Id"], idMVideo, "musicvideo", API().getChecksum(MBitem)))
#### UPDATE THE VIDEO #####
else:
utils.logMsg("UPDATE musicvideo to Kodi library","Id: %s - Title: %s" % (embyId, title))
#get the file ID
cursor.execute("SELECT idFile as fileid FROM musicvideo WHERE idMVideo = ?", (idMVideo,))
fileid = cursor.fetchone()[0]
#always update the filepath (fix for path change)
query = "UPDATE files SET strFilename = ?, dateAdded = ? WHERE idFile = ?"
cursor.execute(query, (filename, dateadded, fileid))
2015-07-26 22:09:09 +10:00
pathsql="update musicvideo SET c00 = ?, c04 = ?, c05 = ?, c06 = ?, c07 = ?, c08 = ?, c09 = ?, c10 = ?, c11 = ?, c12 = ? WHERE idMVideo = ?"
cursor.execute(pathsql, (title, runtime, director, studio, year, plot, album, artist, genre, track, idMVideo))
#update the checksum in emby table
cursor.execute("UPDATE emby SET checksum = ? WHERE emby_id = ?", (API().getChecksum(MBitem),MBitem["Id"]))
2015-05-03 06:02:06 +10:00
2015-10-19 13:15:49 +11:00
# Add tags to item, view tag and emby tags
tags = MBitem['Tags']
self.AddTagsToMedia(idMVideo, tags, "musicvideo", cursor)
2015-07-26 22:09:09 +10:00
#update or insert actors
2015-11-10 12:59:45 +11:00
people = MBitem['People']
2015-10-24 12:00:34 +11:00
artists = MBitem['ArtistItems']
for artist in artists:
artist['Type'] = "Artist"
2015-11-10 12:59:45 +11:00
people.extend(artists)
self.AddPeopleToMedia(idMVideo,people,"musicvideo", connection, cursor)
# Update artwork
self.textureCache.addArtwork(API().getAllArtwork(MBitem), idMVideo, "musicvideo", cursor)
2015-05-03 06:02:06 +10:00
2015-07-26 22:09:09 +10:00
#update genres
2015-05-03 06:02:06 +10:00
self.AddGenresToMedia(idMVideo, genres, "musicvideo", cursor)
2015-07-26 22:09:09 +10:00
#update studios
2015-05-03 06:02:06 +10:00
self.AddStudiosToMedia(idMVideo, studios, "musicvideo", cursor)
2015-07-26 22:09:09 +10:00
#add streamdetails
2015-08-01 06:17:05 +10:00
self.AddStreamDetailsToMedia(API().getMediaStreams(MBitem), runtime ,fileid, cursor)
2015-07-26 22:09:09 +10:00
#set resume point
resume = int(round(float(timeInfo.get("ResumeTime"))))*60
total = int(round(float(timeInfo.get("TotalTime"))))*60
self.setKodiResumePoint(fileid, resume, total, cursor, playcount, dateplayed, realpath, realfile)
2015-05-03 06:02:06 +10:00
2015-07-26 22:09:09 +10:00
def addOrUpdateTvShowToKodiLibrary(self, embyId, connection, cursor, viewTag ):
MBitem = ReadEmbyDB().getFullItem(embyId)
2015-05-13 02:16:30 +10:00
if not MBitem:
2015-07-25 13:49:15 +10:00
self.logMsg("ADD tvshow to Kodi library FAILED, Item %s not found on server!" % embyId)
2015-05-13 02:16:30 +10:00
return
# 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
cursor.execute("SELECT kodi_id FROM emby WHERE emby_id = ?", (embyId,))
try:
showid = cursor.fetchone()[0]
except:
showid = None
2015-07-26 22:09:09 +10:00
self.logMsg("TV Show Id: %s not found." % embyId, 1)
timeInfo = API().getTimeInfo(MBitem)
userData = API().getUserData(MBitem)
people = API().getPeople(MBitem)
genres = MBitem.get('Genres')
studios = API().getStudios(MBitem)
#### The tv show details ####
playcount = userData.get('PlayCount')
dateplayed = userData.get("LastPlayedDate")
dateadded = API().getDateCreated(MBitem)
checksum = API().getChecksum(MBitem)
title = MBitem['Name']
plot = API().getOverview(MBitem)
rating = MBitem.get('CommunityRating')
premieredate = API().getPremiereDate(MBitem)
genre = " / ".join(genres)
tvdb = API().getProvider(MBitem, "tvdb")
mpaa = API().getMpaa(MBitem)
studio = " / ".join(studios)
sorttitle = MBitem['SortName']
2015-07-26 22:09:09 +10:00
#create toplevel path as monitored source - needed for things like actors and stuff to work (no clue why)
if self.directpath:
2015-07-24 17:52:36 +10:00
# Network share
playurl = PlayUtils().directPlay(MBitem)
2015-07-24 17:52:36 +10:00
if "/" in playurl:
# Network path
path = "%s/" % playurl
toplevelpath = "%s/" % dirname(dirname(path))
else:
# Local path
path = "%s\\" % playurl
toplevelpath = "%s\\" % dirname(dirname(path))
2015-07-26 22:09:09 +10:00
else:# Set plugin path
path = "plugin://plugin.video.emby/tvshows/%s/" % embyId
toplevelpath = "plugin://plugin.video.emby/"
##### UPDATE THE TV SHOW #####
if showid:
self.logMsg("UPDATE tvshow to Kodi library, Id: %s - Title: %s" % (embyId, title))
query = "UPDATE tvshow SET c00 = ?, c01 = ?, c04 = ?, c05 = ?, c08 = ?, c09 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ? WHERE idShow = ?"
cursor.execute(query, (title, plot, rating, premieredate, genre, title, tvdb, mpaa, studio, sorttitle, showid))
2015-05-02 20:51:46 +10:00
# Update the checksum in emby table
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
cursor.execute(query, (checksum, embyId))
##### OR ADD THE TV SHOW #####
else:
self.logMsg("ADD tvshow to Kodi library, Id: %s - Title: %s" % (embyId, title))
# Create the TV show path
cursor.execute("select coalesce(max(idPath),0) as pathid from path")
pathid = cursor.fetchone()[0] + 1
query = "INSERT INTO path(idPath, strPath, strContent, strScraper, noUpdate) values(?, ?, ?, ?, ?)"
cursor.execute(query, (pathid, path, None, None, 1))
# Validate the top level path in database
cursor.execute("SELECT idPath as tlpathid FROM path WHERE strPath = ?", (toplevelpath,))
try:
cursor.fetchone()[0]
except:
# Top level path does not exist yet
cursor.execute("select coalesce(max(idPath),0) as tlpathid from path")
tlpathid = cursor.fetchone()[0] + 1
query = "INSERT INTO path(idPath, strPath, strContent, strScraper, noUpdate) values(?, ?, ?, ?, ?)"
cursor.execute(query, (tlpathid, toplevelpath, "tvshows", "metadata.local", 1))
# Create the TV show
cursor.execute("select coalesce(max(idShow),0) as showid from tvshow")
showid = cursor.fetchone()[0] + 1
query = "INSERT INTO tvshow(idShow, c00, c01, c04, c05, c08, c09, c12, c13, c14, c15) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (showid, title, plot, rating, premieredate, genre, title, tvdb, mpaa, studio, sorttitle))
2015-05-02 10:26:06 +10:00
# Create the reference in emby table
query = "INSERT INTO emby(emby_id, kodi_id, media_type, checksum) values(?, ?, ?, ?)"
cursor.execute(query, (embyId, showid, "tvshow", checksum))
# Link the path
query = "INSERT INTO tvshowlinkpath(idShow, idPath) values(?, ?)"
cursor.execute(query, (showid, pathid))
2015-10-02 13:30:11 +10:00
# Add tags to item, view tag, emby tags and favourite
tags = [viewTag]
tags.extend(MBitem['Tags'])
if userData['Favorite']:
tags.append("Favorite tvshows")
self.AddTagsToMedia(showid, tags, "tvshow", cursor)
# Update artwork
self.textureCache.addArtwork(API().getAllArtwork(MBitem), showid, "tvshow", cursor)
# Update or insert people
self.AddPeopleToMedia(showid, MBitem.get('People'),"tvshow", connection, cursor)
# Update genres
2015-05-02 01:13:22 +10:00
self.AddGenresToMedia(showid, genres, "tvshow", cursor)
# Update studios
2015-05-02 01:13:22 +10:00
self.AddStudiosToMedia(showid, studios, "tvshow", cursor)
# Update season details
self.updateSeasons(embyId, showid, connection, cursor)
def addOrUpdateEpisodeToKodiLibrary(self, embyId, showid, connection, cursor):
2015-10-12 09:50:46 +11:00
kodiVersion = self.kodiversion
MBitem = ReadEmbyDB().getFullItem(embyId)
2015-05-13 02:16:30 +10:00
if not MBitem:
self.logMsg("ADD episode to Kodi library FAILED, Item %s not found on server!" % embyId, 1)
2015-05-13 02:16:30 +10:00
return
# If the episode 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
2015-03-18 06:02:42 +11:00
2015-07-26 22:09:09 +10:00
cursor.execute("SELECT kodi_id FROM emby WHERE emby_id = ?", (MBitem['Id'],))
try:
episodeid = cursor.fetchone()[0]
except:
episodeid = None
2015-07-26 22:09:09 +10:00
self.logMsg("Episode Id: %s not found." % embyId, 1)
timeInfo = API().getTimeInfo(MBitem)
userData = API().getUserData(MBitem)
people = API().getPeople(MBitem)
##### The episode details #####
seriesId = MBitem['SeriesId']
seriesName = MBitem['SeriesName']
season = MBitem.get('ParentIndexNumber')
episode = MBitem.get('IndexNumber', 0)
if utils.settings('syncSpecialsOrder') == "true":
airsBeforeSeason = MBitem.get('AirsBeforeSeasonNumber', "-1")
airsBeforeEpisode = MBitem.get('AirsBeforeEpisodeNumber', "-1")
else:
airsBeforeSeason = "-1"
airsBeforeEpisode = "-1"
playcount = userData.get('PlayCount')
dateplayed = userData.get("LastPlayedDate")
dateadded = API().getDateCreated(MBitem)
checksum = API().getChecksum(MBitem)
title = MBitem['Name']
plot = API().getOverview(MBitem)
rating = MBitem.get('CommunityRating')
2015-07-26 22:09:09 +10:00
writer = " / ".join(people.get('Writer'))
premieredate = API().getPremiereDate(MBitem)
runtime = timeInfo.get('TotalTime')
director = " / ".join(people.get('Director'))
2015-07-26 22:09:09 +10:00
playurl = PlayUtils().directPlay(MBitem)
realfile = ""
realpath = ""
2015-07-26 22:09:09 +10:00
if self.directpath:
if playurl == False:
return
elif "\\" in playurl:
filename = playurl.rsplit("\\",1)[-1]
path = playurl.replace(filename, "")
elif "/" in playurl:
filename = playurl.rsplit("/",1)[-1]
path = playurl.replace(filename, "")
else:
self.logMsg("Invalid path: %s" % playurl, 1)
return
2015-07-26 22:09:09 +10:00
else: # Set plugin path and media flags - real filename with extension
realfile = ""
realpath = ""
2015-07-26 22:09:09 +10:00
try:
if not "plugin://" in playurl:
realpath, realfile = ntsplit(playurl)
if "/" in playurl:
realpath = realpath + "/"
else:
realpath = realpath + "\\"
except:
pass
filename = "plugin://plugin.video.emby/tvshows/%s/?filename=%s&id=%s&mode=play" % (seriesId, realfile, embyId)
path = "plugin://plugin.video.emby/tvshows/%s/" % seriesId
2015-07-26 22:09:09 +10:00
# Validate the season exists in Emby and in database
if season is None:
self.logMsg("SKIP adding episode to Kodi Library, no season assigned - ID: %s - %s" % (embyId, title))
return False
idSeason = None
count = 0
while idSeason is None:
cursor.execute("SELECT idSeason FROM seasons WHERE idShow = ? and season = ?", (showid, season,))
try:
idSeason = cursor.fetchone()[0]
except: # Season does not exist, update seasons
if not count:
self.updateSeasons(seriesId, showid, connection, cursor)
count += 1
else:
# Season is still not found, skip episode.
self.logMsg("Skipping episode: %s. Season number is missing at season level in the metadata manager." % title, 1)
return False
##### UPDATE THE EPISODE #####
if episodeid:
self.logMsg("UPDATE episode from // %s - S%s // to Kodi library, Id: %s - E%s: %s" % (seriesName, season, embyId, episode, title))
#get the file ID
cursor.execute("SELECT idFile as fileid FROM episode WHERE idEpisode = ?", (episodeid,))
fileid = cursor.fetchone()[0]
#always update the filepath (fix for path change)
query = "UPDATE files SET strFilename = ?, dateAdded = ? WHERE idFile = ?"
cursor.execute(query, (filename, dateadded, fileid))
2015-10-12 09:50:46 +11:00
if kodiVersion == 16:
query = "UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ?, idSeason = ? WHERE idEpisode = ?"
cursor.execute(query, (title, plot, rating, writer, premieredate, runtime, director, season, episode, title, airsBeforeSeason, airsBeforeEpisode, idSeason ,episodeid))
else:
query = "UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ? WHERE idEpisode = ?"
cursor.execute(query, (title, plot, rating, writer, premieredate, runtime, director, season, episode, title, airsBeforeSeason, airsBeforeEpisode, episodeid))
2015-05-02 10:26:06 +10:00
#update the checksum in emby table
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
cursor.execute(query, (checksum, embyId))
##### OR ADD THE EPISODE #####
else:
self.logMsg("ADD episode from // %s - S%s // to Kodi library, Id: %s - E%s: %s" % (seriesName, season, embyId, episode, title))
# Validate the path in database
if self.directpath:
2015-07-28 09:16:24 +10:00
cursor.execute("SELECT idPath as pathid FROM path WHERE strPath = ?", (path,))
else:
2015-07-28 09:16:24 +10:00
cursor.execute("SELECT idPath as pathid FROM path WHERE strPath LIKE ?", (seriesId,))
try:
pathid = cursor.fetchone()[0]
except:
# Path does not exist yet
cursor.execute("select coalesce(max(idPath),0) as pathid from path")
pathid = cursor.fetchone()[0] + 1
query = "INSERT INTO path(idPath, strPath, strContent, strScraper, noUpdate) values(?, ?, ?, ?, ?)"
cursor.execute(query, (pathid, path, None, None, 1))
# Validate the file in database
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename = ? and idPath = ?", (filename, pathid,))
try:
fileid = cursor.fetchone()[0]
except:
# File does not exist yet
cursor.execute("select coalesce(max(idFile),0) as fileid from files")
fileid = cursor.fetchone()[0] + 1
query = "INSERT INTO files(idFile, idPath, strFilename, playCount, lastPlayed, dateAdded) values(?, ?, ?, ?, ?, ?)"
cursor.execute(query, (fileid, pathid, filename, playcount, dateplayed, dateadded))
# Create the episode
cursor.execute("select coalesce(max(idEpisode),0) as episodeid from episode")
episodeid = cursor.fetchone()[0] + 1
2015-10-12 09:50:46 +11:00
if kodiVersion == 16:
query = "INSERT INTO episode(idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16, idSeason) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (episodeid, fileid, title, plot, rating, writer, premieredate, runtime, director, season, episode, title, showid, airsBeforeSeason, airsBeforeEpisode, idSeason))
else:
query = "INSERT INTO episode(idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (episodeid, fileid, title, plot, rating, writer, premieredate, runtime, director, season, episode, title, showid, airsBeforeSeason, airsBeforeEpisode))
# Create the reference in emby table
query = "INSERT INTO emby(emby_id, kodi_id, kodi_file_id, media_type, checksum, parent_id) values(?, ?, ?, ?, ?, ?)"
cursor.execute(query, (embyId, episodeid, fileid, "episode", checksum, showid))
# Update or insert actors
self.AddPeopleToMedia(episodeid, MBitem.get('People'), "episode", connection, cursor)
# Add streamdetails
2015-08-01 06:17:05 +10:00
self.AddStreamDetailsToMedia(API().getMediaStreams(MBitem), runtime, fileid, cursor)
# Update artwork
artworks = API().getAllArtwork(MBitem)
self.textureCache.addOrUpdateArt(artworks['Primary'], episodeid, "episode", "thumb", cursor)
# Set resume point and round to 6th decimal
resume = round(float(timeInfo.get('ResumeTime')), 6)
total = round(float(timeInfo.get('TotalTime')), 6)
jumpback = int(utils.settings('resumeJumpBack'))
if resume > jumpback:
# To avoid negative bookmark
resume = resume - jumpback
self.setKodiResumePoint(fileid, resume, total, cursor, playcount, dateplayed, realpath, realfile)
2015-07-26 22:09:09 +10:00
def deleteItemFromKodiLibrary(self, id, connection, cursor):
cursor.execute("SELECT kodi_id, media_type FROM emby WHERE emby_id = ?", (id,))
try:
result = cursor.fetchone()
kodi_id = result[0]
media_type = result[1]
except: pass
else: # Delete entry from database
if "movie" in media_type:
self.logMsg("Deleting movie from Kodi library, Id: %s" % id, 1)
cursor.execute("DELETE FROM movie WHERE idMovie = ?", (kodi_id,))
elif "episode" in media_type:
cursor.execute("DELETE FROM episode WHERE idEpisode = ?", (kodi_id,))
self.logMsg("Deleting episode from Kodi library, Id: %s" % id, 1)
elif "tvshow" in media_type:
cursor.execute("DELETE FROM tvshow WHERE idShow = ?", (kodi_id,))
self.logMsg("Deleting tvshow from Kodi library, Id: %s" % id, 1)
2015-05-02 22:57:43 +10:00
elif "musicvideo" in media_type:
cursor.execute("DELETE FROM musicvideo WHERE idMVideo = ?", (kodi_id,))
self.logMsg("Deleting musicvideo from Kodi library, Id: %s" % id, 1)
# Delete the record in emby table
2015-05-02 22:57:43 +10:00
cursor.execute("DELETE FROM emby WHERE emby_id = ?", (id,))
2015-07-26 22:09:09 +10:00
def updateSeasons(self, embyTvShowId, kodiTvShowId, connection, cursor):
2015-03-21 02:16:39 +11:00
textureCache = self.textureCache
seasonData = ReadEmbyDB().getTVShowSeasons(embyTvShowId)
2015-08-04 19:45:26 +10:00
for season in seasonData:
seasonNum = season.get('IndexNumber')
cursor.execute("SELECT idSeason as seasonid FROM seasons WHERE idShow = ? and season = ?", (kodiTvShowId, seasonNum,))
try:
seasonid = cursor.fetchone()[0]
except: # Create the season
cursor.execute("select coalesce(max(idSeason),0) as seasonid from seasons")
seasonid = cursor.fetchone()[0] + 1
query = "INSERT INTO seasons(idSeason, idShow, season) values(?, ?, ?)"
cursor.execute(query, (seasonid, kodiTvShowId, seasonNum))
finally: # Update artwork
textureCache.addArtwork(API().getAllArtwork(season), seasonid, "season", cursor)
# All season entry
MBitem = ReadEmbyDB().getFullItem(embyTvShowId)
seasonNum = -1
cursor.execute("SELECT idSeason as seasonid FROM seasons WHERE idShow = ? and season = ?", (kodiTvShowId, seasonNum,))
try:
seasonid = cursor.fetchone()[0]
except: # Create all season entry
cursor.execute("select coalesce(max(idSeason),0) as seasonid from seasons")
seasonid = cursor.fetchone()[0] + 1
query = "INSERT INTO seasons(idSeason, idShow, season) values(?, ?, ?)"
cursor.execute(query, (seasonid, kodiTvShowId, seasonNum))
finally: # Update the artwork
textureCache.addArtwork(API().getAllArtwork(MBitem), seasonid, "season", cursor)
2015-05-01 21:57:24 +10:00
def addOrUpdateArt(self, imageUrl, kodiId, mediaType, imageType, cursor):
2015-07-26 22:09:09 +10:00
2015-05-01 21:57:24 +10:00
if imageUrl:
2015-07-26 22:09:09 +10:00
cacheimage = False
cursor.execute("SELECT url FROM art WHERE media_id = ? AND media_type = ? AND type = ?", (kodiId, mediaType, imageType,))
try: # Update the artwork
url = cursor.fetchone()[0]
except: # Add the artwork
cacheimage = True
self.logMsg("Adding Art Link for kodiId: %s (%s)" % (kodiId, imageUrl), 1)
query = "INSERT INTO art(media_id, media_type, type, url) values(?, ?, ?, ?)"
cursor.execute(query, (kodiId, mediaType, imageType, imageUrl))
else:
if url != imageUrl:
cacheimage = True
self.logMsg("Updating Art Link for kodiId: %s (%s) -> (%s)" % (kodiId, url, imageUrl), 1)
2015-07-26 22:09:09 +10:00
query = "UPDATE art set url = ? WHERE media_id = ? AND media_type = ? AND type = ?"
cursor.execute(query, (imageUrl, kodiId, mediaType, imageType))
# Cache fanart and poster in Kodi texture cache
if cacheimage and imageType in ("fanart", "poster"):
self.textureCache.CacheTexture(imageUrl)
2015-07-29 10:51:57 +10:00
def setKodiResumePoint(self, fileid, resume_seconds, total_seconds, cursor, playcount, dateplayed=None, realpath=None, realfile=None):
2015-03-18 06:02:42 +11:00
if realpath:
#delete any existing resume point for the real filepath
cursor.execute("SELECT idPath as pathid FROM path WHERE strPath = ?", (realpath,))
result = cursor.fetchone()
if result:
pathid = result[0]
cursor.execute("SELECT idFile as fileid FROM files WHERE strFilename = ? and idPath = ?", (realfile, pathid,))
result = cursor.fetchone()
if result:
cursor.execute("DELETE FROM bookmark WHERE idFile = ?", (result[0],))
#delete existing resume point for the actual filepath
cursor.execute("DELETE FROM bookmark WHERE idFile = ?", (fileid,))
#set watched count
2015-07-29 10:51:57 +10:00
query = "UPDATE files SET playCount = ?, lastPlayed = ? WHERE idFile = ?"
cursor.execute(query, (playcount, dateplayed, fileid))
#set the resume bookmark
if resume_seconds:
2015-05-02 10:26:06 +10:00
cursor.execute("select coalesce(max(idBookmark),0) as bookmarkId from bookmark")
bookmarkId = cursor.fetchone()[0] + 1
query = "INSERT INTO bookmark(idBookmark, idFile, timeInSeconds, totalTimeInSeconds, thumbNailImage, player, playerState, type) values(?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (bookmarkId, fileid, resume_seconds, total_seconds, None, "DVDPlayer", None, 1))
2015-05-02 10:26:06 +10:00
def AddPeopleToMedia(self, id, people, mediatype, connection, cursor):
kodiVersion = self.kodiversion
if people:
2015-07-16 06:05:12 +10:00
castorder = 1
2015-07-16 06:05:12 +10:00
for person in people:
name = person['Name']
type = person['Type']
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard/jarvis
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT actor_id as actorid FROM actor WHERE name = ? COLLATE NOCASE", (name,))
else:
# Kodi Gotham or Helix
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT idActor as actorid FROM actors WHERE strActor = ? COLLATE NOCASE", (name,))
try: # Update person in database
actorid = cursor.fetchone()[0]
except:
# Person entry does not exist yet.
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard
cursor.execute("select coalesce(max(actor_id),0) as actorid from actor")
query = "INSERT INTO actor(actor_id, name) values(?, ?)"
else:
# Kodi Gotham or Helix
cursor.execute("select coalesce(max(idActor),0) as actorid from actors")
query = "INSERT INTO actors(idActor, strActor) values(?, ?)"
actorid = cursor.fetchone()[0] + 1
self.logMsg("Adding people to Media, processing: %s" % name, 2)
cursor.execute(query, (actorid, name))
finally:
2015-09-15 22:48:42 +10:00
query = ""
# Add person image to art table
thumb = API().imageUrl(person['Id'], "Primary", 0, 400, 400)
if thumb:
arttype = type.lower()
if "writing" in arttype:
arttype = "writer"
self.textureCache.addOrUpdateArt(thumb, actorid, arttype, "thumb", cursor)
# Link person to content in database
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard
if "Actor" in type:
Role = person.get('Role')
query = "INSERT OR REPLACE INTO actor_link(actor_id, media_id, media_type, role, cast_order) values(?, ?, ?, ?, ?)"
cursor.execute(query, (actorid, id, mediatype, Role, castorder))
castorder += 1
elif "Director" in type:
query = "INSERT OR REPLACE INTO director_link(actor_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (actorid, id, mediatype))
elif type in ("Writing", "Writer"):
query = "INSERT OR REPLACE INTO writer_link(actor_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (actorid, id, mediatype))
2015-10-24 12:00:34 +11:00
elif "Artist" in type:
query = "INSERT OR REPLACE INTO actor_link(actor_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (actorid, id, mediatype))
else:
# Kodi Gotham or Helix
if "Actor" in type:
Role = person.get('Role')
query = None
if "movie" in mediatype:
query = "INSERT OR REPLACE INTO actorlinkmovie(idActor, idMovie, strRole, iOrder) values(?, ?, ?, ?)"
elif "tvshow" in mediatype:
query = "INSERT OR REPLACE INTO actorlinktvshow(idActor, idShow, strRole, iOrder) values(?, ?, ?, ?)"
elif "episode" in mediatype:
query = "INSERT OR REPLACE INTO actorlinkepisode(idActor, idEpisode, strRole, iOrder) values(?, ?, ?, ?)"
if query:
cursor.execute(query, (actorid, id, Role, castorder))
castorder += 1
elif "Director" in type:
if "movie" in mediatype:
query = "INSERT OR REPLACE INTO directorlinkmovie(idDirector, idMovie) values(?, ?)"
elif "tvshow" in mediatype:
query = "INSERT OR REPLACE INTO directorlinktvshow(idDirector, idShow) values(?, ?)"
elif "musicvideo" in mediatype:
query = "INSERT OR REPLACE INTO directorlinkmusicvideo(idDirector, idMVideo) values(?, ?)"
elif "episode" in mediatype:
query = "INSERT OR REPLACE INTO directorlinkepisode(idDirector, idEpisode) values(?, ?)"
2015-09-15 22:48:42 +10:00
if query:
cursor.execute(query, (actorid, id))
elif type in ("Writing", "Writer"):
if "movie" in mediatype:
query = "INSERT OR REPLACE INTO writerlinkmovie(idWriter, idMovie) values(?, ?)"
elif "episode" in mediatype:
query = "INSERT OR REPLACE INTO writerlinkepisode(idWriter, idEpisode) values(?, ?)"
2015-09-15 22:48:42 +10:00
if query:
cursor. execute(query, (actorid, id))
2015-10-24 12:00:34 +11:00
elif "Artist" in type:
query = "INSERT OR REPLACE INTO artistlinkmusicvideo(idArtist, idMVideo) values(?, ?)"
cursor.execute(query, (actorid, id))
2015-05-02 02:32:13 +10:00
2015-05-06 00:20:09 +10:00
def AddGenresToMedia(self, id, genres, mediatype, cursor):
kodiVersion = self.kodiversion
2015-05-06 00:20:09 +10:00
if genres:
# Delete current genres for clean slate
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
cursor.execute("DELETE FROM genre_link WHERE media_id = ? AND media_type = ?", (id, mediatype,))
else:
if "movie" in mediatype:
cursor.execute("DELETE FROM genrelinkmovie WHERE idMovie = ?", (id,))
elif "tvshow" in mediatype:
cursor.execute("DELETE FROM genrelinktvshow WHERE idShow = ?", (id,))
elif "musicvideo" in mediatype:
cursor.execute("DELETE FROM genrelinkmusicvideo WHERE idMVideo = ?", (id,))
# Add Genres
2015-05-06 00:20:09 +10:00
for genre in genres:
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT genre_id as genre_id FROM genre WHERE name = ? COLLATE NOCASE", (genre,))
try:
2015-05-06 00:20:09 +10:00
genre_id = cursor.fetchone()[0]
except:
# Create genre in database
cursor.execute("select coalesce(max(genre_id),0) as genre_id from genre")
genre_id = cursor.fetchone()[0] + 1
query = "INSERT INTO genre(genre_id, name) values(?, ?)"
2015-07-26 22:09:09 +10:00
cursor.execute(query, (genre_id, genre))
self.logMsg("Add Genres to media, processing: %s" % genre, 1)
finally:
# Assign genre to item
query = "INSERT OR REPLACE INTO genre_link(genre_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (genre_id, id, mediatype))
2015-05-06 00:20:09 +10:00
else:
# Kodi Gotham or Helix
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT idGenre as idGenre FROM genre WHERE strGenre = ? COLLATE NOCASE", (genre,))
try:
2015-05-06 00:20:09 +10:00
idGenre = cursor.fetchone()[0]
except:
# Create genre in database
cursor.execute("select coalesce(max(idGenre),0) as idGenre from genre")
idGenre = cursor.fetchone()[0] + 1
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
2015-07-26 22:09:09 +10:00
cursor.execute(query, (idGenre, genre))
self.logMsg("Add Genres to media, processing: %s" % genre, 1)
finally:
# Assign genre to item
if "movie" in mediatype:
query = "INSERT OR REPLACE into genrelinkmovie(idGenre, idMovie) values(?, ?)"
elif "tvshow" in mediatype:
query = "INSERT OR REPLACE into genrelinktvshow(idGenre, idShow) values(?, ?)"
elif "musicvideo" in mediatype:
query = "INSERT OR REPLACE into genrelinkmusicvideo(idGenre, idMVideo) values(?, ?)"
2015-07-26 22:09:09 +10:00
else: # Item is invalid
return
cursor.execute(query, (idGenre, id))
2015-07-26 22:09:09 +10:00
def AddCountriesToMedia(self, id, countries, mediatype, cursor):
kodiVersion = self.kodiversion
if countries:
for country in countries:
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT country_id as country_id FROM country WHERE name = ? COLLATE NOCASE", (country,))
try:
country_id = cursor.fetchone()[0]
except:
# Country entry does not exists
cursor.execute("select coalesce(max(country_id),0) as country_id from country")
country_id = cursor.fetchone()[0] + 1
query = "INSERT INTO country(country_id, name) values(?, ?)"
cursor.execute(query, (country_id, country))
self.logMsg("Add Countries to Media, processing: %s" % country)
finally:
# Assign country to content
query = "INSERT OR REPLACE INTO country_link(country_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (country_id, id, mediatype))
else:
# Kodi Gotham or Helix
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT idCountry as idCountry FROM country WHERE strCountry = ? COLLATE NOCASE", (country,))
try:
idCountry = cursor.fetchone()[0]
except:
# Country entry does not exists
cursor.execute("select coalesce(max(idCountry),0) as idCountry from country")
idCountry = cursor.fetchone()[0] + 1
query = "INSERT INTO country(idCountry, strCountry) values(?, ?)"
cursor.execute(query, (idCountry, country))
finally:
# Only movies have a country field
if "movie" in mediatype:
query = "INSERT OR REPLACE INTO countrylinkmovie(idCountry, idMovie) values(?, ?)"
cursor.execute(query, (idCountry, id))
def AddStudiosToMedia(self, id, studios, mediatype, cursor):
kodiVersion = self.kodiversion
if studios:
for studio in studios:
2015-07-24 21:29:41 +10:00
if kodiVersion == 15 or kodiVersion == 16:
# Kodi Isengard
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT studio_id as studio_id FROM studio WHERE name = ? COLLATE NOCASE", (studio,))
try:
studio_id = cursor.fetchone()[0]
except: # Studio does not exists.
cursor.execute("select coalesce(max(studio_id),0) as studio_id from studio")
studio_id = cursor.fetchone()[0] + 1
query = "INSERT INTO studio(studio_id, name) values(?, ?)"
cursor.execute(query, (studio_id,studio))
self.logMsg("Add Studios to media, processing: %s" % studio, 1)
finally: # Assign studio to item
query = "INSERT OR REPLACE INTO studio_link(studio_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (studio_id, id, mediatype))
else:
# Kodi Gotham or Helix
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT idstudio as idstudio FROM studio WHERE strstudio = ? COLLATE NOCASE",(studio,))
try:
idstudio = cursor.fetchone()[0]
except: # Studio does not exists.
cursor.execute("select coalesce(max(idstudio),0) as idstudio from studio")
idstudio = cursor.fetchone()[0] + 1
query = "INSERT INTO studio(idstudio, strstudio) values(?, ?)"
cursor.execute(query, (idstudio,studio))
self.logMsg("Add Studios to media, processing: %s" % studio, 1)
finally: # Assign studio to item
2015-07-26 22:09:09 +10:00
if "movie" in mediatype:
query = "INSERT OR REPLACE INTO studiolinkmovie(idstudio, idMovie) values(?, ?)"
elif "musicvideo" in mediatype:
query = "INSERT OR REPLACE INTO studiolinkmusicvideo(idstudio, idMVideo) values(?, ?)"
elif "tvshow" in mediatype:
query = "INSERT OR REPLACE INTO studiolinktvshow(idstudio, idShow) values(?, ?)"
elif "episode" in mediatype:
query = "INSERT OR REPLACE INTO studiolinkepisode(idstudio, idEpisode) values(?, ?)"
cursor.execute(query, (idstudio, id))
2015-10-02 13:30:11 +10:00
2015-10-02 13:30:11 +10:00
def AddTagsToMedia(self, id, tags, mediatype, cursor):
2015-10-02 13:30:11 +10:00
# First, delete any existing tags associated to the id
2015-10-13 04:22:51 +11:00
if self.kodiversion in (15, 16):
2015-10-02 13:30:11 +10:00
# Kodi Isengard, Jarvis
query = "DELETE FROM tag_link WHERE media_id = ? AND media_type = ?"
cursor.execute(query, (id, mediatype))
2015-10-02 13:30:11 +10:00
else: # Kodi Helix
query = "DELETE FROM taglinks WHERE idMedia = ? AND media_type = ?"
cursor.execute(query, (id, mediatype))
# Add tags
self.logMsg("Adding Tags: %s" % tags, 1)
for tag in tags:
self.AddTagToMedia(id, tag, mediatype, cursor)
2015-10-02 13:30:11 +10:00
def AddTagToMedia(self, id, tag, mediatype, cursor, doRemove=False):
2015-10-13 04:22:51 +11:00
if self.kodiversion in (15, 16):
2015-10-02 13:30:11 +10:00
# Kodi Isengard, Jarvis
cursor.execute("SELECT tag_id FROM tag WHERE name = ? COLLATE NOCASE", (tag,))
try:
tag_id = cursor.fetchone()[0]
except:
# Create the tag, because it does not exist
cursor.execute("select coalesce(max(tag_id),0) as tag_id from tag")
tag_id = cursor.fetchone()[0] + 1
query = "INSERT INTO tag(tag_id, name) values(?, ?)"
cursor.execute(query, (tag_id, tag))
self.logMsg("Add Tag to media, adding tag: %s" % tag, 2)
finally:
# Assign tag to item
if not doRemove:
query = "INSERT OR REPLACE INTO tag_link(tag_id, media_id, media_type) values(?, ?, ?)"
cursor.execute(query, (tag_id, id, mediatype))
else:
query = "DELETE FROM tag_link WHERE media_id = ? AND media_type = ? AND tag_id = ?"
cursor.execute(query, (id, mediatype, tag_id))
else:
# Kodi Helix
cursor.execute("SELECT idTag FROM tag WHERE strTag = ? COLLATE NOCASE", (tag,))
try:
idTag = cursor.fetchone()[0]
except:
# Create the tag
cursor.execute("select coalesce(max(idTag),0) as idTag from tag")
idTag = cursor.fetchone()[0] + 1
query = "INSERT INTO tag(idTag, strTag) values(?, ?)"
cursor.execute(query, (idTag, tag))
self.logMsg("Add Tag to media, adding tag: %s" % tag, 2)
finally:
# Assign tag to item
if not doRemove:
query = "INSERT OR REPLACE INTO taglinks(idTag, idMedia, media_type) values(?, ?, ?)"
cursor.execute(query, (idTag, id, mediatype))
else:
query = "DELETE FROM taglinks WHERE idMedia = ? AND media_type = ? AND idTag = ?"
cursor.execute(query, (id, mediatype, idTag))
2015-05-02 10:57:58 +10:00
2015-10-02 13:30:11 +10:00
2015-08-01 06:17:05 +10:00
def AddStreamDetailsToMedia(self, streamdetails, runtime , fileid, cursor):
2015-05-02 10:57:58 +10:00
# First remove any existing entries
cursor.execute("DELETE FROM streamdetails WHERE idFile = ?", (fileid,))
2015-05-02 10:57:58 +10:00
if streamdetails:
# Video details
for videotrack in streamdetails['videocodec']:
2015-08-01 06:17:05 +10:00
query = "INSERT INTO streamdetails(idFile, iStreamType, strVideoCodec, fVideoAspect, iVideoWidth, iVideoHeight, iVideoDuration ,strStereoMode) values(?, ?, ?, ?, ?, ?, ?, ?)"
cursor.execute(query, (fileid, 0, videotrack.get('videocodec'), videotrack.get('aspectratio'), videotrack.get('width'), videotrack.get('height'), runtime ,videotrack.get('Video3DFormat')))
# Audio details
for audiotrack in streamdetails['audiocodec']:
query = "INSERT INTO streamdetails(idFile, iStreamType, strAudioCodec, iAudioChannels, strAudioLanguage) values(?, ?, ?, ?, ?)"
cursor.execute(query, (fileid, 1, audiotrack.get('audiocodec'), audiotrack.get('channels'), audiotrack.get('audiolanguage')))
# Subtitles details
for subtitletrack in streamdetails['subtitlelanguage']:
query = "INSERT INTO streamdetails(idFile, iStreamType, strSubtitleLanguage) values(?, ?, ?)"
cursor.execute(query, (fileid, 2, subtitletrack))
2015-05-03 00:49:47 +10:00
2015-04-05 03:20:48 +10:00
def addBoxsetToKodiLibrary(self, boxset, connection, cursor):
2015-03-25 03:47:26 +11:00
strSet = boxset['Name']
2015-03-25 03:47:26 +11:00
cursor.execute("SELECT idSet FROM sets WHERE strSet = ?", (strSet,))
try:
setid = cursor.fetchone()[0]
except:
# Boxset does not exists
query = "INSERT INTO sets(idSet, strSet) values(?, ?)"
cursor.execute(query, (None, strSet))
# Get the setid of the new boxset
cursor.execute("SELECT idSet FROM sets WHERE strSet = ?", (strSet,))
setid = cursor.fetchone()[0]
finally:
# Assign artwork
2015-07-26 22:09:09 +10:00
cursor.execute('SELECT type, url FROM art WHERE media_type = ? AND media_id = ? and url != ""', ("set", setid,))
2015-03-25 03:47:26 +11:00
existing_type_map = {}
rows = cursor.fetchall()
for row in rows:
existing_type_map[row[0] ] = row[1]
artwork = {}
artwork['poster'] = API().getArtwork(boxset, "Primary", mediaType = "boxset")
artwork['banner'] = API().getArtwork(boxset, "Banner", mediaType = "boxset")
artwork['clearlogo'] = API().getArtwork(boxset, "Logo", mediaType = "boxset")
artwork['clearart'] = API().getArtwork(boxset, "Art", mediaType = "boxset")
artwork['landscape'] = API().getArtwork(boxset, "Thumb", mediaType = "boxset")
artwork['discart'] = API().getArtwork(boxset, "Disc", mediaType = "boxset")
artwork['fanart'] = API().getArtwork(boxset, "Backdrop", mediaType = "boxset")
2015-03-25 03:47:26 +11:00
2015-03-25 05:58:13 +11:00
art_types = ['poster','fanart','landscape','clearlogo','clearart','banner','discart']
2015-03-25 03:47:26 +11:00
for update_type in art_types:
if ( update_type in existing_type_map ):
if ( existing_type_map[update_type] != artwork[update_type] ) and artwork[update_type] != '':
setupdateartsql = "UPDATE art SET url = ? where media_type = ? and media_id = ? and type = ?"
cursor.execute(setupdateartsql,(artwork[update_type],"set",setid,update_type))
elif artwork[update_type] != '':
setartsql = "INSERT INTO art(media_id, media_type, type, url) VALUES(?, ?, ?, ?)"
2015-03-25 03:47:26 +11:00
cursor.execute(setartsql,(setid,"set",update_type,artwork[update_type]))
2015-03-25 03:47:26 +11:00
return True
2015-05-03 22:44:23 +10:00
def updateBoxsetToKodiLibrary(self, boxsetmovie, boxset, connection, cursor):
strSet = boxset['Name']
boxsetmovieid = boxsetmovie['Id']
2015-07-26 22:09:09 +10:00
cursor.execute("SELECT kodi_id FROM emby WHERE emby_id = ?", (boxsetmovieid,))
try:
movieid = cursor.fetchone()[0]
2015-07-31 05:23:50 +10:00
cursor.execute("SELECT idSet FROM sets WHERE strSet = ? COLLATE NOCASE", (strSet,))
setid = cursor.fetchone()[0]
except: pass
else:
query = "UPDATE movie SET idSet = ? WHERE idMovie = ?"
cursor.execute(query, (setid, movieid))
# Update the checksum in emby table
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
cursor.execute(query, (API().getChecksum(boxsetmovie), boxsetmovieid))
def removeMoviesFromBoxset(self, boxset, connection, cursor):
strSet = boxset['Name']
try:
cursor.execute("SELECT idSet FROM sets WHERE strSet = ? COLLATE NOCASE", (strSet,))
setid = cursor.fetchone()[0]
except: pass
else:
query = "UPDATE movie SET idSet = null WHERE idSet = ?"
cursor.execute(query, (setid,))
def updateUserdata(self, userdata, connection, cursor):
# This updates: Favorite, LastPlayedDate, Playcount, PlaybackPositionTicks
embyId = userdata['ItemId']
MBitem = ReadEmbyDB().getItem(embyId)
if not MBitem:
self.logMsg("UPDATE userdata to Kodi library FAILED, Item %s not found on server!" % embyId, 1)
return
# Get details
checksum = API().getChecksum(MBitem)
userdata = API().getUserData(MBitem)
timeInfo = API().getTimeInfo(MBitem)
# Find the Kodi Id
cursor.execute("SELECT kodi_id, kodi_file_id, media_type FROM emby WHERE emby_id = ?", (embyId,))
try:
result = cursor.fetchone()
kodiid = result[0]
fileid = result[1]
mediatype = result[2]
self.logMsg("Found embyId: %s in database - kodiId: %s fileId: %s type: %s" % (embyId, kodiid, fileid, mediatype), 1)
except:
self.logMsg("Id: %s not found in the emby database table." % embyId, 1)
else:
if mediatype in ("movie", "episode"):
playcount = userdata['PlayCount']
dateplayed = userdata['LastPlayedDate']
# Set resume point and round to 6th decimal
resume = round(float(timeInfo.get('ResumeTime')), 6)
total = round(float(timeInfo.get('TotalTime')), 6)
jumpback = int(utils.settings('resumeJumpBack'))
if resume > jumpback:
# To avoid negative bookmark
resume = resume - jumpback
self.setKodiResumePoint(fileid, resume, total, cursor, playcount, dateplayed)
#update the checksum in emby table
query = "UPDATE emby SET checksum = ? WHERE emby_id = ?"
cursor.execute(query, (checksum, embyId))
if mediatype in ("movie", "tvshow"):
# Add to or remove from favorites tag
if userdata['Favorite']:
self.AddTagToMedia(kodiid, "Favorite %ss" % mediatype, mediatype, cursor)
else:
self.AddTagToMedia(kodiid, "Favorite %ss" % mediatype, mediatype, cursor, True)