PlexKodiConnect/resources/lib/plex_db/common.py

286 lines
10 KiB
Python
Raw Normal View History

2018-10-22 01:56:13 +11:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, unicode_literals
2018-10-25 02:53:20 +11:00
from .. import utils, variables as v
2018-10-22 01:56:13 +11:00
2018-10-25 02:17:02 +11:00
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,
2018-11-04 02:59:59 +11:00
v.KODI_TYPE_SONG
)
2018-10-25 02:17:02 +11:00
2018-10-23 22:54:09 +11:00
class PlexDBBase(object):
2018-10-22 01:56:13 +11:00
"""
2018-10-24 19:57:52 +11:00
Plex database methods used for all types of items
2018-10-22 01:56:13 +11:00
"""
2018-10-23 22:54:09 +11:00
def __init__(self, cursor=None):
# Allows us to use this class with a cursor instead of context mgr
2018-10-22 01:56:13 +11:00
self.cursor = cursor
2018-10-23 22:54:09 +11:00
def __enter__(self):
self.plexconn = utils.kodi_sql('plex')
self.cursor = self.plexconn.cursor()
return self
def __exit__(self, e_typ, e_val, trcbak):
if e_typ:
# re-raise any exception
return False
2018-10-23 22:54:09 +11:00
self.plexconn.commit()
self.plexconn.close()
2018-10-26 01:07:34 +11:00
def is_recorded(self, plex_id, plex_type):
"""
FAST method to check whether a plex_id has already been recorded
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute('SELECT plex_id FROM %s WHERE plex_id = ?' % plex_type,
(plex_id, ))
2018-10-26 01:07:34 +11:00
return self.cursor.fetchone() is not None
2018-10-24 19:57:52 +11:00
def item_by_id(self, plex_id, plex_type=None):
"""
2018-10-25 02:17:02 +11:00
Returns the item for plex_id or None.
Supply with the correct plex_type to speed up lookup
2018-10-24 19:57:52 +11:00
"""
answ = None
if plex_type == v.PLEX_TYPE_MOVIE:
2018-11-04 02:56:51 +11:00
answ = self.movie(plex_id)
2018-10-24 19:57:52 +11:00
elif plex_type == v.PLEX_TYPE_EPISODE:
2018-11-04 02:56:51 +11:00
answ = self.episode(plex_id)
2018-10-24 19:57:52 +11:00
elif plex_type == v.PLEX_TYPE_SHOW:
2018-11-04 02:56:51 +11:00
answ = self.show(plex_id)
2018-10-24 19:57:52 +11:00
elif plex_type == v.PLEX_TYPE_SEASON:
2018-11-04 02:56:51 +11:00
answ = self.season(plex_id)
2018-11-04 02:59:46 +11:00
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)
2018-10-24 19:57:52 +11:00
else:
# SLOW - lookup plex_id in all our tables
for kind in (v.PLEX_TYPE_MOVIE,
v.PLEX_TYPE_EPISODE,
2018-11-04 02:59:46 +11:00
v.PLEX_TYPE_SHOW,
v.PLEX_TYPE_SEASON,
2018-11-05 03:00:34 +11:00
'song', # darn
2018-11-04 02:59:46 +11:00
v.PLEX_TYPE_ALBUM,
v.PLEX_TYPE_ARTIST):
2018-10-24 19:57:52 +11:00
method = getattr(self, kind)
2018-11-04 02:56:51 +11:00
answ = method(plex_id)
if answ:
2018-10-24 19:57:52 +11:00
break
return answ
2018-10-25 02:17:02 +11:00
def item_by_kodi_id(self, kodi_id, kodi_type):
"""
"""
if kodi_type not in SUPPORTED_KODI_TYPES:
return
2018-11-09 01:15:52 +11:00
self.cursor.execute('SELECT * from %s WHERE kodi_id = ? LIMIT 1'
% v.PLEX_TYPE_FROM_KODI_TYPE[kodi_type],
(kodi_id, ))
2018-10-25 02:17:02 +11:00
method = getattr(self, 'entry_to_%s' % v.PLEX_TYPE_FROM_KODI_TYPE[kodi_type])
return method(self.cursor.fetchone())
2018-10-23 22:54:09 +11:00
def plex_id_by_last_sync(self, plex_type, last_sync):
"""
Returns an iterator for all items where the last_sync is NOT identical
2018-10-22 01:56:13 +11:00
"""
2018-11-09 01:15:52 +11:00
return (x[0] for x in
self.cursor.execute('SELECT plex_id FROM %s WHERE last_sync <> ?' % plex_type,
(last_sync, )))
2018-10-22 01:56:13 +11:00
def checksum(self, plex_id, plex_type):
2018-10-29 03:02:06 +11:00
"""
Returns the checksum for plex_id
2018-10-29 03:02:06 +11:00
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute('SELECT checksum FROM %s WHERE plex_id = ? LIMIT 1' % plex_type,
(plex_id, ))
2018-10-29 03:02:06 +11:00
try:
return self.cursor.fetchone()[0]
except TypeError:
pass
def update_last_sync(self, plex_id, plex_type, last_sync):
2018-10-22 01:56:13 +11:00
"""
2018-10-23 22:54:09 +11:00
Sets a new timestamp for plex_id
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute('UPDATE %s SET last_sync = ? WHERE plex_id = ?' % plex_type,
(last_sync, plex_id))
2018-10-23 22:54:09 +11:00
def remove(self, plex_id, plex_type):
"""
Removes the item from our Plex db
"""
2018-11-25 00:06:21 +11:00
self.cursor.execute('DELETE FROM %s WHERE plex_id = ?' % plex_type, (plex_id, ))
2018-10-23 22:54:09 +11:00
2018-11-03 20:36:37 +11:00
def every_plex_id(self, plex_type):
"""
Returns an iterator for plex_type for every single plex_id
"""
2018-11-09 01:15:52 +11:00
return (x[0] for x in
self.cursor.execute('SELECT plex_id from %s' % plex_type))
2018-11-03 20:36:37 +11:00
def missing_fanart(self, plex_type):
2018-10-24 19:57:52 +11:00
"""
Returns an iterator for plex_type for all plex_id, where fanart_synced
has not yet been set to 1
"""
2018-11-09 01:15:52 +11:00
return (x[0] for x in
self.cursor.execute('SELECT plex_id from %s WHERE fanart_synced = 0' % plex_type))
2018-10-24 19:57:52 +11:00
def set_fanart_synced(self, plex_id, plex_type):
"""
Toggles fanart_synced to 1 for plex_id
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute('UPDATE %s SET fanart_synced = 1 WHERE plex_id = ?' % plex_type,
(plex_id, ))
2018-10-24 19:57:52 +11:00
2018-11-26 06:15:38 +11:00
def plexid_by_sectionid(self, section_id, plex_type):
return (x[0] for x in
self.cursor.execute('SELECT plex_id FROM %s WHERE section_id = ?' % plex_type,
(section_id, )))
2018-10-23 22:54:09 +11:00
def initialize():
"""
2018-10-24 19:57:52 +11:00
Run once upon PKC startup to verify that plex db exists.
2018-10-23 22:54:09 +11:00
"""
2018-10-24 19:57:52 +11:00
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, ))
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS sections(
section_id INTEGER PRIMARY KEY,
section_name TEXT,
plex_type TEXT,
kodi_tagid INTEGER,
sync_to_kodi INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS movie(
2018-10-24 19:57:52 +11:00
plex_id INTEGER PRIMARY KEY,
2018-10-23 22:54:09 +11:00
checksum INTEGER UNIQUE,
section_id INTEGER,
kodi_id INTEGER,
kodi_fileid INTEGER,
kodi_pathid INTEGER,
fanart_synced INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS show(
2018-10-24 19:57:52 +11:00
plex_id INTEGER PRIMARY KEY,
2018-10-23 22:54:09 +11:00
checksum INTEGER UNIQUE,
section_id INTEGER,
kodi_id INTEGER,
kodi_pathid INTEGER,
fanart_synced INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS season(
plex_id INTEGER PRIMARY KEY,
checksum INTEGER UNIQUE,
section_id INTEGER,
2018-10-25 03:07:51 +11:00
show_id INTEGER,
parent_id INTEGER,
2018-10-23 22:54:09 +11:00
kodi_id INTEGER,
fanart_synced INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS episode(
plex_id INTEGER PRIMARY KEY,
checksum INTEGER UNIQUE,
section_id INTEGER,
2018-10-25 03:07:51 +11:00
show_id INTEGER,
grandparent_id INTEGER,
season_id INTEGER,
parent_id INTEGER,
2018-10-23 22:54:09 +11:00
kodi_id INTEGER,
kodi_fileid INTEGER,
kodi_pathid INTEGER,
fanart_synced INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS artist(
2018-10-24 19:57:52 +11:00
plex_id INTEGER PRIMARY KEY,
2018-10-23 22:54:09 +11:00
checksum INTEGER UNIQUE,
section_id INTEGER,
kodi_id INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS album(
plex_id INTEGER PRIMARY KEY,
checksum INTEGER UNIQUE,
section_id INTEGER,
2018-10-25 03:07:51 +11:00
artist_id INTEGER,
parent_id INTEGER,
2018-10-23 22:54:09 +11:00
kodi_id INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS track(
plex_id INTEGER PRIMARY KEY,
checksum INTEGER UNIQUE,
section_id INTEGER,
2018-10-25 03:07:51 +11:00
artist_id INTEGER,
grandparent_id INTEGER,
album_id INTEGER,
parent_id INTEGER,
2018-10-23 22:54:09 +11:00
kodi_id INTEGER,
kodi_pathid INTEGER,
last_sync INTEGER)
''')
2018-10-24 19:57:52 +11:00
plexdb.cursor.execute('''
2018-10-23 22:54:09 +11:00
CREATE TABLE IF NOT EXISTS playlists(
2018-10-24 19:57:52 +11:00
plex_id INTEGER PRIMARY KEY,
2018-10-23 22:54:09 +11:00
plex_name TEXT,
plex_updatedat INTEGER,
kodi_path TEXT,
kodi_type TEXT,
kodi_hash TEXT)
''')
2018-11-12 05:22:32 +11:00
# DB indicees for faster lookups
2018-11-12 06:11:19 +11:00
commands = (
'CREATE UNIQUE INDEX IF NOT EXISTS ix_movie_1 ON movie (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_show_1 ON show (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_season_1 ON season (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_episode_1 ON episode (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_artist_1 ON artist (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_album_1 ON album (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_track_1 ON track (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_playlists_1 ON playlists (plex_id)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_playlists_2 ON playlists (kodi_path)',
'CREATE UNIQUE INDEX IF NOT EXISTS ix_playlists_3 ON playlists (kodi_hash)',
)
for cmd in commands:
plexdb.cursor.execute(cmd)
2018-10-23 22:54:09 +11:00
def wipe():
"""
Completely resets the Plex database
"""
2018-10-24 19:57:52 +11:00
with PlexDBBase() as plexdb:
2018-11-09 01:15:52 +11:00
plexdb.cursor.execute("SELECT name FROM sqlite_master WHERE type = 'table'")
tables = [i[0] for i in plexdb.cursor.fetchall()]
2018-10-23 22:54:09 +11:00
for table in tables:
2018-11-09 01:15:52 +11:00
plexdb.cursor.execute('DROP table IF EXISTS %s' % table)