PlexKodiConnect/resources/lib/plex_db/tvshows.py

245 lines
7.7 KiB
Python
Raw Permalink 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 00:20:25 +11:00
from .. import variables as v
2018-10-22 01:56:13 +11:00
2018-10-23 22:54:09 +11:00
class TVShows(object):
def add_show(self, plex_id, checksum, section_id, kodi_id, kodi_pathid,
last_sync):
2018-10-23 22:54:09 +11:00
"""
Appends or replaces tv show entry into the plex table
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute(
'''
2018-10-23 22:54:09 +11:00
INSERT OR REPLACE INTO show(
2018-10-25 00:20:25 +11:00
plex_id,
checksum,
section_id,
kodi_id,
kodi_pathid,
fanart_synced,
2018-10-24 19:57:52 +11:00
last_sync)
VALUES (?, ?, ?, ?, ?, ?, ?)
2018-11-09 01:15:52 +11:00
''',
2018-10-25 00:20:25 +11:00
(plex_id,
checksum,
section_id,
kodi_id,
kodi_pathid,
0,
2018-10-23 22:54:09 +11:00
last_sync))
2018-10-22 01:56:13 +11:00
def add_season(self, plex_id, checksum, section_id, show_id, parent_id,
kodi_id, last_sync):
2018-10-23 22:54:09 +11:00
"""
Appends or replaces an entry into the plex table
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute(
'''
2018-10-23 22:54:09 +11:00
INSERT OR REPLACE INTO season(
2018-10-25 00:20:25 +11:00
plex_id,
checksum,
section_id,
show_id,
parent_id,
kodi_id,
fanart_synced,
2018-10-24 19:57:52 +11:00
last_sync)
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
2018-11-09 01:15:52 +11:00
''',
2018-10-25 00:20:25 +11:00
(plex_id,
checksum,
section_id,
show_id,
parent_id,
kodi_id,
0,
last_sync))
2018-10-22 01:56:13 +11:00
def add_episode(self, plex_id, checksum, section_id, show_id,
2018-10-25 00:20:25 +11:00
grandparent_id, season_id, parent_id, kodi_id, kodi_fileid,
kodi_fileid_2, kodi_pathid, last_sync):
2018-10-22 01:56:13 +11:00
"""
Appends or replaces an entry into the plex table
"""
2018-11-09 01:15:52 +11:00
self.cursor.execute(
'''
2018-10-23 22:54:09 +11:00
INSERT OR REPLACE INTO episode(
2018-10-25 00:20:25 +11:00
plex_id,
checksum,
section_id,
show_id,
grandparent_id,
season_id,
parent_id,
kodi_id,
kodi_fileid,
kodi_fileid_2,
2018-10-25 00:20:25 +11:00
kodi_pathid,
fanart_synced,
2018-10-24 19:57:52 +11:00
last_sync)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
2018-11-09 01:15:52 +11:00
''',
2018-10-25 00:20:25 +11:00
(plex_id,
checksum,
section_id,
show_id,
grandparent_id,
season_id,
parent_id,
kodi_id,
kodi_fileid,
kodi_fileid_2,
2018-10-25 00:20:25 +11:00
kodi_pathid,
0,
last_sync))
2018-10-22 01:56:13 +11:00
def show(self, plex_id):
"""
Returns the show info as a tuple for the TV show with plex_id:
plex_id INTEGER PRIMARY KEY ASC,
checksum INTEGER UNIQUE,
section_id INTEGER,
kodi_id INTEGER,
kodi_pathid INTEGER,
fanart_synced INTEGER,
last_sync INTEGER
"""
2018-10-25 00:20:25 +11:00
if plex_id is None:
return
2018-10-23 22:54:09 +11:00
self.cursor.execute('SELECT * FROM show WHERE plex_id = ? LIMIT 1',
2018-10-22 01:56:13 +11:00
(plex_id, ))
2018-10-25 00:20:25 +11:00
return self.entry_to_show(self.cursor.fetchone())
2018-10-22 01:56:13 +11:00
def season(self, plex_id):
"""
Returns the show info as a tuple for the TV show with plex_id:
plex_id INTEGER PRIMARY KEY,
checksum INTEGER UNIQUE,
section_id INTEGER,
show_id INTEGER, # plex_id of the parent show
parent_id INTEGER, # kodi_id of the parent show
kodi_id INTEGER,
fanart_synced INTEGER,
last_sync INTEGER
"""
2018-10-25 00:20:25 +11:00
if plex_id is None:
return
2018-10-23 22:54:09 +11:00
self.cursor.execute('SELECT * FROM season WHERE plex_id = ? LIMIT 1',
2018-10-22 01:56:13 +11:00
(plex_id, ))
2018-10-25 00:20:25 +11:00
return self.entry_to_season(self.cursor.fetchone())
2018-10-22 01:56:13 +11:00
def episode(self, plex_id):
2018-10-25 00:20:25 +11:00
if plex_id is None:
return
2018-10-23 22:54:09 +11:00
self.cursor.execute('SELECT * FROM episode WHERE plex_id = ? LIMIT 1',
2018-10-22 01:56:13 +11:00
(plex_id, ))
2018-10-25 00:20:25 +11:00
return self.entry_to_episode(self.cursor.fetchone())
@staticmethod
def entry_to_episode(entry):
if not entry:
return
return {
'plex_type': v.PLEX_TYPE_EPISODE,
'kodi_type': v.KODI_TYPE_EPISODE,
'plex_id': entry[0],
'checksum': entry[1],
'section_id': entry[2],
'show_id': entry[3],
'grandparent_id': entry[4],
'season_id': entry[5],
'parent_id': entry[6],
'kodi_id': entry[7],
'kodi_fileid': entry[8],
'kodi_fileid_2': entry[9],
'kodi_pathid': entry[10],
'fanart_synced': entry[11],
'last_sync': entry[12]
2018-10-25 00:20:25 +11:00
}
@staticmethod
def entry_to_show(entry):
if not entry:
return
return {
'plex_type': v.PLEX_TYPE_SHOW,
'kodi_type': v.KODI_TYPE_SHOW,
'plex_id': entry[0],
'checksum': entry[1],
'section_id': entry[2],
'kodi_id': entry[3],
'kodi_pathid': entry[4],
'fanart_synced': entry[5],
'last_sync': entry[6]
2018-10-25 00:20:25 +11:00
}
@staticmethod
def entry_to_season(entry):
if not entry:
return
return {
'plex_type': v.PLEX_TYPE_SEASON,
'kodi_type': v.KODI_TYPE_SEASON,
'plex_id': entry[0],
'checksum': entry[1],
'section_id': entry[2],
'show_id': entry[3],
'parent_id': entry[4],
'kodi_id': entry[5],
'fanart_synced': entry[6],
'last_sync': entry[7]
2018-10-25 00:20:25 +11:00
}
2018-10-24 19:57:52 +11:00
def season_has_episodes(self, plex_id):
"""
Returns True if there are episodes left for the season with plex_id
"""
self.cursor.execute('SELECT plex_id FROM episode WHERE season_id = ? LIMIT 1',
(plex_id, ))
return self.cursor.fetchone() is not None
def show_has_seasons(self, plex_id):
"""
Returns True if there are seasons left for the show with plex_id
"""
self.cursor.execute('SELECT plex_id FROM season WHERE show_id = ? LIMIT 1',
(plex_id, ))
return self.cursor.fetchone() is not None
def show_has_episodes(self, plex_id):
"""
Returns True if there are episodes left for the show with plex_id
"""
self.cursor.execute('SELECT plex_id FROM episode WHERE show_id = ? LIMIT 1',
(plex_id, ))
return self.cursor.fetchone() is not None
def episode_by_season(self, plex_id):
"""
Returns an iterator for all episodes that have a parent season_id with
a value of plex_id
"""
2018-11-09 01:15:52 +11:00
return (self.entry_to_episode(x) for x in
self.cursor.execute('SELECT * FROM episode WHERE season_id = ?',
(plex_id, )))
2018-10-24 19:57:52 +11:00
def episode_by_show(self, plex_id):
"""
Returns an iterator for all episodes that have a grandparent show_id
with a value of plex_id
"""
2018-11-09 01:15:52 +11:00
return (self.entry_to_episode(x) for x in
self.cursor.execute('SELECT * FROM episode WHERE show_id = ?',
(plex_id, )))
2018-10-24 19:57:52 +11:00
def season_by_show(self, plex_id):
"""
Returns an iterator for all seasons that have a parent show_id
with a value of plex_id
"""
2018-11-09 01:15:52 +11:00
return (self.entry_to_season(x) for x in
self.cursor.execute('SELECT * FROM season WHERE show_id = ?',
(plex_id, )))