jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 07:07:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #435 from toddeye/notify-smtp-retry

Browse Source 
Added retry logic if the SMTP connection is disconnected by the server.
This commit is contained in:
Paulus Schoutsen 2015-09-24 17:15:05 -07:00
commit 9f986c55e6
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
homeassistant/components/notify/smtp.py
View File

@ -140,13 +140,19 @@ class MailNotificationService(BaseNotificationService):
self.username = username self.username = username
self.password = password self.password = password
self.recipient = recipient self.recipient = recipient
self.tries = 2
self.mail = None
self.connect()
def connect(self):
""" Connect/Authenticate to SMTP Server """
self.mail = smtplib.SMTP(self._server, self._port) self.mail = smtplib.SMTP(self._server, self._port)
self.mail.ehlo_or_helo_if_needed() self.mail.ehlo_or_helo_if_needed()
if self.starttls == 1: if self.starttls == 1:
self.mail.starttls() self.mail.starttls()
self.mail.ehlo() self.mail.ehlo()
self.mail.login(self.username, self.password) self.mail.login(self.username, self.password)
def send_message(self, message="", **kwargs): def send_message(self, message="", **kwargs):
@ -160,4 +166,12 @@ class MailNotificationService(BaseNotificationService):
msg['From'] = self._sender msg['From'] = self._sender
msg['X-Mailer'] = 'HomeAssistant' msg['X-Mailer'] = 'HomeAssistant'
self.mail.sendmail(self._sender, self.recipient, msg.as_string()) for _ in range(self.tries):
try:
self.mail.sendmail(self._sender, self.recipient,
msg.as_string())
break
except smtplib.SMTPException:
_LOGGER.warning('SMTPException sending mail: '
'retrying connection')
self.connect()