PlexKodiConnect/resources/lib/plex_db/tvshows.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, unicode_literals
from .. import variables as v


class TVShows(object):
    def add_show(self, plex_id, checksum, section_id, kodi_id, kodi_pathid,
                 last_sync):
        """
        Appends or replaces tv show entry into the plex table
        """
        self.cursor.execute(
            '''
            INSERT OR REPLACE INTO show(
                plex_id,
                checksum,
                section_id,
                kodi_id,
                kodi_pathid,
                fanart_synced,
                last_sync)
            VALUES (?, ?, ?, ?, ?, ?, ?)
            ''',
            (plex_id,
             checksum,
             section_id,
             kodi_id,
             kodi_pathid,
             0,
             last_sync))

    def add_season(self, plex_id, checksum, section_id, show_id, parent_id,
                   kodi_id, last_sync):
        """
        Appends or replaces an entry into the plex table
        """
        self.cursor.execute(
            '''
            INSERT OR REPLACE INTO season(
                plex_id,
                checksum,
                section_id,
                show_id,
                parent_id,
                kodi_id,
                fanart_synced,
                last_sync)
            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
            ''',
            (plex_id,
             checksum,
             section_id,
             show_id,
             parent_id,
             kodi_id,
             0,
             last_sync))

    def add_episode(self, plex_id, checksum, section_id, show_id,
                    grandparent_id, season_id, parent_id, kodi_id, kodi_fileid,
                    kodi_fileid_2, kodi_pathid, last_sync):
        """
        Appends or replaces an entry into the plex table
        """
        self.cursor.execute(
            '''
            INSERT OR REPLACE INTO episode(
                plex_id,
                checksum,
                section_id,
                show_id,
                grandparent_id,
                season_id,
                parent_id,
                kodi_id,
                kodi_fileid,
                kodi_fileid_2,
                kodi_pathid,
                fanart_synced,
                last_sync)
            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
            ''',
            (plex_id,
             checksum,
             section_id,
             show_id,
             grandparent_id,
             season_id,
             parent_id,
             kodi_id,
             kodi_fileid,
             kodi_fileid_2,
             kodi_pathid,
             0,
             last_sync))

    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
        """
        if plex_id is None:
            return
        self.cursor.execute('SELECT * FROM show WHERE plex_id = ? LIMIT 1',
                            (plex_id, ))
        return self.entry_to_show(self.cursor.fetchone())

    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
        """
        if plex_id is None:
            return
        self.cursor.execute('SELECT * FROM season WHERE plex_id = ? LIMIT 1',
                            (plex_id, ))
        return self.entry_to_season(self.cursor.fetchone())

    def episode(self, plex_id):
        if plex_id is None:
            return
        self.cursor.execute('SELECT * FROM episode WHERE plex_id = ? LIMIT 1',
                            (plex_id, ))
        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]
        }

    @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]
        }

    @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]
        }

    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
        """
        return (self.entry_to_episode(x) for x in
                self.cursor.execute('SELECT * FROM episode WHERE season_id = ?',
                                    (plex_id, )))

    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
        """
        return (self.entry_to_episode(x) for x in
                self.cursor.execute('SELECT * FROM episode WHERE show_id = ?',
                                    (plex_id, )))

    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
        """
        return (self.entry_to_season(x) for x in
                self.cursor.execute('SELECT * FROM season WHERE show_id = ?',
                                    (plex_id, )))