mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-28 05:33:38 +00:00
8ddffb7882
Remove py2 part from _23.py Remove more mapped stuff. Replace filter_iter with native filter. Replace map_iter with native map. Remove unidecode from _23 (empty wrapper on py3). Remove map_list and replace with native list(map( for performance reasons. Replace filter_list with list(filter. Replace list_keys with list(. Replace list_values with list(...values()). Replace list_items with list(....items()). Replace ordered_dict with dict. Fix tvinfo base type docs. Remove py2 parts from sg_futures. Remove scandir lib ... it's a sub module of os in py3. Remove PY2 stuff. Ignore unknown ids for characters/persons. Fix tvdb image parsing. Ignore unknown id sources on person page.
166 lines
5.6 KiB
Python
166 lines
5.6 KiB
Python
#
|
|
# 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 = []
|