# 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/>. from __future__ import absolute_import import os import re from ..common import USER_AGENT from .generic import Notifier # noinspection PyPep8Naming import encodingKludge as ek from exceptions_helper import ex import sickgear from sickgear.image_cache import ImageCache from sg_helpers import get_url class TelegramNotifier(Notifier): def __init__(self): super(TelegramNotifier, self).__init__() def _notify(self, title, body, send_icon='', quiet=False, access_token='', chatid='', ep_obj=None, **kwargs): result = None use_icon = bool(self._choose(send_icon, sickgear.TELEGRAM_SEND_IMAGE)) quiet = bool(self._choose(quiet, sickgear.TELEGRAM_QUIET)) access_token = self._choose(access_token, sickgear.TELEGRAM_ACCESS_TOKEN) cid = self._choose(chatid, sickgear.TELEGRAM_CHATID) try: msg = self._body_only(('' if not title else u'<b>%s</b>' % title), body) msg = msg.replace(u'<b>%s</b>: ' % title, u'<b>%s:</b>\r\n' % ('SickGear ' + title, title)[use_icon]) # HTML spaces ( ) and tabs ( ) aren't supported # See https://core.telegram.org/bots/api#html-style msg = re.sub('(?i) ?', ' ', msg) # Tabs become 3 spaces msg = re.sub('(?i) ?', ' ', msg) if use_icon: image_path = ek.ek(os.path.join, sickgear.PROG_DIR, 'gui', 'slick', 'images', 'banner_thumb.jpg') if not self._testing: show_obj = ep_obj.show_obj banner_path = ImageCache().banner_thumb_path(show_obj.tvid, show_obj.prodid) if ek.ek(os.path.isfile, banner_path): image_path = banner_path with open(image_path, 'rb') as f: response = self.post('sendPhoto', access_token, cid, quiet, dict(files={'photo': ('image.png', f)}, post_data=dict(caption=msg))) else: response = self.post('sendMessage', access_token, cid, quiet, dict(post_data=dict(text=msg))) result = response and response.get('ok') or False 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)) @staticmethod def post(action, access_token, cid, quiet, params): params.update(dict(headers={'User-Agent': USER_AGENT}, verify=True, json=True)) params['post_data'].update(dict(chat_id=cid, parse_mode='HTML', disable_notification=quiet)) return get_url('https://api.telegram.org/bot%s/%s' % (access_token, action), **params) notifier = TelegramNotifier