mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-27 13:13:37 +00:00
1eec1e7492
Change exclude Trakt notifier from redundant library updates when renaming files. Change filename checker regex to be less strict. Change remove naming pattern code duplication.
167 lines
5.7 KiB
Python
167 lines
5.7 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, include_online=True):
|
|
|
|
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,
|
|
include_online=include_online)
|
|
|
|
if flush_q:
|
|
sickgear.QUEUE_UPDATE_LIBRARY = []
|