SickGear/sickgear/watchedstate_queue.py

#
# This file is part of SickGear.
#
# SickGear is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# SickGear is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with SickGear.  If not, see <http://www.gnu.org/licenses/>.

from . import logger, generic_queue
from .webserve import History

EMBYWATCHEDSTATE = 10
PLEXWATCHEDSTATE = 20


class WatchedStateQueue(generic_queue.GenericQueue):
    def __init__(self):
        super(WatchedStateQueue, self).__init__()
        # self.queue_name = 'WATCHEDSTATEQUEUE'
        self.queue_name = 'Q'

    def is_in_queue(self, itemtype):
        with self.lock:
            for cur_item in self.queue + [self.currentItem]:
                if isinstance(cur_item, itemtype):
                    return True
            return False

    # method for possible UI usage, can be removed if not used
    def queue_length(self):
        length = {'emby': 0, 'plex': 0}
        with self.lock:
            for cur_item in [self.currentItem] + self.queue:
                if isinstance(cur_item, EmbyWatchedStateQueueItem):
                    length['emby'] += 1
                elif isinstance(cur_item, PlexWatchedStateQueueItem):
                    length['plex'] += 1

        return length

    def add_item(self, item, **kwargs):
        if isinstance(item, EmbyWatchedStateQueueItem) and not self.is_in_queue(EmbyWatchedStateQueueItem):
            # emby watched state item
            generic_queue.GenericQueue.add_item(self, item)
        elif isinstance(item, PlexWatchedStateQueueItem) and not self.is_in_queue(PlexWatchedStateQueueItem):
            # plex watched state item
            generic_queue.GenericQueue.add_item(self, item)
        else:
            logger.debug("Not adding item, it's already in the queue")


class EmbyWatchedStateQueueItem(generic_queue.QueueItem):
    def __init__(self):
        super(EmbyWatchedStateQueueItem, self).__init__('Emby Watched', EMBYWATCHEDSTATE)

    def run(self):
        super(EmbyWatchedStateQueueItem, self).run()
        try:
            History.update_watched_state_emby()
        finally:
            self.finish()


class PlexWatchedStateQueueItem(generic_queue.QueueItem):
    def __init__(self):
        super(PlexWatchedStateQueueItem, self).__init__('Plex Watched', PLEXWATCHEDSTATE)

    def run(self):
        super(PlexWatchedStateQueueItem, self).run()
        try:
            History.update_watched_state_plex()
        finally:
            self.finish()