self cursor
This commit is contained in:
SpootDev
parent
6922eea15f
commit
a6a3da60f2
1 changed files with 1153 additions and 1190 deletions
|
|
@ -33,20 +33,18 @@ class Kodidb_Functions():
|
|||
|
||||
def addPath(self, path):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
query = ' '.join((
|
||||
|
||||
"SELECT idPath",
|
||||
"FROM path",
|
||||
"WHERE strPath = ?"
|
||||
))
|
||||
cursor.execute(query, (path,))
|
||||
self.cursor.execute(query, (path,))
|
||||
try:
|
||||
pathid = cursor.fetchone()[0]
|
||||
pathid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idPath),0) from path")
|
||||
pathid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idPath),0) from path")
|
||||
pathid = self.cursor.fetchone()[0] + 1
|
||||
query = (
|
||||
'''
|
||||
INSERT INTO path(
|
||||
|
|
@ -55,23 +53,21 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (pathid, path))
|
||||
self.cursor.execute(query, (pathid, path))
|
||||
|
||||
return pathid
|
||||
|
||||
def getPath(self, path):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
query = ' '.join((
|
||||
|
||||
"SELECT idPath",
|
||||
"FROM path",
|
||||
"WHERE strPath = ?"
|
||||
))
|
||||
cursor.execute(query, (path,))
|
||||
self.cursor.execute(query, (path,))
|
||||
try:
|
||||
pathid = cursor.fetchone()[0]
|
||||
pathid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
pathid = None
|
||||
|
||||
|
|
@ -79,8 +75,6 @@ class Kodidb_Functions():
|
|||
|
||||
def addFile(self, filename, pathid):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
query = ' '.join((
|
||||
|
||||
"SELECT idFile",
|
||||
|
|
@ -88,12 +82,12 @@ class Kodidb_Functions():
|
|||
"WHERE strFilename = ?",
|
||||
"AND idPath = ?"
|
||||
))
|
||||
cursor.execute(query, (filename, pathid,))
|
||||
self.cursor.execute(query, (filename, pathid,))
|
||||
try:
|
||||
fileid = cursor.fetchone()[0]
|
||||
fileid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idFile),0) from files")
|
||||
fileid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idFile),0) from files")
|
||||
fileid = self.cursor.fetchone()[0] + 1
|
||||
query = (
|
||||
'''
|
||||
INSERT INTO files(
|
||||
|
|
@ -102,23 +96,21 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (fileid, filename))
|
||||
self.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,))
|
||||
self.cursor.execute(query, (fileid,))
|
||||
try:
|
||||
filename = cursor.fetchone()[0]
|
||||
filename = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
filename = ""
|
||||
|
||||
|
|
@ -139,8 +131,6 @@ class Kodidb_Functions():
|
|||
|
||||
def addCountries(self, kodiid, countries, mediatype):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
if self.kodiversion in (15, 16, 17):
|
||||
# Kodi Isengard, Jarvis, Krypton
|
||||
for country in countries:
|
||||
|
|
@ -151,18 +141,18 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (country,))
|
||||
self.cursor.execute(query, (country,))
|
||||
|
||||
try:
|
||||
country_id = cursor.fetchone()[0]
|
||||
country_id = self.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
|
||||
self.cursor.execute("select coalesce(max(country_id),0) from country")
|
||||
country_id = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO country(country_id, name) values(?, ?)"
|
||||
cursor.execute(query, (country_id, country))
|
||||
self.cursor.execute(query, (country_id, country))
|
||||
self.logMsg("Add country to media, processing: %s" % country, 2)
|
||||
|
||||
finally: # Assign country to content
|
||||
|
|
@ -174,7 +164,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (country_id, kodiid, mediatype))
|
||||
self.cursor.execute(query, (country_id, kodiid, mediatype))
|
||||
else:
|
||||
# Kodi Helix
|
||||
for country in countries:
|
||||
|
|
@ -185,18 +175,18 @@ class Kodidb_Functions():
|
|||
"WHERE strCountry = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (country,))
|
||||
self.cursor.execute(query, (country,))
|
||||
|
||||
try:
|
||||
idCountry = cursor.fetchone()[0]
|
||||
idCountry = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Country entry does not exists
|
||||
cursor.execute("select coalesce(max(idCountry),0) from country")
|
||||
idCountry = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idCountry),0) from country")
|
||||
idCountry = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO country(idCountry, strCountry) values(?, ?)"
|
||||
cursor.execute(query, (idCountry, country))
|
||||
self.cursor.execute(query, (idCountry, country))
|
||||
self.logMsg("Add country to media, processing: %s" % country, 2)
|
||||
|
||||
finally:
|
||||
|
|
@ -210,14 +200,10 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (idCountry, kodiid))
|
||||
self.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:
|
||||
|
||||
|
|
@ -226,7 +212,7 @@ class Kodidb_Functions():
|
|||
thumb = person['imageurl']
|
||||
|
||||
# Kodi Isengard, Jarvis, Krypton
|
||||
if kodiversion in (15, 16, 17):
|
||||
if self.kodiversion in (15, 16, 17):
|
||||
query = ' '.join((
|
||||
|
||||
"SELECT actor_id",
|
||||
|
|
@ -234,18 +220,18 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (name,))
|
||||
self.cursor.execute(query, (name,))
|
||||
|
||||
try:
|
||||
actorid = cursor.fetchone()[0]
|
||||
actorid = self.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
|
||||
self.cursor.execute("select coalesce(max(actor_id),0) from actor")
|
||||
actorid = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO actor(actor_id, name) values(?, ?)"
|
||||
cursor.execute(query, (actorid, name))
|
||||
self.cursor.execute(query, (actorid, name))
|
||||
self.logMsg("Add people to media, processing: %s" % name, 2)
|
||||
|
||||
finally:
|
||||
|
|
@ -260,7 +246,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (actorid, kodiid, mediatype, role, castorder))
|
||||
self.cursor.execute(query, (actorid, kodiid, mediatype, role, castorder))
|
||||
castorder += 1
|
||||
|
||||
elif "Director" in type:
|
||||
|
|
@ -272,7 +258,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
self.cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
|
||||
elif type in ("Writing", "Writer"):
|
||||
query = (
|
||||
|
|
@ -283,7 +269,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
self.cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
|
||||
elif "Artist" in type:
|
||||
query = (
|
||||
|
|
@ -294,7 +280,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
self.cursor.execute(query, (actorid, kodiid, mediatype))
|
||||
# Kodi Helix
|
||||
else:
|
||||
query = ' '.join((
|
||||
|
|
@ -304,18 +290,18 @@ class Kodidb_Functions():
|
|||
"WHERE strActor = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (name,))
|
||||
self.cursor.execute(query, (name,))
|
||||
|
||||
try:
|
||||
actorid = cursor.fetchone()[0]
|
||||
actorid = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Cast entry does not exists
|
||||
cursor.execute("select coalesce(max(idActor),0) from actors")
|
||||
actorid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idActor),0) from actors")
|
||||
actorid = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO actors(idActor, strActor) values(?, ?)"
|
||||
cursor.execute(query, (actorid, name))
|
||||
self.cursor.execute(query, (actorid, name))
|
||||
self.logMsg("Add people to media, processing: %s" % name, 2)
|
||||
|
||||
finally:
|
||||
|
|
@ -352,7 +338,7 @@ class Kodidb_Functions():
|
|||
)
|
||||
else: return # Item is invalid
|
||||
|
||||
cursor.execute(query, (actorid, kodiid, role, castorder))
|
||||
self.cursor.execute(query, (actorid, kodiid, role, castorder))
|
||||
castorder += 1
|
||||
|
||||
elif "Director" in type:
|
||||
|
|
@ -395,7 +381,7 @@ class Kodidb_Functions():
|
|||
)
|
||||
else: return # Item is invalid
|
||||
|
||||
cursor.execute(query, (actorid, kodiid))
|
||||
self.cursor.execute(query, (actorid, kodiid))
|
||||
|
||||
elif type in ("Writing", "Writer"):
|
||||
if "movie" in mediatype:
|
||||
|
|
@ -418,7 +404,7 @@ class Kodidb_Functions():
|
|||
)
|
||||
else: return # Item is invalid
|
||||
|
||||
cursor.execute(query, (actorid, kodiid))
|
||||
self.cursor.execute(query, (actorid, kodiid))
|
||||
|
||||
elif "Artist" in type:
|
||||
query = (
|
||||
|
|
@ -429,7 +415,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (actorid, kodiid))
|
||||
self.cursor.execute(query, (actorid, kodiid))
|
||||
|
||||
# Add person image to art table
|
||||
if thumb:
|
||||
|
|
@ -438,11 +424,10 @@ class Kodidb_Functions():
|
|||
if "writing" in arttype:
|
||||
arttype = "writer"
|
||||
|
||||
artwork.addOrUpdateArt(thumb, actorid, arttype, "thumb", cursor)
|
||||
self.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):
|
||||
|
|
@ -453,7 +438,7 @@ class Kodidb_Functions():
|
|||
"WHERE media_id = ?",
|
||||
"AND media_type = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype,))
|
||||
self.cursor.execute(query, (kodiid, mediatype,))
|
||||
|
||||
# Add genres
|
||||
for genre in genres:
|
||||
|
|
@ -465,18 +450,18 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (genre,))
|
||||
self.cursor.execute(query, (genre,))
|
||||
|
||||
try:
|
||||
genre_id = cursor.fetchone()[0]
|
||||
genre_id = self.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
|
||||
self.cursor.execute("select coalesce(max(genre_id),0) from genre")
|
||||
genre_id = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO genre(genre_id, name) values(?, ?)"
|
||||
cursor.execute(query, (genre_id, genre))
|
||||
self.cursor.execute(query, (genre_id, genre))
|
||||
self.logMsg("Add Genres to media, processing: %s" % genre, 2)
|
||||
|
||||
finally:
|
||||
|
|
@ -489,16 +474,16 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (genre_id, kodiid, mediatype))
|
||||
self.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,))
|
||||
self.cursor.execute("DELETE FROM genrelinkmovie WHERE idMovie = ?", (kodiid,))
|
||||
elif "tvshow" in mediatype:
|
||||
cursor.execute("DELETE FROM genrelinktvshow WHERE idShow = ?", (kodiid,))
|
||||
self.cursor.execute("DELETE FROM genrelinktvshow WHERE idShow = ?", (kodiid,))
|
||||
elif "musicvideo" in mediatype:
|
||||
cursor.execute("DELETE FROM genrelinkmusicvideo WHERE idMVideo = ?", (kodiid,))
|
||||
self.cursor.execute("DELETE FROM genrelinkmusicvideo WHERE idMVideo = ?", (kodiid,))
|
||||
|
||||
# Add genres
|
||||
for genre in genres:
|
||||
|
|
@ -510,18 +495,18 @@ class Kodidb_Functions():
|
|||
"WHERE strGenre = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (genre,))
|
||||
self.cursor.execute(query, (genre,))
|
||||
|
||||
try:
|
||||
idGenre = cursor.fetchone()[0]
|
||||
idGenre = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Create genre in database
|
||||
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
idGenre = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
idGenre = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
||||
cursor.execute(query, (idGenre, genre))
|
||||
self.cursor.execute(query, (idGenre, genre))
|
||||
self.logMsg("Add Genres to media, processing: %s" % genre, 2)
|
||||
|
||||
finally:
|
||||
|
|
@ -555,11 +540,10 @@ class Kodidb_Functions():
|
|||
)
|
||||
else: return # Item is invalid
|
||||
|
||||
cursor.execute(query, (idGenre, kodiid))
|
||||
self.cursor.execute(query, (idGenre, kodiid))
|
||||
|
||||
def addStudios(self, kodiid, studios, mediatype):
|
||||
|
||||
cursor = self.cursor
|
||||
kodiversion = self.kodiversion
|
||||
|
||||
for studio in studios:
|
||||
|
|
@ -573,17 +557,17 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (studio,))
|
||||
self.cursor.execute(query, (studio,))
|
||||
try:
|
||||
studioid = cursor.fetchone()[0]
|
||||
studioid = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Studio does not exists.
|
||||
cursor.execute("select coalesce(max(studio_id),0) from studio")
|
||||
studioid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(studio_id),0) from studio")
|
||||
studioid = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO studio(studio_id, name) values(?, ?)"
|
||||
cursor.execute(query, (studioid, studio))
|
||||
self.cursor.execute(query, (studioid, studio))
|
||||
self.logMsg("Add Studios to media, processing: %s" % studio, 2)
|
||||
|
||||
finally: # Assign studio to item
|
||||
|
|
@ -594,7 +578,7 @@ class Kodidb_Functions():
|
|||
|
||||
VALUES (?, ?, ?)
|
||||
''')
|
||||
cursor.execute(query, (studioid, kodiid, mediatype))
|
||||
self.cursor.execute(query, (studioid, kodiid, mediatype))
|
||||
else:
|
||||
# Kodi Helix
|
||||
query = ' '.join((
|
||||
|
|
@ -604,17 +588,17 @@ class Kodidb_Functions():
|
|||
"WHERE strstudio = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (studio,))
|
||||
self.cursor.execute(query, (studio,))
|
||||
try:
|
||||
studioid = cursor.fetchone()[0]
|
||||
studioid = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Studio does not exists.
|
||||
cursor.execute("select coalesce(max(idstudio),0) from studio")
|
||||
studioid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idstudio),0) from studio")
|
||||
studioid = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO studio(idstudio, strstudio) values(?, ?)"
|
||||
cursor.execute(query, (studioid, studio))
|
||||
self.cursor.execute(query, (studioid, studio))
|
||||
self.logMsg("Add Studios to media, processing: %s" % studio, 2)
|
||||
|
||||
finally: # Assign studio to item
|
||||
|
|
@ -642,14 +626,12 @@ class Kodidb_Functions():
|
|||
INSERT OR REPLACE INTO studiolinkepisode(idstudio, idEpisode)
|
||||
VALUES (?, ?)
|
||||
''')
|
||||
cursor.execute(query, (studioid, kodiid))
|
||||
self.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,))
|
||||
self.cursor.execute("DELETE FROM streamdetails WHERE idFile = ?", (fileid,))
|
||||
if streamdetails:
|
||||
# Video details
|
||||
for videotrack in streamdetails['video']:
|
||||
|
|
@ -662,7 +644,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (fileid, 0, videotrack['codec'],
|
||||
self.cursor.execute(query, (fileid, 0, videotrack['codec'],
|
||||
videotrack['aspect'], videotrack['width'], videotrack['height'],
|
||||
runtime ,videotrack['video3DFormat']))
|
||||
|
||||
|
|
@ -676,7 +658,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (fileid, 1, audiotrack['codec'],
|
||||
self.cursor.execute(query, (fileid, 1, audiotrack['codec'],
|
||||
audiotrack['channels'], audiotrack['language']))
|
||||
|
||||
# Subtitles details
|
||||
|
|
@ -689,19 +671,17 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (fileid, 2, subtitletrack))
|
||||
self.cursor.execute(query, (fileid, 2, subtitletrack))
|
||||
|
||||
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,))
|
||||
self.cursor.execute(query, (fileid,))
|
||||
|
||||
# Set watched count
|
||||
query = ' '.join((
|
||||
|
|
@ -710,12 +690,12 @@ class Kodidb_Functions():
|
|||
"SET playCount = ?, lastPlayed = ?",
|
||||
"WHERE idFile = ?"
|
||||
))
|
||||
cursor.execute(query, (playcount, dateplayed, fileid))
|
||||
self.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
|
||||
self.cursor.execute("select coalesce(max(idBookmark),0) from bookmark")
|
||||
bookmarkId = self.cursor.fetchone()[0] + 1
|
||||
query = (
|
||||
'''
|
||||
INSERT INTO bookmark(
|
||||
|
|
@ -724,13 +704,11 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?, ?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (bookmarkId, fileid, resume_seconds, total_seconds,
|
||||
self.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
|
||||
|
|
@ -740,7 +718,7 @@ class Kodidb_Functions():
|
|||
"WHERE media_id = ?",
|
||||
"AND media_type = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype))
|
||||
self.cursor.execute(query, (kodiid, mediatype))
|
||||
else:
|
||||
# Kodi Helix
|
||||
query = ' '.join((
|
||||
|
|
@ -749,7 +727,7 @@ class Kodidb_Functions():
|
|||
"WHERE idMedia = ?",
|
||||
"AND media_type = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype))
|
||||
self.cursor.execute(query, (kodiid, mediatype))
|
||||
|
||||
# Add tags
|
||||
self.logMsg("Adding Tags: %s" % tags, 2)
|
||||
|
|
@ -758,8 +736,6 @@ class Kodidb_Functions():
|
|||
|
||||
def addTag(self, kodiid, tag, mediatype):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
if self.kodiversion in (15, 16, 17):
|
||||
# Kodi Isengard, Jarvis, Krypton
|
||||
query = ' '.join((
|
||||
|
|
@ -769,9 +745,9 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (tag,))
|
||||
self.cursor.execute(query, (tag,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Create the tag, because it does not exist
|
||||
|
|
@ -788,7 +764,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (tag_id, kodiid, mediatype))
|
||||
self.cursor.execute(query, (tag_id, kodiid, mediatype))
|
||||
else:
|
||||
# Kodi Helix
|
||||
query = ' '.join((
|
||||
|
|
@ -798,9 +774,9 @@ class Kodidb_Functions():
|
|||
"WHERE strTag = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (tag,))
|
||||
self.cursor.execute(query, (tag,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
# Create the tag
|
||||
|
|
@ -817,12 +793,10 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (tag_id, kodiid, mediatype))
|
||||
self.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
|
||||
|
|
@ -833,16 +807,16 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (name,))
|
||||
self.cursor.execute(query, (name,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(tag_id),0) from tag")
|
||||
tag_id = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(tag_id),0) from tag")
|
||||
tag_id = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO tag(tag_id, name) values(?, ?)"
|
||||
cursor.execute(query, (tag_id, name))
|
||||
self.cursor.execute(query, (tag_id, name))
|
||||
self.logMsg("Create tag_id: %s name: %s" % (tag_id, name), 2)
|
||||
else:
|
||||
# Kodi Helix
|
||||
|
|
@ -853,23 +827,22 @@ class Kodidb_Functions():
|
|||
"WHERE strTag = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (name,))
|
||||
self.cursor.execute(query, (name,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idTag),0) from tag")
|
||||
tag_id = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idTag),0) from tag")
|
||||
tag_id = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO tag(idTag, strTag) values(?, ?)"
|
||||
cursor.execute(query, (tag_id, name))
|
||||
self.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):
|
||||
|
|
@ -883,7 +856,7 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND tag_id = ?"
|
||||
))
|
||||
cursor.execute(query, (newtag, kodiid, mediatype, oldtag,))
|
||||
self.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
|
||||
|
|
@ -895,7 +868,7 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND tag_id = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype, oldtag,))
|
||||
self.cursor.execute(query, (kodiid, mediatype, oldtag,))
|
||||
else:
|
||||
# Kodi Helix
|
||||
try:
|
||||
|
|
@ -907,7 +880,7 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND idTag = ?"
|
||||
))
|
||||
cursor.execute(query, (newtag, kodiid, mediatype, oldtag,))
|
||||
self.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
|
||||
|
|
@ -919,12 +892,10 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND idTag = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype, oldtag,))
|
||||
self.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((
|
||||
|
|
@ -934,9 +905,9 @@ class Kodidb_Functions():
|
|||
"WHERE name = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (tagname,))
|
||||
self.cursor.execute(query, (tagname,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
return
|
||||
else:
|
||||
|
|
@ -947,7 +918,7 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND tag_id = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype, tag_id,))
|
||||
self.cursor.execute(query, (kodiid, mediatype, tag_id,))
|
||||
else:
|
||||
# Kodi Helix
|
||||
query = ' '.join((
|
||||
|
|
@ -957,9 +928,9 @@ class Kodidb_Functions():
|
|||
"WHERE strTag = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (tagname,))
|
||||
self.cursor.execute(query, (tagname,))
|
||||
try:
|
||||
tag_id = cursor.fetchone()[0]
|
||||
tag_id = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
return
|
||||
else:
|
||||
|
|
@ -970,11 +941,10 @@ class Kodidb_Functions():
|
|||
"AND media_type = ?",
|
||||
"AND idTag = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid, mediatype, tag_id,))
|
||||
self.cursor.execute(query, (kodiid, mediatype, tag_id,))
|
||||
|
||||
def createBoxset(self, boxsetname):
|
||||
|
||||
cursor = self.cursor
|
||||
self.logMsg("Adding boxset: %s" % boxsetname, 2)
|
||||
query = ' '.join((
|
||||
|
||||
|
|
@ -983,16 +953,16 @@ class Kodidb_Functions():
|
|||
"WHERE strSet = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (boxsetname,))
|
||||
self.cursor.execute(query, (boxsetname,))
|
||||
try:
|
||||
setid = cursor.fetchone()[0]
|
||||
setid = self.cursor.fetchone()[0]
|
||||
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idSet),0) from sets")
|
||||
setid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idSet),0) from sets")
|
||||
setid = self.cursor.fetchone()[0] + 1
|
||||
|
||||
query = "INSERT INTO sets(idSet, strSet) values(?, ?)"
|
||||
cursor.execute(query, (setid, boxsetname))
|
||||
self.cursor.execute(query, (setid, boxsetname))
|
||||
|
||||
return setid
|
||||
|
||||
|
|
@ -1018,8 +988,6 @@ class Kodidb_Functions():
|
|||
|
||||
def addSeason(self, showid, seasonnumber):
|
||||
|
||||
cursor = self.cursor
|
||||
|
||||
query = ' '.join((
|
||||
|
||||
"SELECT idSeason",
|
||||
|
|
@ -1027,30 +995,28 @@ class Kodidb_Functions():
|
|||
"WHERE idShow = ?",
|
||||
"AND season = ?"
|
||||
))
|
||||
cursor.execute(query, (showid, seasonnumber,))
|
||||
self.cursor.execute(query, (showid, seasonnumber,))
|
||||
try:
|
||||
seasonid = cursor.fetchone()[0]
|
||||
seasonid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idSeason),0) from seasons")
|
||||
seasonid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idSeason),0) from seasons")
|
||||
seasonid = self.cursor.fetchone()[0] + 1
|
||||
query = "INSERT INTO seasons(idSeason, idShow, season) values(?, ?, ?)"
|
||||
cursor.execute(query, (seasonid, showid, seasonnumber))
|
||||
self.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,))
|
||||
self.cursor.execute(query, (musicbrainz,))
|
||||
try:
|
||||
result = cursor.fetchone()
|
||||
result = self.cursor.fetchone()
|
||||
artistid = result[0]
|
||||
artistname = result[1]
|
||||
|
||||
|
|
@ -1063,12 +1029,12 @@ class Kodidb_Functions():
|
|||
"WHERE strArtist = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (name,))
|
||||
self.cursor.execute(query, (name,))
|
||||
try:
|
||||
artistid = cursor.fetchone()[0]
|
||||
artistid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
cursor.execute("select coalesce(max(idArtist),0) from artist")
|
||||
artistid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idArtist),0) from artist")
|
||||
artistid = self.cursor.fetchone()[0] + 1
|
||||
query = (
|
||||
'''
|
||||
INSERT INTO artist(idArtist, strArtist, strMusicBrainzArtistID)
|
||||
|
|
@ -1076,18 +1042,17 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (artistid, name, musicbrainz))
|
||||
self.cursor.execute(query, (artistid, name, musicbrainz))
|
||||
else:
|
||||
if artistname != name:
|
||||
query = "UPDATE artist SET strArtist = ? WHERE idArtist = ?"
|
||||
cursor.execute(query, (name, artistid,))
|
||||
self.cursor.execute(query, (name, artistid,))
|
||||
|
||||
return artistid
|
||||
|
||||
def addAlbum(self, name, musicbrainz):
|
||||
|
||||
kodiversion = self.kodiversion
|
||||
cursor = self.cursor
|
||||
|
||||
query = ' '.join((
|
||||
|
||||
|
|
@ -1095,13 +1060,13 @@ class Kodidb_Functions():
|
|||
"FROM album",
|
||||
"WHERE strMusicBrainzAlbumID = ?"
|
||||
))
|
||||
cursor.execute(query, (musicbrainz,))
|
||||
self.cursor.execute(query, (musicbrainz,))
|
||||
try:
|
||||
albumid = cursor.fetchone()[0]
|
||||
albumid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
# Create the album
|
||||
cursor.execute("select coalesce(max(idAlbum),0) from album")
|
||||
albumid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idAlbum),0) from album")
|
||||
albumid = self.cursor.fetchone()[0] + 1
|
||||
if kodiversion in (15, 16, 17):
|
||||
query = (
|
||||
'''
|
||||
|
|
@ -1110,7 +1075,7 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (albumid, name, musicbrainz, "album"))
|
||||
self.cursor.execute(query, (albumid, name, musicbrainz, "album"))
|
||||
else: # Helix
|
||||
query = (
|
||||
'''
|
||||
|
|
@ -1119,14 +1084,12 @@ class Kodidb_Functions():
|
|||
VALUES (?, ?, ?)
|
||||
'''
|
||||
)
|
||||
cursor.execute(query, (albumid, name, musicbrainz))
|
||||
self.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
|
||||
|
|
@ -1135,7 +1098,7 @@ class Kodidb_Functions():
|
|||
"DELETE FROM album_genre",
|
||||
"WHERE idAlbum = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid,))
|
||||
self.cursor.execute(query, (kodiid,))
|
||||
|
||||
for genre in genres:
|
||||
query = ' '.join((
|
||||
|
|
@ -1145,18 +1108,18 @@ class Kodidb_Functions():
|
|||
"WHERE strGenre = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (genre,))
|
||||
self.cursor.execute(query, (genre,))
|
||||
try:
|
||||
genreid = cursor.fetchone()[0]
|
||||
genreid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
# Create the genre
|
||||
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
genreid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
genreid = self.cursor.fetchone()[0] + 1
|
||||
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
||||
cursor.execute(query, (genreid, genre))
|
||||
self.cursor.execute(query, (genreid, genre))
|
||||
|
||||
query = "INSERT OR REPLACE INTO album_genre(idGenre, idAlbum) values(?, ?)"
|
||||
cursor.execute(query, (genreid, kodiid))
|
||||
self.cursor.execute(query, (genreid, kodiid))
|
||||
|
||||
elif mediatype == "song":
|
||||
|
||||
|
|
@ -1166,7 +1129,7 @@ class Kodidb_Functions():
|
|||
"DELETE FROM song_genre",
|
||||
"WHERE idSong = ?"
|
||||
))
|
||||
cursor.execute(query, (kodiid,))
|
||||
self.cursor.execute(query, (kodiid,))
|
||||
|
||||
for genre in genres:
|
||||
query = ' '.join((
|
||||
|
|
@ -1176,15 +1139,15 @@ class Kodidb_Functions():
|
|||
"WHERE strGenre = ?",
|
||||
"COLLATE NOCASE"
|
||||
))
|
||||
cursor.execute(query, (genre,))
|
||||
self.cursor.execute(query, (genre,))
|
||||
try:
|
||||
genreid = cursor.fetchone()[0]
|
||||
genreid = self.cursor.fetchone()[0]
|
||||
except TypeError:
|
||||
# Create the genre
|
||||
cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
genreid = cursor.fetchone()[0] + 1
|
||||
self.cursor.execute("select coalesce(max(idGenre),0) from genre")
|
||||
genreid = self.cursor.fetchone()[0] + 1
|
||||
query = "INSERT INTO genre(idGenre, strGenre) values(?, ?)"
|
||||
cursor.execute(query, (genreid, genre))
|
||||
self.cursor.execute(query, (genreid, genre))
|
||||
|
||||
query = "INSERT OR REPLACE INTO song_genre(idGenre, idSong) values(?, ?)"
|
||||
cursor.execute(query, (genreid, kodiid))
|
||||
self.cursor.execute(query, (genreid, kodiid))
|
||||
Loading…
Reference in a new issue