Skip to content

Instantly share code, notes, and snippets.

@mattupstate
Created November 26, 2011 19:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save mattupstate/1396163 to your computer and use it in GitHub Desktop.
Save mattupstate/1396163 to your computer and use it in GitHub Desktop.
Amazon SES email logging handler for Python
import logging
import types
from boto.ses import SESConnection
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
from email.utils import COMMASPACE
class SESHandler(logging.Handler):
"""
A handler class which sends an email using Amazon SES.
"""
def __init__(self, aws_key, aws_secret, fromaddr, toaddrs, subject):
"""
Initialize the handler.
Initialize the instance with the AWS account key and secret, from and
to addresses and subject line of the email.
"""
logging.Handler.__init__(self)
self.aws_key = aws_key
self.aws_secret = aws_secret
self.fromaddr = fromaddr
if isinstance(toaddrs, basestring):
toaddrs = [toaddrs]
self.toaddrs = toaddrs
self.subject = subject
def getSubject(self, record):
"""
Determine the subject for the email.
If you want to specify a subject line which is record-dependent,
override this method.
"""
return self.subject
def emit(self, record):
"""
Emit a record.
Format the record and send it to the specified addressees.
"""
client = SESConnection(self.aws_key, self.aws_secret)
message = MIMEMultipart('alternative')
message.set_charset('UTF-8')
message['Subject'] = self._encode_str(self.getSubject(record))
message['From'] = self._encode_str(self.fromaddr)
message['To'] = self._convert_to_strings(self.toaddrs)
from email.utils import formatdate
body = self.format(record)
body = "From: %s\r\n" \
"To: %s\r\n" \
"Subject: %s\r\n" \
"Date: %s\r\n\r\n" \
"%s" % (
self.fromaddr,
",".join(self.toaddrs),
self.getSubject(record),
formatdate(),
body)
message.attach(MIMEText(self._encode_str(body), 'plain'))
return client.send_raw_email(message.as_string(), self.fromaddr,
destinations=self.toaddrs)
def _convert_to_strings(self, list_of_strs):
if isinstance(list_of_strs, (list, tuple)):
result = COMMASPACE.join(list_of_strs)
else:
result = list_of_strs
return self._encode_str(result)
def _encode_str(self, s):
if type(s) == types.UnicodeType:
return s.encode('utf8')
return s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment