mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 05:33:37 +00:00
3fa750651c
Notifiers are now loaded into memory on demand. Add bubble links to Notifications config tabs. Add Discordapp notifier to Notifications config/Social. Add Gitter notifier to Notifications config/Social. Change order of notifiers in Notifications config tabs. Remove Pushalot notifier. Remove XBMC notifier. Refactor update_library, notify, test notify and test results functions. Change most IDs and vars consistent for HTML, CSS, JS, and Python - related to notifications, camelCase for JS, underscore separated lower_case for python, hyphen separated-lowercase for CSS. A couple of exceptions have been left untouched in this clean up. Change commented out some unused vars in preparation for later removal.
141 lines
4.6 KiB
Python
141 lines
4.6 KiB
Python
# coding=utf-8
|
|
#
|
|
# Author: Nic Wolfe <nic@wolfeden.ca>
|
|
# URL: http://code.google.com/p/sickbeard/
|
|
#
|
|
# 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 sickbeard
|
|
from sickbeard import logger
|
|
|
|
notify_strings = dict(
|
|
snatch='Started download',
|
|
download='Download finished',
|
|
subtitle_download='Subtitle download finished',
|
|
git_updated='SickGear updated',
|
|
git_updated_text='SickGear updated to commit#: ',
|
|
test_title='SickGear notification test',
|
|
test_body=u'Success testing %s settings from SickGear ʕ•ᴥ•ʔ',
|
|
)
|
|
|
|
|
|
class BaseNotifier(object):
|
|
|
|
def __init__(self):
|
|
self.sg_logo_file = 'apple-touch-icon-precomposed.png'
|
|
self._testing = False
|
|
|
|
@property
|
|
def _sg_logo_url(self):
|
|
return 'https://raw.githubusercontent.com/SickGear/SickGear/master/gui/slick/images/ico/' + self.sg_logo_file
|
|
|
|
def _log(self, msg, level=logger.MESSAGE):
|
|
logger.log(u'%s: %s' % (self.name, msg), level)
|
|
|
|
def _log_debug(self, msg):
|
|
self._log(msg, logger.DEBUG)
|
|
|
|
def _log_error(self, msg):
|
|
self._log(msg, logger.ERROR)
|
|
|
|
def _log_warning(self, msg):
|
|
self._log(msg, logger.WARNING)
|
|
|
|
@classmethod
|
|
def id(cls):
|
|
return cls.__name__.replace('Notifier', '').upper()
|
|
|
|
@property
|
|
def name(self):
|
|
return self.__class__.__name__.replace('Notifier', '')
|
|
|
|
@classmethod
|
|
def is_enabled_onsnatch(cls):
|
|
return cls.is_enabled('NOTIFY_ONSNATCH')
|
|
|
|
@classmethod
|
|
def is_enabled_ondownload(cls):
|
|
return cls.is_enabled('NOTIFY_ONDOWNLOAD')
|
|
|
|
@classmethod
|
|
def is_enabled_onsubtitledownload(cls):
|
|
return cls.is_enabled('NOTIFY_ONSUBTITLEDOWNLOAD')
|
|
|
|
@classmethod
|
|
def is_enabled_library(cls):
|
|
return cls.is_enabled('UPDATE_LIBRARY')
|
|
|
|
@classmethod
|
|
def is_enabled(cls, action=None):
|
|
return getattr(sickbeard, action and '%s_%s' % (cls.id(), action) or 'USE_%s' % cls.id(), False)
|
|
|
|
def notify_snatch(self, *args, **kwargs):
|
|
pass
|
|
|
|
def notify_download(self, *args, **kwargs):
|
|
pass
|
|
|
|
def notify_subtitle_download(self, *args, **kwargs):
|
|
pass
|
|
|
|
def notify_git_update(self, *args, **kwargs):
|
|
pass
|
|
|
|
def update_library(self, **kwargs):
|
|
"""
|
|
note: nmj_notifier fires its library update when the notify_download is issued (inside notifiers)
|
|
"""
|
|
pass
|
|
|
|
def _notify(self, *args, **kwargs):
|
|
pass
|
|
|
|
def _choose(self, current=True, saved=True):
|
|
if self._testing:
|
|
return current
|
|
return saved
|
|
|
|
@staticmethod
|
|
def _body_only(title, body):
|
|
# don't use title with updates or testing, as only one str is used
|
|
return body if 'SickGear' in title else '%s: %s' % (title, body.replace('#: ', '# '))
|
|
|
|
|
|
class Notifier(BaseNotifier):
|
|
|
|
def test_notify(self, *args, **kwargs):
|
|
self._testing = True
|
|
r = self._pre_notify('test_title', notify_strings['test_body'] % (self.name + ' notifier'), *args, **kwargs)
|
|
return (r, (('Success, notification sent.', 'Failed to send notification.')[not r]))[r in (True, False)]
|
|
|
|
def notify_snatch(self, ep_name, **kwargs):
|
|
self._pre_notify('snatch', ep_name, **kwargs)
|
|
|
|
def notify_download(self, ep_name, **kwargs):
|
|
self._pre_notify('download', ep_name, **kwargs)
|
|
|
|
def notify_subtitle_download(self, ep_name, lang, **kwargs):
|
|
self._pre_notify('subtitle_download', '%s : %s' % (ep_name, lang), **kwargs)
|
|
|
|
def notify_git_update(self, new_version='??', **kwargs):
|
|
self._pre_notify('git_updated', notify_strings['git_updated_text'] + new_version, **kwargs)
|
|
|
|
def _pre_notify(self, notify_string, message, *args, **kwargs):
|
|
self._log_debug(u'Sending notification "%s"' % (self._body_only(notify_strings[notify_string], message)))
|
|
try:
|
|
return self._notify(notify_strings[notify_string], message, *args, **kwargs)
|
|
except (StandardError, Exception):
|
|
return False
|