#!/usr/bin/env python # -*- coding: utf-8 -*- from __future__ import absolute_import, division, unicode_literals from logging import getLogger from sqlite3 import IntegrityError from . import common from .. import db, path_ops, timing, variables as v LOG = getLogger('PLEX.kodi_db.video') MOVIE_PATH = 'plugin://%s.movies/' % v.ADDON_ID SHOW_PATH = 'plugin://%s.tvshows/' % v.ADDON_ID class KodiVideoDB(common.KodiDBBase): db_kind = 'video' @db.catch_operationalerrors def create_kodi_db_indicees(self): """ Index the "actors" because we got a TON - speed up SELECT and WHEN """ commands = ( 'CREATE UNIQUE INDEX IF NOT EXISTS ix_actor_2 ON actor (actor_id);', 'CREATE UNIQUE INDEX IF NOT EXISTS ix_files_2 ON files (idFile);', ) for cmd in commands: self.cursor.execute(cmd) @db.catch_operationalerrors def setup_path_table(self): """ Use with Kodi video DB Sets strContent to e.g. 'movies' and strScraper to metadata.local For some reason, Kodi ignores this if done via itemtypes while e.g. adding or updating items. (addPath method does NOT work) """ for path, kind in ((MOVIE_PATH, 'movies'), (SHOW_PATH, 'tvshows')): path_id = self.get_path(path) if path_id is None: query = ''' INSERT INTO path(strPath, strContent, strScraper, noUpdate, exclude) VALUES (?, ?, ?, ?, ?) ''' self.cursor.execute(query, (path, kind, 'metadata.local', 1, 0)) @db.catch_operationalerrors def parent_path_id(self, path): """ Video DB: Adds all subdirectories to path table while setting a "trail" of parent path ids """ parentpath = path_ops.path.abspath( path_ops.path.join(path, path_ops.decode_path(path_ops.path.pardir))) pathid = self.get_path(parentpath) if pathid is None: self.cursor.execute(''' INSERT INTO path(strPath, dateAdded) VALUES (?, ?) ''', (parentpath, timing.kodi_now())) pathid = self.cursor.lastrowid if parentpath != path: # In case we end up having media in the filesystem root, C:\ parent_id = self.parent_path_id(parentpath) self.update_parentpath_id(parent_id, pathid) return pathid @db.catch_operationalerrors def update_parentpath_id(self, parent_id, pathid): """ Dedicated method in order to catch OperationalErrors correctly """ self.cursor.execute('UPDATE path SET idParentPath = ? WHERE idPath = ?', (parent_id, pathid)) @db.catch_operationalerrors def add_path(self, path, date_added=None, id_parent_path=None, content=None, scraper=None): """ Returns the idPath from the path table. Creates a new entry if path [unicode] does not yet exist (using date_added [kodi date type], id_parent_path [int], content ['tvshows', 'movies', None], scraper [usually 'metadata.local']) WILL activate noUpdate for the path! """ path = '' if path is None else path self.cursor.execute('SELECT idPath FROM path WHERE strPath = ? LIMIT 1', (path, )) try: pathid = self.cursor.fetchone()[0] except TypeError: self.cursor.execute(''' INSERT INTO path( strPath, dateAdded, idParentPath, strContent, strScraper, noUpdate) VALUES (?, ?, ?, ?, ?, ?) ''', (path, date_added, id_parent_path, content, scraper, 1)) pathid = self.cursor.lastrowid return pathid def get_path(self, path): """ Returns the idPath from the path table for path [unicode] or None """ self.cursor.execute('SELECT idPath FROM path WHERE strPath = ?', (path, )) try: return self.cursor.fetchone()[0] except TypeError: pass @db.catch_operationalerrors def add_file(self, filename, path_id, date_added): """ Adds the filename [unicode] to the table files if not already added and returns the idFile. """ self.cursor.execute(''' INSERT INTO files(idPath, strFilename, dateAdded) VALUES (?, ?, ?) ''', (path_id, filename, date_added)) return self.cursor.lastrowid def modify_file(self, filename, path_id, date_added): self.cursor.execute('SELECT idFile FROM files WHERE idPath = ? AND strFilename = ?', (path_id, filename)) try: file_id = self.cursor.fetchone()[0] except TypeError: file_id = self.add_file(filename, path_id, date_added) return file_id def obsolete_file_ids(self): """ Returns a generator for idFile of all Kodi file ids that do not have a dateAdded set (dateAdded NULL) and the filename start with 'plugin://plugin.video.plexkodiconnect' These entries should be deleted as they're created falsely by Kodi. """ return (x[0] for x in self.cursor.execute(''' SELECT idFile FROM files WHERE dateAdded IS NULL AND strFilename LIKE \'plugin://plugin.video.plexkodiconnect%\' ''')) def show_id_from_path(self, path): """ Returns the idShow for path [unicode] or None """ self.cursor.execute('SELECT idPath FROM path WHERE strPath = ? LIMIT 1', (path, )) try: path_id = self.cursor.fetchone()[0] except TypeError: return self.cursor.execute('SELECT idShow FROM tvshowlinkpath WHERE idPath = ? LIMIT 1', (path_id, )) try: return self.cursor.fetchone()[0] except TypeError: pass @db.catch_operationalerrors def remove_file(self, file_id, remove_orphans=True): """ Removes the entry for file_id from the files table. Will also delete entries from the associated tables: bookmark, settings, streamdetails. If remove_orphans is true, this method will delete any orphaned path entries in the Kodi path table """ self.cursor.execute('SELECT idPath FROM files WHERE idFile = ? LIMIT 1', (file_id,)) try: path_id = self.cursor.fetchone()[0] except TypeError: return self.cursor.execute('DELETE FROM files WHERE idFile = ?', (file_id,)) self.cursor.execute('DELETE FROM bookmark WHERE idFile = ?', (file_id,)) self.cursor.execute('DELETE FROM settings WHERE idFile = ?', (file_id,)) self.cursor.execute('DELETE FROM streamdetails WHERE idFile = ?', (file_id,)) self.cursor.execute('DELETE FROM stacktimes WHERE idFile = ?', (file_id,)) if remove_orphans: # Delete orphaned path entry self.cursor.execute('SELECT idFile FROM files WHERE idPath = ? LIMIT 1', (path_id,)) if self.cursor.fetchone() is None: # Make sure we're not deleting our root paths! query = ''' DELETE FROM path WHERE idPath = ? AND strPath NOT IN (?, ?) ''' self.cursor.execute(query, (path_id, MOVIE_PATH, SHOW_PATH)) @db.catch_operationalerrors def _modify_link_and_table(self, kodi_id, kodi_type, entries, link_table, table, key, first_id=None): first_id = first_id if first_id is not None else 1 entry_ids = [] for entry in entries: self.cursor.execute(''' SELECT %s FROM %s WHERE name = ? COLLATE NOCASE LIMIT 1 ''' % (key, table), (entry, )) try: entry_id = self.cursor.fetchone()[0] except TypeError: self.cursor.execute('INSERT INTO %s(name) VALUES(?)' % table, (entry, )) entry_id = self.cursor.lastrowid finally: entry_ids.append(entry_id) # Now process the ids obtained from the names # Get the existing, old entries outdated_entries = [] for entry_id in self.cursor.execute('SELECT %s FROM %s WHERE media_id = ? AND media_type = ?' % (key, link_table), (kodi_id, kodi_type)): try: entry_ids.remove(entry_id[0]) except ValueError: outdated_entries.append(entry_id[0]) # Add all new entries that haven't already been added for entry_id in entry_ids: try: self.cursor.execute('INSERT INTO %s VALUES (?, ?, ?)' % link_table, (entry_id, kodi_id, kodi_type)) except IntegrityError: LOG.info('IntegrityError: skipping entry %s for table %s', entry_id, link_table) # Delete all outdated references in the link table. Also check whether # we need to delete orphaned entries in the master table for entry_id in outdated_entries: self.cursor.execute(''' DELETE FROM %s WHERE %s = ? AND media_id = ? AND media_type = ? ''' % (link_table, key), (entry_id, kodi_id, kodi_type)) self.cursor.execute('SELECT %s FROM %s WHERE %s = ?' % (key, link_table, key), (entry_id, )) if self.cursor.fetchone() is None: # Delete in the original table because entry is now orphaned self.cursor.execute('DELETE FROM %s WHERE %s = ?' % (table, key), (entry_id, )) def modify_countries(self, kodi_id, kodi_type, countries=None): """ Writes a country (string) in the list countries into the Kodi DB. Will also delete any orphaned country entries. """ self._modify_link_and_table(kodi_id, kodi_type, countries if countries else [], 'country_link', 'country', 'country_id') def modify_genres(self, kodi_id, kodi_type, genres=None): """ Writes a country (string) in the list countries into the Kodi DB. Will also delete any orphaned country entries. """ self._modify_link_and_table(kodi_id, kodi_type, genres if genres else [], 'genre_link', 'genre', 'genre_id') def modify_studios(self, kodi_id, kodi_type, studios=None): """ Writes a country (string) in the list countries into the Kodi DB. Will also delete any orphaned country entries. """ self._modify_link_and_table(kodi_id, kodi_type, studios if studios else [], 'studio_link', 'studio', 'studio_id') def modify_tags(self, kodi_id, kodi_type, tags=None): """ Writes a country (string) in the list countries into the Kodi DB. Will also delete any orphaned country entries. """ self._modify_link_and_table(kodi_id, kodi_type, tags if tags else [], 'tag_link', 'tag', 'tag_id') def add_people(self, kodi_id, kodi_type, people): """ Makes sure that actors, directors and writers are recorded correctly for the elmement kodi_id, kodi_type. Will also delete a freshly orphaned actor entry. """ for kind, people_list in people.iteritems(): self._add_people_kind(kodi_id, kodi_type, kind, people_list) @db.catch_operationalerrors def _add_people_kind(self, kodi_id, kodi_type, kind, people_list): # Save new people to Kodi DB by iterating over the remaining entries if kind == 'actor': for person in people_list: # Make sure the person entry in table actor exists actor_id, new = self._get_actor_id(person[0], art_url=person[1]) if not new and person[1]: # Person might have shown up as a director or writer first # WITHOUT an art url from the Plex side! # Check here if we need to set the actor's art url self._check_actor_art(actor_id, person[1]) # Link the person with the media element try: self.cursor.execute('INSERT INTO actor_link VALUES (?, ?, ?, ?, ?)', (actor_id, kodi_id, kodi_type, person[2], person[3])) except IntegrityError: # With Kodi, an actor may have only one role, unlike Plex pass else: for person in people_list: # Make sure the person entry in table actor exists: actor_id, _ = self._get_actor_id(person[0]) # Link the person with the media element try: self.cursor.execute('INSERT INTO %s_link VALUES (?, ?, ?)' % kind, (actor_id, kodi_id, kodi_type)) except IntegrityError: # Again, Kodi may have only one person assigned to a role pass def modify_people(self, kodi_id, kodi_type, people=None): """ Makes sure that actors, directors and writers are recorded correctly for the elmement kodi_id, kodi_type. Will also delete a freshly orphaned actor entry. """ for kind, people_list in (people if people else {'actor': [], 'director': [], 'writer': []}).iteritems(): self._modify_people_kind(kodi_id, kodi_type, kind, people_list) @db.catch_operationalerrors def _modify_people_kind(self, kodi_id, kodi_type, kind, people_list): # Get the people already saved in the DB for this specific item if kind == 'actor': query = ''' SELECT actor.actor_id, actor.name, art.url, actor_link.role, actor_link.cast_order FROM actor_link LEFT JOIN actor ON actor.actor_id = actor_link.actor_id LEFT JOIN art ON (art.media_id = actor_link.actor_id AND art.media_type = 'actor') WHERE actor_link.media_id = ? AND actor_link.media_type = ? ''' else: query = ''' SELECT actor.actor_id, actor.name FROM {0}_link LEFT JOIN actor ON actor.actor_id = {0}_link.actor_id WHERE {0}_link.media_id = ? AND {0}_link.media_type = ? '''.format(kind) self.cursor.execute(query, (kodi_id, kodi_type)) old_people = self.cursor.fetchall() # Determine which people we need to save or delete outdated_people = [] for person in old_people: try: people_list.remove(person[1:]) except ValueError: outdated_people.append(person) # Get rid of old entries query = ''' DELETE FROM %s_link WHERE actor_id = ? AND media_id = ? AND media_type = ? ''' % kind query_actor_check = 'SELECT actor_id FROM %s_link WHERE actor_id = ?' query_actor_delete = 'DELETE FROM actor WHERE actor_id = ?' for person in outdated_people: # Delete the outdated entry self.cursor.execute(query, (person[0], kodi_id, kodi_type)) # Do we now have orphaned entries? for person_kind in ('actor', 'writer', 'director'): self.cursor.execute(query_actor_check % person_kind, (person[0],)) if self.cursor.fetchone() is not None: break else: # person entry in actor table is now orphaned # Delete the person from actor table self.cursor.execute(query_actor_delete, (person[0],)) if kind == 'actor': # Delete any associated artwork self.delete_artwork(person[0], 'actor') # Save new people to Kodi DB by iterating over the remaining entries self._add_people_kind(kodi_id, kodi_type, kind, people_list) @db.catch_operationalerrors def _new_actor_id(self, name, art_url): # Not yet in actor DB, add person self.cursor.execute('INSERT INTO actor(name) VALUES (?)', (name, )) actor_id = self.cursor.lastrowid if art_url: self.add_art(art_url, actor_id, 'actor', 'thumb') return actor_id def _get_actor_id(self, name, art_url=None): """ Returns the tuple (actor_id [int], new_entry [bool]) for name [unicode] in table actor (without ensuring that the name matches)."new_entry" will be True if a new DB entry has just been created. If not, will create a new record with actor_id, name, art_url Uses Plex ids and thus assumes that Plex person id is unique! """ self.cursor.execute('SELECT actor_id FROM actor WHERE name=? LIMIT 1', (name,)) try: return (self.cursor.fetchone()[0], False) except TypeError: return (self._new_actor_id(name, art_url), True) def _check_actor_art(self, actor_id, url): """ Sets the actor's art url [unicode] for actor_id [int] """ self.cursor.execute(''' SELECT EXISTS(SELECT 1 FROM art WHERE media_id = ? AND media_type = 'actor' LIMIT 1)''', (actor_id, )) if not self.cursor.fetchone()[0]: # We got a new artwork url for this actor! self.add_art(url, actor_id, 'actor', 'thumb') def get_art(self, kodi_id, kodi_type): """ Returns a dict of all available artwork with unicode urls/paths: { 'thumb' 'poster' 'banner' 'clearart' 'clearlogo' 'discart' 'fanart' and also potentially more fanart 'fanart1', 'fanart2', } Missing fanart will not appear in the dict. 'landscape' and 'icon' might be implemented in the future. """ self.cursor.execute('SELECT type, url FROM art WHERE media_id=? AND media_type=?', (kodi_id, kodi_type)) return dict(self.cursor.fetchall()) def get_trailer(self, kodi_id, kodi_type): """ Returns the trailer's URL for kodi_type from the Kodi database or None """ if kodi_type == v.KODI_TYPE_MOVIE: self.cursor.execute('SELECT c19 FROM movie WHERE idMovie=?', (kodi_id, )) else: raise NotImplementedError('trailers for %s not implemented' % kodi_type) try: return self.cursor.fetchone()[0] except TypeError: pass @db.catch_operationalerrors def set_trailer(self, kodi_id, kodi_type, url): """ Writes the trailer's url to the Kodi DB """ if kodi_type == v.KODI_TYPE_MOVIE: self.cursor.execute('UPDATE movie SET c19=? WHERE idMovie=?', (url, kodi_id)) else: raise NotImplementedError('trailers for not implemented' % kodi_type) @db.catch_operationalerrors def modify_streams(self, fileid, streamdetails=None, runtime=None): """ Leave streamdetails and runtime empty to delete all stream entries for fileid """ # First remove any existing entries self.cursor.execute('DELETE FROM streamdetails WHERE idFile = ?', (fileid,)) if not streamdetails: return for videotrack in streamdetails['video']: self.cursor.execute(''' INSERT OR REPLACE INTO streamdetails( idFile, iStreamType, strVideoCodec, fVideoAspect, iVideoWidth, iVideoHeight, iVideoDuration ,strStereoMode) VALUES (?, ?, ?, ?, ?, ?, ?, ?) ''', (fileid, 0, videotrack['codec'], videotrack['aspect'], videotrack['width'], videotrack['height'], runtime, videotrack['video3DFormat'])) for audiotrack in streamdetails['audio']: self.cursor.execute(''' INSERT OR REPLACE INTO streamdetails( idFile, iStreamType, strAudioCodec, iAudioChannels, strAudioLanguage) VALUES (?, ?, ?, ?, ?) ''', (fileid, 1, audiotrack['codec'], audiotrack['channels'], audiotrack['language'])) for subtitletrack in streamdetails['subtitle']: self.cursor.execute(''' INSERT OR REPLACE INTO streamdetails(idFile, iStreamType, strSubtitleLanguage) VALUES (?, ?, ?) ''', (fileid, 2, subtitletrack)) def video_id_from_filename(self, filename, path): """ Returns the tuple (itemId, type) where itemId: Kodi DB unique Id for either movie or episode type: either 'movie' or 'episode' Returns None if not found OR if too many entries were found """ self.cursor.execute('SELECT idFile, idPath FROM files WHERE strFilename = ?', (filename,)) files = self.cursor.fetchall() if len(files) == 0: LOG.debug('Did not find any file, abort') return # result will contain a list of all idFile with matching filename and # matching path result = [] for file in files: # Use idPath to get path as a string self.cursor.execute('SELECT strPath FROM path WHERE idPath = ?', (file[1], )) try: path_str = self.cursor.fetchone()[0] except TypeError: # idPath not found; skip continue # For whatever reason, double might have become triple path_str = path_str.replace('///', '//').replace('\\\\\\', '\\\\') if path_str == path: result.append(file[0]) if len(result) == 0: LOG.info('Did not find matching paths, abort') return # Kodi seems to make ONE temporary entry; we only want the earlier, # permanent one if len(result) > 2: LOG.warn('We found too many items with matching filenames and ' ' paths, aborting') return file_id = result[0] # Try movies first self.cursor.execute('SELECT idMovie FROM movie WHERE idFile = ?', (file_id, )) try: movie_id = self.cursor.fetchone()[0] typus = v.KODI_TYPE_MOVIE except TypeError: # Try tv shows next self.cursor.execute('SELECT idEpisode FROM episode WHERE idFile = ?', (file_id, )) try: movie_id = self.cursor.fetchone()[0] typus = v.KODI_TYPE_EPISODE except TypeError: LOG.debug('Did not find a video DB match') return return movie_id, typus def get_resume(self, file_id): """ Returns the first resume point in seconds (int) if found, else None for the Kodi file_id provided """ self.cursor.execute('SELECT timeInSeconds FROM bookmark WHERE idFile = ? LIMIT 1', (file_id,)) try: return self.cursor.fetchone()[0] except TypeError: pass def get_playcount(self, file_id): """ Returns the playcount for the item file_id or None if not found """ self.cursor.execute('SELECT playCount FROM files WHERE idFile = ? LIMIT 1', (file_id, )) try: return self.cursor.fetchone()[0] except TypeError: pass @db.catch_operationalerrors def set_resume(self, file_id, resume_seconds, total_seconds, playcount, dateplayed): """ Adds a resume marker for a video library item. Will even set 2, considering add-on path widget hacks. """ # Delete existing resume point self.cursor.execute('DELETE FROM bookmark WHERE idFile = ?', (file_id,)) # Set watched count # Be careful to set playCount to None, NOT the int zero! self.cursor.execute('UPDATE files SET playCount = ?, lastPlayed = ? WHERE idFile = ?', (playcount or None, dateplayed, file_id)) # Set the resume bookmark if resume_seconds: self.cursor.execute(''' INSERT INTO bookmark( idFile, timeInSeconds, totalTimeInSeconds, thumbNailImage, player, playerState, type) VALUES (?, ?, ?, ?, ?, ?, ?) ''', (file_id, resume_seconds, total_seconds, '', 'VideoPlayer', '', 1)) @db.catch_operationalerrors def create_tag(self, name): """ Will create a new tag if needed and return the tag_id """ self.cursor.execute('SELECT tag_id FROM tag WHERE name = ? COLLATE NOCASE', (name,)) try: tag_id = self.cursor.fetchone()[0] except TypeError: self.cursor.execute('INSERT INTO tag(name) VALUES(?)', (name, )) tag_id = self.cursor.lastrowid return tag_id @db.catch_operationalerrors def update_tag(self, oldtag, newtag, kodiid, mediatype): """ Updates the tag_id by replaying oldtag with newtag """ try: self.cursor.execute(''' UPDATE tag_link SET tag_id = ? WHERE media_id = ? AND media_type = ? AND tag_id = ? ''', (newtag, kodiid, mediatype, oldtag,)) except Exception: # The new tag we are going to apply already exists for this item # delete current tag instead self.cursor.execute(''' DELETE FROM tag_link WHERE media_id = ? AND media_type = ? AND tag_id = ? ''', (kodiid, mediatype, oldtag,)) @db.catch_operationalerrors def create_collection(self, set_name): """ Returns the collection/set id for set_name [unicode] """ self.cursor.execute('SELECT idSet FROM sets WHERE strSet = ? COLLATE NOCASE', (set_name,)) try: setid = self.cursor.fetchone()[0] except TypeError: self.cursor.execute('INSERT INTO sets(strSet) VALUES(?)', (set_name, )) setid = self.cursor.lastrowid return setid @db.catch_operationalerrors def assign_collection(self, setid, movieid): """ Assign the movie to one set/collection """ self.cursor.execute('UPDATE movie SET idSet = ? WHERE idMovie = ?', (setid, movieid,)) @db.catch_operationalerrors def remove_from_set(self, movieid): """ Remove the movie with movieid [int] from an associated movie set, movie collection """ self.cursor.execute('UPDATE movie SET idSet = null WHERE idMovie = ?', (movieid,)) def get_set_id(self, kodi_id): """ Returns the set_id for the movie with kodi_id or None """ self.cursor.execute('SELECT idSet FROM movie WHERE idMovie = ?', (kodi_id, )) try: return self.cursor.fetchone()[0] except TypeError: pass @db.catch_operationalerrors def delete_possibly_empty_set(self, set_id): """ Checks whether there are other movies in the set set_id. If not, deletes the set """ self.cursor.execute('SELECT idSet FROM movie WHERE idSet = ?', (set_id, )) if self.cursor.fetchone() is None: self.cursor.execute('DELETE FROM sets WHERE idSet = ?', (set_id,)) @db.catch_operationalerrors def add_season(self, showid, seasonnumber): """ Adds a TV show season to the Kodi video DB or simply returns the ID, if there already is an entry in the DB """ self.cursor.execute('INSERT INTO seasons(idShow, season) VALUES (?, ?)', (showid, seasonnumber)) return self.cursor.lastrowid @db.catch_operationalerrors def add_uniqueid(self, *args): """ Feed with: media_id: int media_type: string value: string type: e.g. 'imdb' or 'tvdb' """ self.cursor.execute(''' INSERT INTO uniqueid( media_id, media_type, value, type) VALUES (?, ?, ?, ?) ''', (args)) return self.cursor.lastrowid @db.catch_operationalerrors def update_uniqueid(self, *args): """ Pass in value, media_id, media_type, type """ self.cursor.execute(''' INSERT OR REPLACE INTO uniqueid(media_id, media_type, type, value) VALUES(?, ?, ?, ?) ''', (args)) return self.cursor.lastrowid @db.catch_operationalerrors def remove_uniqueid(self, kodi_id, kodi_type): """ Deletes the entry from the uniqueid table for the item """ self.cursor.execute('DELETE FROM uniqueid WHERE media_id = ? AND media_type = ?', (kodi_id, kodi_type)) @db.catch_operationalerrors def update_ratings(self, *args): """ Feed with media_id, media_type, rating_type, rating, votes, rating_id """ self.cursor.execute(''' INSERT OR REPLACE INTO rating(media_id, media_type, rating_type, rating, votes) VALUES (?, ?, ?, ?, ?) ''', (args)) return self.cursor.lastrowid @db.catch_operationalerrors def add_ratings(self, *args): """ feed with: media_id, media_type, rating_type, rating, votes rating_type = 'default' """ self.cursor.execute(''' INSERT INTO rating( media_id, media_type, rating_type, rating, votes) VALUES (?, ?, ?, ?, ?) ''', (args)) return self.cursor.lastrowid @db.catch_operationalerrors def remove_ratings(self, kodi_id, kodi_type): """ Removes all ratings from the rating table for the item """ self.cursor.execute('DELETE FROM rating WHERE media_id = ? AND media_type = ?', (kodi_id, kodi_type)) def new_show_id(self): self.cursor.execute('SELECT COALESCE(MAX(idShow), 0) FROM tvshow') return self.cursor.fetchone()[0] + 1 def new_episode_id(self): self.cursor.execute('SELECT COALESCE(MAX(idEpisode), 0) FROM episode') return self.cursor.fetchone()[0] + 1 @db.catch_operationalerrors def add_episode(self, *args): self.cursor.execute( ''' INSERT INTO episode( idEpisode, idFile, c00, c01, c03, c04, c05, c09, c10, c12, c13, c14, idShow, c15, c16, c18, c19, c20, idSeason, userrating) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''', (args)) @db.catch_operationalerrors def update_episode(self, *args): self.cursor.execute( ''' UPDATE episode SET c00 = ?, c01 = ?, c03 = ?, c04 = ?, c05 = ?, c09 = ?, c10 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ?, c16 = ?, c18 = ?, c19 = ?, c20 = ?, idFile=?, idSeason = ?, userrating = ? WHERE idEpisode = ? ''', (args)) @db.catch_operationalerrors def add_show(self, *args): self.cursor.execute( ''' INSERT INTO tvshow( idShow, c00, c01, c04, c05, c08, c09, c12, c13, c14, c15) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''', (args)) @db.catch_operationalerrors def update_show(self, *args): self.cursor.execute( ''' UPDATE tvshow SET c00 = ?, c01 = ?, c04 = ?, c05 = ?, c08 = ?, c09 = ?, c12 = ?, c13 = ?, c14 = ?, c15 = ? WHERE idShow = ? ''', (args)) @db.catch_operationalerrors def add_showlinkpath(self, kodi_id, kodi_pathid): self.cursor.execute('INSERT INTO tvshowlinkpath(idShow, idPath) VALUES (?, ?)', (kodi_id, kodi_pathid)) @db.catch_operationalerrors def remove_show(self, kodi_id): self.cursor.execute('DELETE FROM tvshow WHERE idShow = ?', (kodi_id,)) @db.catch_operationalerrors def remove_season(self, kodi_id): self.cursor.execute('DELETE FROM seasons WHERE idSeason = ?', (kodi_id,)) @db.catch_operationalerrors def remove_episode(self, kodi_id): self.cursor.execute('DELETE FROM episode WHERE idEpisode = ?', (kodi_id,)) def new_movie_id(self): self.cursor.execute('SELECT COALESCE(MAX(idMovie), 0) FROM movie') return self.cursor.fetchone()[0] + 1 @db.catch_operationalerrors def add_movie(self, *args): self.cursor.execute( ''' INSERT OR REPLACE INTO movie( idMovie, idFile, c00, c01, c02, c03, c04, c05, c06, c07, c09, c10, c11, c12, c14, c15, c16, c18, c19, c21, c22, c23, premiered, userrating) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) ''', (args)) @db.catch_operationalerrors def remove_movie(self, kodi_id): self.cursor.execute('DELETE FROM movie WHERE idMovie = ?', (kodi_id,)) @db.catch_operationalerrors def update_userrating(self, kodi_id, kodi_type, userrating): """ Updates userrating """ if kodi_type == v.KODI_TYPE_MOVIE: table = kodi_type identifier = 'idMovie' elif kodi_type == v.KODI_TYPE_EPISODE: table = kodi_type identifier = 'idEpisode' elif kodi_type == v.KODI_TYPE_SEASON: table = 'seasons' identifier = 'idSeason' elif kodi_type == v.KODI_TYPE_SHOW: table = kodi_type identifier = 'idShow' self.cursor.execute('''UPDATE %s SET userrating = ? WHERE ? = ?''' % table, (userrating, identifier, kodi_id))