JackDandy
5 years ago
10 changed files with 246 additions and 12 deletions
|
After Width: | Height: | Size: 588 B |
@ -0,0 +1,61 @@ |
|||
# coding=utf-8 |
|||
# |
|||
# 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 |
|||
from .generic import Notifier |
|||
import sickbeard |
|||
|
|||
# noinspection PyPep8Naming |
|||
import encodingKludge as ek |
|||
from exceptions_helper import ex |
|||
from lib.apprise.plugins.NotifyTelegram import NotifyTelegram |
|||
|
|||
|
|||
class TelegramNotifier(Notifier): |
|||
|
|||
def __init__(self): |
|||
super(TelegramNotifier, self).__init__() |
|||
|
|||
def _notify(self, title, body, send_icon='', access_token='', chatid='', **kwargs): |
|||
result = None |
|||
cid = '' |
|||
use_icon = bool(self._choose(send_icon, sickbeard.TELEGRAM_SEND_ICON)) |
|||
try: |
|||
tg = NotifyTelegram( |
|||
bot_token=self._choose(access_token, sickbeard.TELEGRAM_ACCESS_TOKEN), |
|||
targets=self._choose(chatid, sickbeard.TELEGRAM_CHATID), |
|||
include_image=use_icon |
|||
) |
|||
cid = chatid or isinstance(tg.targets, list) and 1 == len(tg.targets) and tg.targets[0] or '' |
|||
if use_icon: |
|||
tg.icon_path = ek.ek(os.path.join, sickbeard.PROG_DIR, 'gui', 'slick', |
|||
'images', 'ico', 'apple-touch-icon-76x76.png') |
|||
tg.image_size = '72x72' |
|||
result = tg.send(body, title=title) |
|||
except (BaseException, Exception) as e: |
|||
if 'No chat_id' in ex(e): |
|||
result = 'a chat id is not set, and a msg has not been sent to the bot to auto detect one.' |
|||
|
|||
if True is not result: |
|||
self._log_error('Failed to send message, %s' % result) |
|||
|
|||
return dict(chatid=cid, |
|||
result=self._choose(('Successful test notice sent.', |
|||
'Error sending notification, %s' % result)[True is not result], result)) |
|||
|
|||
|
|||
notifier = TelegramNotifier |
|||
Loading…
Reference in new issue