SickGear/sickgear/notifiers/__init__.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/>.

import os
import re

# import pushalot
# from lib import api_trakt
from . import emby, kodi, plex, xbmc, \
    boxcar2, nmj, nmjv2, pushbullet, pushover, pytivo, synoindex, synologynotifier, \
    discord, emailnotify, gitter, libnotify, growl, prowl, slack, telegram, trakt

import sickgear


class NotifierFactory(object):

    def __init__(self):
        self.notifiers = dict(
            # home theater / nas
            EMBY=emby.EmbyNotifier,
            KODI=kodi.KodiNotifier,
            PLEX=plex.PLEXNotifier,
            # ### XBMC=xbmc.XBMCNotifier,
            NMJ=nmj.NMJNotifier,
            NMJV2=nmjv2.NMJv2Notifier,
            SYNOINDEX=synoindex.SynoIndexNotifier,
            SYNOLOGY=synologynotifier.SynologyNotifier,
            PYTIVO=pytivo.PyTivoNotifier,

            # devices,
            BOXCAR2=boxcar2.Boxcar2Notifier,
            # PUSHALOT=pushalot.PushalotNotifier,
            PUSHBULLET=pushbullet.PushbulletNotifier,
            PUSHOVER=pushover.PushoverNotifier,
            GROWL=growl.GrowlNotifier,
            PROWL=prowl.ProwlNotifier,
            LIBNOTIFY=libnotify.LibnotifyNotifier,

            # social
            TRAKT=trakt.TraktNotifier,
            SLACK=slack.SlackNotifier,
            DISCORD=discord.DiscordNotifier,
            GITTER=gitter.GitterNotifier,
            TELEGRAM=telegram.TelegramNotifier,
            EMAIL=emailnotify.EmailNotifier,
        )

    @property
    def enabled(self):
        """
        Generator to yield iterable IDs for enabled notifiers
        :return: ID String
        :rtype: String
        """
        for n in filter(lambda v: v.is_enabled(), list(self.notifiers.values())):
            yield n.id()

    @property
    def enabled_onsnatch(self):
        for n in filter(lambda v: v.is_enabled() and v.is_enabled_onsnatch(), list(self.notifiers.values())):
            yield n.id()

    @property
    def enabled_ondownload(self):
        for n in filter(lambda v: v.is_enabled() and v.is_enabled_ondownload(), list(self.notifiers.values())):
            yield n.id()

    @property
    def enabled_onsubtitledownload(self):
        for n in filter(lambda v: v.is_enabled() and v.is_enabled_onsubtitledownload(), list(self.notifiers.values())):
            yield n.id()

    @property
    def enabled_library(self):
        for n in filter(lambda v: v.is_enabled() and v.is_enabled_library(), list(self.notifiers.values())):
            yield n.id()

    def get(self, nid):
        """
        Get a notifier instance
        :param nid: Notified ID
        :type nid: String
        :return: Notifier instance
        :rtype: Notifier
        """
        return self.notifiers[nid]()

    def get_enabled(self, kind=None):
        """
        Generator to yield iterable notifier instance(s) that are either enabled or enabled for requested actions
        :param kind: Action
        :type kind: String
        :return: Notifier instance
        :rtype: Notifier
        """
        for n in getattr(self, 'enabled' + ('' if None is kind else ('_' + kind))):
            yield self.get(n)


def notify_snatch(ep_obj):
    for n in NotifierFactory().get_enabled('onsnatch'):
        n.notify_snatch(ep_obj)


def notify_download(ep_obj):
    for n in NotifierFactory().get_enabled('ondownload'):
        n.notify_download(ep_obj)


def notify_subtitle_download(ep_obj, lang):
    for n in NotifierFactory().get_enabled('onsubtitledownload'):
        n.notify_subtitle_download(ep_obj, lang)


def notify_git_update(new_version=''):
    if sickgear.NOTIFY_ON_UPDATE:
        for n in NotifierFactory().get_enabled():
            n.notify_git_update(new_version)


def notify_update_library(ep_obj, flush_q=False):

    if not flush_q or sickgear.QUEUE_UPDATE_LIBRARY:

        for n in NotifierFactory().get_enabled('library'):

            if isinstance(n, (plex.PLEXNotifier, kodi.KodiNotifier)):
                if not flush_q:
                    sickgear.QUEUE_UPDATE_LIBRARY += [(ep_obj.show_obj.name, ep_obj.location)]
                else:
                    shows = set()
                    locations = set()
                    for show_name, location in sickgear.QUEUE_UPDATE_LIBRARY:
                        if isinstance(n, kodi.KodiNotifier):
                            if show_name in shows:
                                continue
                            shows.add(show_name)
                        else:
                            parent_dir = re.sub(r'[/\\]+%s.*' % show_name, '', os.path.dirname(location))
                            parent_dir = re.sub(r'^(.{,2})[/\\]', '', parent_dir)
                            if parent_dir in locations:
                                continue
                            locations.add(parent_dir)

                        n.update_library(show_name=show_name, location=location)

            elif not flush_q:

                n.update_library(show_obj=ep_obj.show_obj, show_name=ep_obj.show_obj.name, ep_obj=ep_obj)

        if flush_q:
            sickgear.QUEUE_UPDATE_LIBRARY = []