From 922f4150a0c0082884a8607bd9a69ca2a0af679a Mon Sep 17 00:00:00 2001 From: tomkat83 Date: Mon, 13 Feb 2017 20:26:30 +0100 Subject: [PATCH] Fix getting unique id --- resources/lib/itemtypes.py | 8 +++++--- resources/lib/kodidb_functions.py | 9 ++++++--- 2 files changed, 11 insertions(+), 6 deletions(-) diff --git a/resources/lib/itemtypes.py b/resources/lib/itemtypes.py index 1b265cc1..e729e80a 100644 --- a/resources/lib/itemtypes.py +++ b/resources/lib/itemtypes.py @@ -323,7 +323,8 @@ class Movies(Items): votecount, ratingid) # update new uniqueid Kodi 17 - uniqueid = self.kodi_db.get_uniqueid(movieid) + uniqueid = self.kodi_db.get_uniqueid(movieid, + v.KODI_TYPE_MOVIE) self.kodi_db.update_uniqueid(movieid, v.KODI_TYPE_MOVIE, imdb, @@ -605,7 +606,7 @@ class TVShows(Items): None, # votecount ratingid) # update new uniqueid Kodi 17 - uniqueid = self.kodi_db.get_uniqueid(showid) + uniqueid = self.kodi_db.get_uniqueid(showid, v.KODI_TYPE_SHOW) self.kodi_db.update_uniqueid(showid, v.KODI_TYPE_SHOW, tvdb, @@ -927,7 +928,8 @@ class TVShows(Items): votecount, ratingid) # update new uniqueid Kodi 17 - uniqueid = self.kodi_db.get_uniqueid(episodeid) + uniqueid = self.kodi_db.get_uniqueid(episodeid, + v.KODI_TYPE_EPISODE) self.kodi_db.update_uniqueid(episodeid, v.KODI_TYPE_EPISODE, tvdb, diff --git a/resources/lib/kodidb_functions.py b/resources/lib/kodidb_functions.py index 7f5262fa..322588fc 100644 --- a/resources/lib/kodidb_functions.py +++ b/resources/lib/kodidb_functions.py @@ -1434,9 +1434,12 @@ class Kodidb_Functions(): ''' self.cursor.execute(query, (args)) - def get_uniqueid(self, media_id): - query = "SELECT uniqueid_id FROM uniqueid WHERE media_id = ?" - self.cursor.execute(query, (media_id,)) + def get_uniqueid(self, kodi_id, kodi_type): + query = ''' + SELECT uniqueid_id FROM uniqueid + WHERE media_id = ? AND media_type = ? + ''' + self.cursor.execute(query, (kodi_id, kodi_type)) try: uniqueid = self.cursor.fetchone()[0] except TypeError: