# Authors: # Derek Battams # Pedro Jose Pereira Vieito (@pvieito) # # URL: https://github.com/sickgear/sickgear # # 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 . from email.mime.multipart import MIMEMultipart from email.mime.text import MIMEText from email.utils import formatdate import re import smtplib from .generic import Notifier, notify_strings from .. import db import sickgear from exceptions_helper import ex from six import text_type class EmailNotifier(Notifier): def __init__(self): super(EmailNotifier, self).__init__() self.last_err = None def _sendmail(self, host, port, smtp_from, use_tls, user, pwd, to, msg, smtp_debug=False): use_tls = 1 == sickgear.helpers.try_int(use_tls) login = any(user) and any(pwd) self._log_debug(u'Sendmail HOST: %s; PORT: %s; LOGIN: %s, TLS: %s, USER: %s, FROM: %s, TO: %s' % ( host, port, login, use_tls, user, smtp_from, to)) try: srv = smtplib.SMTP(host, int(port)) if smtp_debug: srv.set_debuglevel(1) if use_tls or login: srv.ehlo() self._log_debug(u'Sent initial EHLO command') if use_tls: srv.starttls() srv.ehlo() self._log_debug(u'Sent STARTTLS and EHLO command') if login: srv.login(user, pwd) self._log_debug(u'Sent LOGIN command') srv.sendmail(smtp_from, to, msg.as_string()) srv.quit() except (BaseException, Exception) as e: self.last_err = '%s' % ex(e) return False return True @staticmethod def _get_recipients(show_name=None): email_list = [] # Grab the global recipients if sickgear.EMAIL_LIST: for email_address in sickgear.EMAIL_LIST.split(','): if any(email_address.strip()): email_list.append(email_address) # Grab the recipients for the show if None is not show_name: my_db = db.DBConnection() for result in my_db.select('SELECT notify_list FROM tv_shows WHERE show_name = ?', (show_name,)): if result['notify_list']: for email_address in result['notify_list'].split(','): if any(email_address.strip()): email_list.append(email_address) return list(set(email_list)) def _notify(self, title, body, lang='', extra='', **kwargs): show_name = body.split(' - ')[0] to = self._get_recipients(show_name) if not any(to): self._log_warning(u'No email recipients to notify, skipping') return self._log_debug(u'Email recipients to notify: %s' % to) try: msg = MIMEMultipart('alternative') msg.attach(MIMEText( '' + '

SickGear Notification - %s

\n' % title + '

Show: ' + show_name.encode('ascii', 'xmlcharrefreplace') + '

Episode: ' + text_type(re.search('.+ - (.+?-.+) -.+', body).group(1)).encode('ascii', 'xmlcharrefreplace') + extra + '

\n\n' + '', 'html')) except (BaseException, Exception): try: msg = MIMEText(body) except (BaseException, Exception): msg = MIMEText('Episode %s' % title) msg['Subject'] = '%s%s: %s' % (lang, title, body) msg['From'] = sickgear.EMAIL_FROM msg['To'] = ','.join(to) msg['Date'] = formatdate(localtime=True) if self._sendmail(sickgear.EMAIL_HOST, sickgear.EMAIL_PORT, sickgear.EMAIL_FROM, sickgear.EMAIL_TLS, sickgear.EMAIL_USER, sickgear.EMAIL_PASSWORD, to, msg): self._log_debug(u'%s notification sent to [%s] for "%s"' % (title, to, body)) else: self._log_error(u'%s notification ERROR: %s' % (title, self.last_err)) def test_notify(self, host, port, smtp_from, use_tls, user, pwd, to): self._testing = True msg = MIMEText('Success. This is a SickGear test message. Typically sent on, %s' % notify_strings['download']) msg['Subject'] = 'SickGear: Test message' msg['From'] = smtp_from msg['To'] = to msg['Date'] = formatdate(localtime=True) r = self._sendmail(host, port, smtp_from, use_tls, user, pwd, [to], msg, True) return self._choose(('Success, notification sent.', 'Failed to send notification: %s' % self.last_err)[not r], r) def notify_snatch(self, ep_obj, title=None, **kwargs): """ Send a notification that an episode was snatched :param ep_obj: The snatched episode :param title: The title of the notification (optional) """ title = sickgear.EMAIL_OLD_SUBJECTS and 'Snatched' or title or notify_strings['snatch'] # noinspection PyProtectedMember self._notify(title, self.pretty_name(ep_obj)) def notify_download(self, ep_obj, title=None, **kwargs): """ Send a notification that an episode was downloaded :param ep_obj: The downloaded episode :param title: The title of the notification (optional) """ title = sickgear.EMAIL_OLD_SUBJECTS and 'Downloaded' or title or notify_strings['download'] self._notify(title, self.pretty_name(ep_obj)) def notify_subtitle_download(self, ep_obj, lang, title=None, **kwargs): """ Send a notification that a subtitle was downloaded :param ep_obj: The downloaded episode :param lang: Subtitle language :param title: The title of the notification (optional) """ title = sickgear.EMAIL_OLD_SUBJECTS and 'Subtitle Downloaded' or title or notify_strings['subtitle_download'] self._notify(title, ep_obj.pretty_name(), '%s ' % lang, '

Language: %s' % lang) notifier = EmailNotifier