cffe8acb69
However, Kodi STILL tries to gather info from music files unsuccessfully (PMS http folder cannot be accessed)
1273 lines
No EOL
41 KiB
Python
1273 lines
No EOL
41 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
###############################################################################
|
|
|
|
import xbmc
|
|
|
|
import artwork
|
|
import clientinfo
|
|
from utils import logging, kodiSQL
|
|
|
|
###############################################################################
|
|
|
|
|
|
class GetKodiDB():
|
|
"""
|
|
Usage: with GetKodiDB(itemType) as kodi_db:
|
|
do stuff with kodi_db
|
|
|
|
Parameters:
|
|
itemType: itemtype for Kodi DB, e.g. 'video', 'music'
|
|
|
|
On exiting "with" (no matter what), commits get automatically committed
|
|
and the db gets closed
|
|
"""
|
|
def __init__(self, itemType):
|
|
self.itemType = itemType
|
|
|
|
def __enter__(self):
|
|
self.kodiconn = kodiSQL(self.itemType)
|
|
self.emby_db = Kodidb_Functions(self.kodiconn.cursor())
|
|
return self.emby_db
|
|
|
|
def __exit__(self, type, value, traceback):
|
|
self.kodiconn.commit()
|
|
self.kodiconn.close()
|
|
|
|
|
|
@logging
|
|
class Kodidb_Functions():
|
|
|
|
kodiversion = int(xbmc.getInfoLabel("System.BuildVersion")[:2])
|
|
|
|
def __init__(self, cursor):
|
|
|
|
self.cursor = cursor
|
|
|
|
self.clientInfo = clientinfo.ClientInfo()
|
|
self.artwork = artwork.Artwork()
|
|
|
|
def addPath(self, path, strHash=None):
|
|
# SQL won't return existing paths otherwise
|
|
if path is None:
|
|
path = ""
|
|
cursor = self.cursor
|
|
query = ' '.join((
|
|
|
|
"SELECT idPath",
|
|
"FROM path",
|
|
"WHERE strPath = ?"
|
|
))
|
|
cursor.execute(query, (path,))
|
|
try:
|
|
pathid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idPath),0) from path")
|
|
pathid = cursor.fetchone()[0] + 1
|
|
if strHash is None:
|
|
query = (
|
|
'''
|
|
INSERT INTO path(
|
|
idPath, strPath)
|
|
|
|
VALUES (?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (pathid, path))
|
|
else:
|
|
query = (
|
|
'''
|
|
INSERT INTO path(
|
|
idPath, strPath, strHash)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (pathid, path, strHash))
|
|
|
|
return pathid
|
|
|
|
def getPath(self, path):
|
|
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idPath",
|
|
"FROM path",
|
|
"WHERE strPath = ?"
|
|
))
|
|
cursor.execute(query, (path,))
|
|
try:
|
|
pathid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
pathid = None
|
|
|
|
return pathid
|
|
|
|
def addFile(self, filename, pathid):
|
|
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idFile",
|
|
"FROM files",
|
|
"WHERE strFilename = ?",
|
|
"AND idPath = ?"
|
|
))
|
|
cursor.execute(query, (filename, pathid,))
|
|
try:
|
|
fileid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idFile),0) from files")
|
|
fileid = cursor.fetchone()[0] + 1
|
|
query = (
|
|
'''
|
|
INSERT INTO files(
|
|
idFile, strFilename)
|
|
|
|
VALUES (?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (fileid, filename))
|
|
|
|
return fileid
|
|
|
|
def getFile(self, fileid):
|
|
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT strFilename",
|
|
"FROM files",
|
|
"WHERE idFile = ?"
|
|
))
|
|
cursor.execute(query, (fileid,))
|
|
try:
|
|
filename = cursor.fetchone()[0]
|
|
except TypeError:
|
|
filename = ""
|
|
|
|
return filename
|
|
|
|
def removeFile(self, path, filename):
|
|
|
|
pathid = self.getPath(path)
|
|
|
|
if pathid is not None:
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM files",
|
|
"WHERE idPath = ?",
|
|
"AND strFilename = ?"
|
|
))
|
|
self.cursor.execute(query, (pathid, filename,))
|
|
|
|
def addCountries(self, kodiid, countries, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
for country in countries:
|
|
query = ' '.join((
|
|
|
|
"SELECT country_id",
|
|
"FROM country",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (country,))
|
|
|
|
try:
|
|
country_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Country entry does not exists
|
|
cursor.execute("select coalesce(max(country_id),0) 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 country to media, processing: %s" % country, 2)
|
|
|
|
finally: # Assign country to content
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO country_link(
|
|
country_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (country_id, kodiid, mediatype))
|
|
else:
|
|
# Kodi Helix
|
|
for country in countries:
|
|
query = ' '.join((
|
|
|
|
"SELECT idCountry",
|
|
"FROM country",
|
|
"WHERE strCountry = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (country,))
|
|
|
|
try:
|
|
idCountry = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Country entry does not exists
|
|
cursor.execute("select coalesce(max(idCountry),0) from country")
|
|
idCountry = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO country(idCountry, strCountry) values(?, ?)"
|
|
cursor.execute(query, (idCountry, country))
|
|
self.logMsg("Add country to media, processing: %s" % country, 2)
|
|
|
|
finally:
|
|
# Only movies have a country field
|
|
if "movie" in mediatype:
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO countrylinkmovie(
|
|
idCountry, idMovie)
|
|
|
|
VALUES (?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (idCountry, kodiid))
|
|
|
|
def addPeople(self, kodiid, people, mediatype):
|
|
|
|
cursor = self.cursor
|
|
artwork = self.artwork
|
|
kodiversion = self.kodiversion
|
|
|
|
castorder = 1
|
|
for person in people:
|
|
|
|
name = person['Name']
|
|
type = person['Type']
|
|
thumb = person['imageurl']
|
|
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
if kodiversion in (15, 16, 17):
|
|
query = ' '.join((
|
|
|
|
"SELECT actor_id",
|
|
"FROM actor",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (name,))
|
|
|
|
try:
|
|
actorid = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Cast entry does not exists
|
|
cursor.execute("select coalesce(max(actor_id),0) from actor")
|
|
actorid = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO actor(actor_id, name) values(?, ?)"
|
|
cursor.execute(query, (actorid, name))
|
|
|
|
finally:
|
|
# Link person to content
|
|
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, kodiid, 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, kodiid, mediatype))
|
|
|
|
elif type in ("Writing", "Writer"):
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO writer_link(
|
|
actor_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (actorid, kodiid, mediatype))
|
|
|
|
elif "Artist" in type:
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO actor_link(
|
|
actor_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (actorid, kodiid, mediatype))
|
|
# Kodi Helix
|
|
else:
|
|
query = ' '.join((
|
|
|
|
"SELECT idActor",
|
|
"FROM actors",
|
|
"WHERE strActor = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (name,))
|
|
|
|
try:
|
|
actorid = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Cast entry does not exists
|
|
cursor.execute("select coalesce(max(idActor),0) from actors")
|
|
actorid = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO actors(idActor, strActor) values(?, ?)"
|
|
cursor.execute(query, (actorid, name))
|
|
|
|
finally:
|
|
# Link person to content
|
|
if "Actor" in type:
|
|
role = person.get('Role')
|
|
|
|
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 (?, ?, ?, ?)
|
|
'''
|
|
)
|
|
else: return # Item is invalid
|
|
|
|
cursor.execute(query, (actorid, kodiid, 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 (?, ?)
|
|
'''
|
|
)
|
|
else: return # Item is invalid
|
|
|
|
cursor.execute(query, (actorid, kodiid))
|
|
|
|
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 (?, ?)
|
|
'''
|
|
)
|
|
else: return # Item is invalid
|
|
|
|
cursor.execute(query, (actorid, kodiid))
|
|
|
|
elif "Artist" in type:
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO artistlinkmusicvideo(
|
|
idArtist, idMVideo)
|
|
|
|
VALUES (?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (actorid, kodiid))
|
|
|
|
# Add person image to art table
|
|
if thumb:
|
|
arttype = type.lower()
|
|
|
|
if "writing" in arttype:
|
|
arttype = "writer"
|
|
|
|
artwork.addOrUpdateArt(thumb, actorid, arttype, "thumb", cursor)
|
|
|
|
def addGenres(self, kodiid, genres, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Delete current genres for clean slate
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM genre_link",
|
|
"WHERE media_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype,))
|
|
|
|
# Add genres
|
|
for genre in genres:
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT genre_id",
|
|
"FROM genre",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (genre,))
|
|
|
|
try:
|
|
genre_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Create genre in database
|
|
cursor.execute("select coalesce(max(genre_id),0) from genre")
|
|
genre_id = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO genre(genre_id, name) values(?, ?)"
|
|
cursor.execute(query, (genre_id, genre))
|
|
|
|
finally:
|
|
# Assign genre to item
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO genre_link(
|
|
genre_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (genre_id, kodiid, mediatype))
|
|
else:
|
|
# Kodi Helix
|
|
# Delete current genres for clean slate
|
|
if "movie" in mediatype:
|
|
cursor.execute("DELETE FROM genrelinkmovie WHERE idMovie = ?", (kodiid,))
|
|
elif "tvshow" in mediatype:
|
|
cursor.execute("DELETE FROM genrelinktvshow WHERE idShow = ?", (kodiid,))
|
|
elif "musicvideo" in mediatype:
|
|
cursor.execute("DELETE FROM genrelinkmusicvideo WHERE idMVideo = ?", (kodiid,))
|
|
|
|
# Add genres
|
|
for genre in genres:
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idGenre",
|
|
"FROM genre",
|
|
"WHERE strGenre = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (genre,))
|
|
|
|
try:
|
|
idGenre = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Create genre in database
|
|
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
|
idGenre = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
|
cursor.execute(query, (idGenre, genre))
|
|
|
|
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 (?, ?)
|
|
'''
|
|
)
|
|
else: return # Item is invalid
|
|
|
|
cursor.execute(query, (idGenre, kodiid))
|
|
|
|
def addStudios(self, kodiid, studios, mediatype):
|
|
|
|
cursor = self.cursor
|
|
kodiversion = self.kodiversion
|
|
|
|
for studio in studios:
|
|
|
|
if kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
query = ' '.join((
|
|
|
|
"SELECT studio_id",
|
|
"FROM studio",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (studio,))
|
|
try:
|
|
studioid = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Studio does not exists.
|
|
cursor.execute("select coalesce(max(studio_id),0) from studio")
|
|
studioid = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO studio(studio_id, name) values(?, ?)"
|
|
cursor.execute(query, (studioid, studio))
|
|
|
|
finally: # Assign studio to item
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO studio_link(
|
|
studio_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
''')
|
|
cursor.execute(query, (studioid, kodiid, mediatype))
|
|
else:
|
|
# Kodi Helix
|
|
query = ' '.join((
|
|
|
|
"SELECT idstudio",
|
|
"FROM studio",
|
|
"WHERE strstudio = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (studio,))
|
|
try:
|
|
studioid = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Studio does not exists.
|
|
cursor.execute("select coalesce(max(idstudio),0) from studio")
|
|
studioid = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO studio(idstudio, strstudio) values(?, ?)"
|
|
cursor.execute(query, (studioid, studio))
|
|
|
|
finally: # Assign studio to item
|
|
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, (studioid, kodiid))
|
|
|
|
def addStreams(self, fileid, streamdetails, runtime):
|
|
|
|
cursor = self.cursor
|
|
|
|
# First remove any existing entries
|
|
cursor.execute("DELETE FROM streamdetails WHERE idFile = ?", (fileid,))
|
|
if streamdetails:
|
|
# Video details
|
|
for videotrack in streamdetails['video']:
|
|
query = (
|
|
'''
|
|
INSERT INTO streamdetails(
|
|
idFile, iStreamType, strVideoCodec, fVideoAspect,
|
|
iVideoWidth, iVideoHeight, iVideoDuration ,strStereoMode)
|
|
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (fileid, 0, videotrack['codec'],
|
|
videotrack['aspect'], videotrack['width'], videotrack['height'],
|
|
runtime ,videotrack['video3DFormat']))
|
|
|
|
# Audio details
|
|
for audiotrack in streamdetails['audio']:
|
|
query = (
|
|
'''
|
|
INSERT INTO streamdetails(
|
|
idFile, iStreamType, strAudioCodec, iAudioChannels, strAudioLanguage)
|
|
|
|
VALUES (?, ?, ?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (fileid, 1, audiotrack['codec'],
|
|
audiotrack['channels'], audiotrack['language']))
|
|
|
|
# Subtitles details
|
|
for subtitletrack in streamdetails['subtitle']:
|
|
query = (
|
|
'''
|
|
INSERT INTO streamdetails(
|
|
idFile, iStreamType, strSubtitleLanguage)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (fileid, 2, subtitletrack))
|
|
|
|
def getResumes(self):
|
|
"""
|
|
VIDEOS
|
|
|
|
Returns all Kodi idFile that have a resume point set (not unwatched
|
|
ones or items that have already been completely watched)
|
|
"""
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
"SELECT idFile",
|
|
"FROM bookmark"
|
|
))
|
|
try:
|
|
rows = cursor.execute(query)
|
|
except:
|
|
return []
|
|
ids = []
|
|
for row in rows:
|
|
ids.append(row[0])
|
|
return ids
|
|
|
|
def getUnplayedMusicItems(self):
|
|
"""
|
|
MUSIC
|
|
|
|
Returns all Kodi Item idFile that have not yet been completely played
|
|
"""
|
|
query = ' '.join((
|
|
"SELECT idSong",
|
|
"FROM song",
|
|
"WHERE iTimesPlayed IS NULL OR iTimesPlayed = ''"
|
|
))
|
|
try:
|
|
rows = self.cursor.execute(query)
|
|
except:
|
|
return []
|
|
ids = []
|
|
for row in rows:
|
|
ids.append(row[0])
|
|
return ids
|
|
|
|
def getUnplayedItems(self):
|
|
"""
|
|
VIDEOS
|
|
|
|
Returns all Kodi Item idFile that have not yet been completely played
|
|
"""
|
|
query = ' '.join((
|
|
"SELECT idFile",
|
|
"FROM files",
|
|
"WHERE playCount IS NULL OR playCount = ''"
|
|
))
|
|
try:
|
|
rows = self.cursor.execute(query)
|
|
except:
|
|
return []
|
|
ids = []
|
|
for row in rows:
|
|
ids.append(row[0])
|
|
return ids
|
|
|
|
def addPlaystate(self, fileid, resume_seconds, total_seconds, playcount, dateplayed):
|
|
|
|
cursor = self.cursor
|
|
|
|
# Delete existing resume point
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM bookmark",
|
|
"WHERE idFile = ?"
|
|
))
|
|
cursor.execute(query, (fileid,))
|
|
|
|
# Set watched count
|
|
query = ' '.join((
|
|
|
|
"UPDATE files",
|
|
"SET playCount = ?, lastPlayed = ?",
|
|
"WHERE idFile = ?"
|
|
))
|
|
cursor.execute(query, (playcount, dateplayed, fileid))
|
|
|
|
# Set the resume bookmark
|
|
if resume_seconds:
|
|
cursor.execute("select coalesce(max(idBookmark),0) from bookmark")
|
|
bookmarkId = cursor.fetchone()[0] + 1
|
|
query = (
|
|
'''
|
|
INSERT INTO bookmark(
|
|
idBookmark, idFile, timeInSeconds, totalTimeInSeconds, player, type)
|
|
|
|
VALUES (?, ?, ?, ?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (bookmarkId, fileid, resume_seconds, total_seconds,
|
|
"DVDPlayer", 1))
|
|
|
|
def addTags(self, kodiid, tags, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
# First, delete any existing tags associated to the id
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM tag_link",
|
|
"WHERE media_id = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype))
|
|
else:
|
|
# Kodi Helix
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM taglinks",
|
|
"WHERE idMedia = ?",
|
|
"AND media_type = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype))
|
|
|
|
# Add tags
|
|
for tag in tags:
|
|
self.addTag(kodiid, tag, mediatype)
|
|
|
|
def addTag(self, kodiid, tag, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
query = ' '.join((
|
|
|
|
"SELECT tag_id",
|
|
"FROM tag",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (tag,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Create the tag, because it does not exist
|
|
tag_id = self.createTag(tag)
|
|
self.logMsg("Adding tag: %s" % tag, 2)
|
|
|
|
finally:
|
|
# Assign tag to item
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO tag_link(
|
|
tag_id, media_id, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (tag_id, kodiid, mediatype))
|
|
else:
|
|
# Kodi Helix
|
|
query = ' '.join((
|
|
|
|
"SELECT idTag",
|
|
"FROM tag",
|
|
"WHERE strTag = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (tag,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
# Create the tag
|
|
tag_id = self.createTag(tag)
|
|
self.logMsg("Adding tag: %s" % tag, 2)
|
|
|
|
finally:
|
|
# Assign tag to item
|
|
query = (
|
|
'''
|
|
INSERT OR REPLACE INTO taglinks(
|
|
idTag, idMedia, media_type)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (tag_id, kodiid, mediatype))
|
|
|
|
def createTag(self, name):
|
|
|
|
cursor = self.cursor
|
|
|
|
# This will create and return the tag_id
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
query = ' '.join((
|
|
|
|
"SELECT tag_id",
|
|
"FROM tag",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (name,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(tag_id),0) from tag")
|
|
tag_id = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO tag(tag_id, name) values(?, ?)"
|
|
cursor.execute(query, (tag_id, name))
|
|
self.logMsg("Create tag_id: %s name: %s" % (tag_id, name), 2)
|
|
else:
|
|
# Kodi Helix
|
|
query = ' '.join((
|
|
|
|
"SELECT idTag",
|
|
"FROM tag",
|
|
"WHERE strTag = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (name,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idTag),0) from tag")
|
|
tag_id = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO tag(idTag, strTag) values(?, ?)"
|
|
cursor.execute(query, (tag_id, name))
|
|
self.logMsg("Create idTag: %s name: %s" % (tag_id, name), 2)
|
|
|
|
return tag_id
|
|
|
|
def updateTag(self, oldtag, newtag, kodiid, mediatype):
|
|
|
|
cursor = self.cursor
|
|
self.logMsg("Updating: %s with %s for %s: %s" % (oldtag, newtag, mediatype, kodiid), 2)
|
|
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
try:
|
|
query = ' '.join((
|
|
|
|
"UPDATE tag_link",
|
|
"SET tag_id = ?",
|
|
"WHERE media_id = ?",
|
|
"AND media_type = ?",
|
|
"AND tag_id = ?"
|
|
))
|
|
cursor.execute(query, (newtag, kodiid, mediatype, oldtag,))
|
|
except Exception as e:
|
|
# The new tag we are going to apply already exists for this item
|
|
# delete current tag instead
|
|
self.logMsg("Exception: %s" % e, 1)
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM tag_link",
|
|
"WHERE media_id = ?",
|
|
"AND media_type = ?",
|
|
"AND tag_id = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype, oldtag,))
|
|
else:
|
|
# Kodi Helix
|
|
try:
|
|
query = ' '.join((
|
|
|
|
"UPDATE taglinks",
|
|
"SET idTag = ?",
|
|
"WHERE idMedia = ?",
|
|
"AND media_type = ?",
|
|
"AND idTag = ?"
|
|
))
|
|
cursor.execute(query, (newtag, kodiid, mediatype, oldtag,))
|
|
except Exception as e:
|
|
# The new tag we are going to apply already exists for this item
|
|
# delete current tag instead
|
|
self.logMsg("Exception: %s" % e, 1)
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM taglinks",
|
|
"WHERE idMedia = ?",
|
|
"AND media_type = ?",
|
|
"AND idTag = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype, oldtag,))
|
|
|
|
def removeTag(self, kodiid, tagname, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
if self.kodiversion in (15, 16, 17):
|
|
# Kodi Isengard, Jarvis, Krypton
|
|
query = ' '.join((
|
|
|
|
"SELECT tag_id",
|
|
"FROM tag",
|
|
"WHERE name = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (tagname,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
except TypeError:
|
|
return
|
|
else:
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM tag_link",
|
|
"WHERE media_id = ?",
|
|
"AND media_type = ?",
|
|
"AND tag_id = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype, tag_id,))
|
|
else:
|
|
# Kodi Helix
|
|
query = ' '.join((
|
|
|
|
"SELECT idTag",
|
|
"FROM tag",
|
|
"WHERE strTag = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (tagname,))
|
|
try:
|
|
tag_id = cursor.fetchone()[0]
|
|
except TypeError:
|
|
return
|
|
else:
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM taglinks",
|
|
"WHERE idMedia = ?",
|
|
"AND media_type = ?",
|
|
"AND idTag = ?"
|
|
))
|
|
cursor.execute(query, (kodiid, mediatype, tag_id,))
|
|
|
|
def createBoxset(self, boxsetname):
|
|
|
|
cursor = self.cursor
|
|
self.logMsg("Adding boxset: %s" % boxsetname, 2)
|
|
query = ' '.join((
|
|
|
|
"SELECT idSet",
|
|
"FROM sets",
|
|
"WHERE strSet = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (boxsetname,))
|
|
try:
|
|
setid = cursor.fetchone()[0]
|
|
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idSet),0) from sets")
|
|
setid = cursor.fetchone()[0] + 1
|
|
|
|
query = "INSERT INTO sets(idSet, strSet) values(?, ?)"
|
|
cursor.execute(query, (setid, boxsetname))
|
|
|
|
return setid
|
|
|
|
def assignBoxset(self, setid, movieid):
|
|
|
|
query = ' '.join((
|
|
|
|
"UPDATE movie",
|
|
"SET idSet = ?",
|
|
"WHERE idMovie = ?"
|
|
))
|
|
self.cursor.execute(query, (setid, movieid,))
|
|
|
|
def removefromBoxset(self, movieid):
|
|
|
|
query = ' '.join((
|
|
|
|
"UPDATE movie",
|
|
"SET idSet = null",
|
|
"WHERE idMovie = ?"
|
|
))
|
|
self.cursor.execute(query, (movieid,))
|
|
|
|
def addSeason(self, showid, seasonnumber):
|
|
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idSeason",
|
|
"FROM seasons",
|
|
"WHERE idShow = ?",
|
|
"AND season = ?"
|
|
))
|
|
cursor.execute(query, (showid, seasonnumber,))
|
|
try:
|
|
seasonid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idSeason),0) from seasons")
|
|
seasonid = cursor.fetchone()[0] + 1
|
|
query = "INSERT INTO seasons(idSeason, idShow, season) values(?, ?, ?)"
|
|
cursor.execute(query, (seasonid, showid, seasonnumber))
|
|
|
|
return seasonid
|
|
|
|
def addArtist(self, name, musicbrainz):
|
|
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idArtist, strArtist",
|
|
"FROM artist",
|
|
"WHERE strMusicBrainzArtistID = ?"
|
|
))
|
|
cursor.execute(query, (musicbrainz,))
|
|
try:
|
|
result = cursor.fetchone()
|
|
artistid = result[0]
|
|
artistname = result[1]
|
|
|
|
except TypeError:
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idArtist",
|
|
"FROM artist",
|
|
"WHERE strArtist = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (name,))
|
|
try:
|
|
artistid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
cursor.execute("select coalesce(max(idArtist),0) from artist")
|
|
artistid = cursor.fetchone()[0] + 1
|
|
query = (
|
|
'''
|
|
INSERT INTO artist(idArtist, strArtist, strMusicBrainzArtistID)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (artistid, name, musicbrainz))
|
|
else:
|
|
if artistname != name:
|
|
query = "UPDATE artist SET strArtist = ? WHERE idArtist = ?"
|
|
cursor.execute(query, (name, artistid,))
|
|
|
|
return artistid
|
|
|
|
def addAlbum(self, name, musicbrainz):
|
|
|
|
kodiversion = self.kodiversion
|
|
cursor = self.cursor
|
|
|
|
query = ' '.join((
|
|
|
|
"SELECT idAlbum",
|
|
"FROM album",
|
|
"WHERE strMusicBrainzAlbumID = ?"
|
|
))
|
|
cursor.execute(query, (musicbrainz,))
|
|
try:
|
|
albumid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
# Create the album
|
|
cursor.execute("select coalesce(max(idAlbum),0) from album")
|
|
albumid = cursor.fetchone()[0] + 1
|
|
if kodiversion in (15, 16, 17):
|
|
query = (
|
|
'''
|
|
INSERT INTO album(idAlbum, strAlbum, strMusicBrainzAlbumID, strReleaseType)
|
|
|
|
VALUES (?, ?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (albumid, name, musicbrainz, "album"))
|
|
else: # Helix
|
|
query = (
|
|
'''
|
|
INSERT INTO album(idAlbum, strAlbum, strMusicBrainzAlbumID)
|
|
|
|
VALUES (?, ?, ?)
|
|
'''
|
|
)
|
|
cursor.execute(query, (albumid, name, musicbrainz))
|
|
|
|
return albumid
|
|
|
|
def addMusicGenres(self, kodiid, genres, mediatype):
|
|
|
|
cursor = self.cursor
|
|
|
|
if mediatype == "album":
|
|
|
|
# Delete current genres for clean slate
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM album_genre",
|
|
"WHERE idAlbum = ?"
|
|
))
|
|
cursor.execute(query, (kodiid,))
|
|
|
|
for genre in genres:
|
|
query = ' '.join((
|
|
|
|
"SELECT idGenre",
|
|
"FROM genre",
|
|
"WHERE strGenre = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (genre,))
|
|
try:
|
|
genreid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
# Create the genre
|
|
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
|
genreid = cursor.fetchone()[0] + 1
|
|
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
|
cursor.execute(query, (genreid, genre))
|
|
|
|
query = "INSERT OR REPLACE INTO album_genre(idGenre, idAlbum) values(?, ?)"
|
|
cursor.execute(query, (genreid, kodiid))
|
|
|
|
elif mediatype == "song":
|
|
|
|
# Delete current genres for clean slate
|
|
query = ' '.join((
|
|
|
|
"DELETE FROM song_genre",
|
|
"WHERE idSong = ?"
|
|
))
|
|
cursor.execute(query, (kodiid,))
|
|
|
|
for genre in genres:
|
|
query = ' '.join((
|
|
|
|
"SELECT idGenre",
|
|
"FROM genre",
|
|
"WHERE strGenre = ?",
|
|
"COLLATE NOCASE"
|
|
))
|
|
cursor.execute(query, (genre,))
|
|
try:
|
|
genreid = cursor.fetchone()[0]
|
|
except TypeError:
|
|
# Create the genre
|
|
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
|
genreid = cursor.fetchone()[0] + 1
|
|
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
|
cursor.execute(query, (genreid, genre))
|
|
|
|
query = "INSERT OR REPLACE INTO song_genre(idGenre, idSong) values(?, ?)"
|
|
cursor.execute(query, (genreid, kodiid)) |