mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-23 19:23:37 +00:00
58 lines
2.5 KiB
Python
58 lines
2.5 KiB
Python
|
# 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 <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
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 = '<br><br>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
|