# coding=utf-8
#
# This file is part of SickGear.
#
# Thanks to: mallen86, generica
#
# 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 .
import re
from .generic import Notifier
import sickgear
class DiscordNotifier(Notifier):
def __init__(self):
super(DiscordNotifier, self).__init__()
def _notify(self, title, body, as_authed=None, username='', icon_url='', as_tts='', access_token='', **kwargs):
params = [] if not bool(self._choose(not as_authed, sickgear.DISCORD_AS_AUTHED)) else \
[('username', self._choose(username, sickgear.DISCORD_USERNAME) or 'SickGear'),
('avatar_url', self._choose(icon_url, sickgear.DISCORD_ICON_URL) or self._sg_logo_url)]
as_tts = self._choose(as_tts, bool(sickgear.DISCORD_AS_TTS))
url = self._choose(access_token, sickgear.DISCORD_ACCESS_TOKEN)
success_and_deprecated_msg = ''
if 'discordapp' in url:
# upgrade webhook url for get_url, and notify user on test to update setting
url = re.sub(r'^(https?://discord)app', r'\1', url)
success_and_deprecated_msg = '
Note2: Please change `discordapp.com` ' \
'in Discord channel webhook to `discord.com`'
resp = sickgear.helpers.get_url(
url=url,
post_json=dict([('content', self._body_only(title, body)), ('tts', as_tts)] + params))
result = '' == resp or self._choose('bad webhook?', None)
if True is not result:
self._log_error('%s failed to send message: %s' % (self.name, result))
return self._choose(('Success, notification sent. (Note: %s displays the icon once in a sequence)%s'
% (self.name, success_and_deprecated_msg),
'Failed to send notification, %s' % result)[True is not result], result)
notifier = DiscordNotifier