#!/usr/bin/env python # -*- coding: utf-8 -*- from __future__ import absolute_import, division, unicode_literals from .. import utils, variables as v SUPPORTED_KODI_TYPES = ( v.KODI_TYPE_MOVIE, v.KODI_TYPE_SHOW, v.KODI_TYPE_SEASON, v.KODI_TYPE_EPISODE, v.KODI_TYPE_ARTIST, v.KODI_TYPE_ALBUM, v.KODI_TYPE_SONG ) class PlexDBBase(object): """ Plex database methods used for all types of items """ def __init__(self, cursor=None): # Allows us to use this class with a cursor instead of context mgr self.cursor = cursor def __enter__(self): self.plexconn = utils.kodi_sql('plex') self.cursor = self.plexconn.cursor() return self def __exit__(self, e_typ, e_val, trcbak): self.plexconn.commit() self.plexconn.close() def is_recorded(self, plex_id, plex_type): """ FAST method to check whether a plex_id has already been recorded """ query = 'SELECT plex_id FROM %s WHERE plex_id = ?' % plex_type self.cursor.execute(query, (plex_id, )) return self.cursor.fetchone() is not None def item_by_id(self, plex_id, plex_type=None): """ Returns the item for plex_id or None. Supply with the correct plex_type to speed up lookup """ answ = None if plex_type == v.PLEX_TYPE_MOVIE: answ = self.movie(plex_id) elif plex_type == v.PLEX_TYPE_EPISODE: answ = self.episode(plex_id) elif plex_type == v.PLEX_TYPE_SHOW: answ = self.show(plex_id) elif plex_type == v.PLEX_TYPE_SEASON: answ = self.season(plex_id) elif plex_type == v.PLEX_TYPE_SONG: answ = self.song(plex_id) elif plex_type == v.PLEX_TYPE_ALBUM: answ = self.album(plex_id) elif plex_type == v.PLEX_TYPE_ARTIST: answ = self.artist(plex_id) else: # SLOW - lookup plex_id in all our tables for kind in (v.PLEX_TYPE_MOVIE, v.PLEX_TYPE_EPISODE, v.PLEX_TYPE_SHOW, v.PLEX_TYPE_SEASON, v.PLEX_TYPE_SONG, v.PLEX_TYPE_ALBUM, v.PLEX_TYPE_ARTIST): method = getattr(self, kind) answ = method(plex_id) if answ: break return answ def item_by_kodi_id(self, kodi_id, kodi_type): """ """ if kodi_type not in SUPPORTED_KODI_TYPES: return query = ('SELECT * from %s WHERE kodi_id = ? LIMIT 1' % v.PLEX_TYPE_FROM_KODI_TYPE[kodi_type]) self.cursor.execute(query, (kodi_id, )) method = getattr(self, 'entry_to_%s' % v.PLEX_TYPE_FROM_KODI_TYPE[kodi_type]) return method(self.cursor.fetchone()) def plex_id_by_last_sync(self, plex_type, last_sync): """ Returns an iterator for all items where the last_sync is NOT identical """ query = 'SELECT plex_id FROM %s WHERE last_sync <> ?' % plex_type self.cursor.execute(query, (last_sync, )) return (x[0] for x in self.cursor) def checksum(self, plex_id, plex_type): """ Returns the checksum for plex_id """ query = 'SELECT checksum FROM %s WHERE plex_id = ?' % plex_type self.cursor.execute(query, (plex_id, )) try: return self.cursor.fetchone()[0] except TypeError: pass def update_last_sync(self, plex_id, plex_type, last_sync): """ Sets a new timestamp for plex_id """ query = 'UPDATE %s SET last_sync = ? WHERE plex_id = ?' % plex_type self.cursor.execute(query, (last_sync, plex_id)) def remove(self, plex_id, plex_type): """ Removes the item from our Plex db """ query = 'DELETE FROM ? WHERE plex_id = ?' % plex_type self.cursor.execute(query, (plex_id, )) def every_plex_id(self, plex_type): """ Returns an iterator for plex_type for every single plex_id """ query = 'SELECT plex_id from %s' % plex_type self.cursor.execute(query) return (x[0] for x in self.cursor) def missing_fanart(self, plex_type): """ Returns an iterator for plex_type for all plex_id, where fanart_synced has not yet been set to 1 """ query = 'SELECT plex_id from %s WHERE fanart_synced = 0' % plex_type self.cursor.execute(query) return (x[0] for x in self.cursor) def set_fanart_synced(self, plex_id, plex_type): """ Toggles fanart_synced to 1 for plex_id """ query = 'UPDATE %s SET fanart_synced = 1 WHERE plex_id = ?' % plex_type self.cursor.execute(query, (plex_id, )) def initialize(): """ Run once upon PKC startup to verify that plex db exists. """ with PlexDBBase() as plexdb: plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS version( idVersion TEXT) ''') plexdb.cursor.execute(''' INSERT OR REPLACE INTO version(idVersion) VALUES (?) ''', (v.ADDON_VERSION, )) plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS sections( section_id INTEGER PRIMARY KEY, section_name TEXT, plex_type TEXT, kodi_tagid INTEGER, sync_to_kodi INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS movie( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, kodi_id INTEGER, kodi_fileid INTEGER, kodi_pathid INTEGER, fanart_synced INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS show( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, kodi_id INTEGER, kodi_pathid INTEGER, fanart_synced INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS season( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, show_id INTEGER, parent_id INTEGER, kodi_id INTEGER, fanart_synced INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS episode( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, show_id INTEGER, grandparent_id INTEGER, season_id INTEGER, parent_id INTEGER, kodi_id INTEGER, kodi_fileid INTEGER, kodi_pathid INTEGER, fanart_synced INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS artist( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, kodi_id INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS album( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, artist_id INTEGER, parent_id INTEGER, kodi_id INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS track( plex_id INTEGER PRIMARY KEY, checksum INTEGER UNIQUE, section_id INTEGER, artist_id INTEGER, grandparent_id INTEGER, album_id INTEGER, parent_id INTEGER, kodi_id INTEGER, kodi_pathid INTEGER, last_sync INTEGER) ''') plexdb.cursor.execute(''' CREATE TABLE IF NOT EXISTS playlists( plex_id INTEGER PRIMARY KEY, plex_name TEXT, plex_updatedat INTEGER, kodi_path TEXT, kodi_type TEXT, kodi_hash TEXT) ''') def wipe(): """ Completely resets the Plex database """ query = "SELECT name FROM sqlite_master WHERE type = 'table'" with PlexDBBase() as plexdb: plexdb.cursor.execute(query) tables = plexdb.cursor.fetchall() tables = [i[0] for i in tables] for table in tables: delete_query = 'DROP table IF EXISTS %s' % table plexdb.cursor.execute(delete_query)