PLC/sendmail.py

   1 import os
   2 import sys
   3 import pprint
   4 from email.mime.text import MIMEText
   5 from email.header import Header
   6 from smtplib import SMTP
   7
   8 from PLC.Logger import logger
   9 from PLC.Faults import *
  10
  11 def sendmail(api, To, Subject, Body, From = None, Cc = None, Bcc = None):
  12     """
  13     Uses sendmail (must be installed and running locally) to send a
  14     message to the specified recipients. If the API is running under
  15     mod_python, the apache user must be listed in e.g.,
  16     /etc/mail/trusted-users.
  17
  18     To, Cc, and Bcc may be addresses or lists of addresses. Each
  19     address may be either a plain text address or a tuple of (name,
  20     address).
  21     """
  22
  23     # Fix up defaults
  24     if not isinstance(To, list):
  25         To = [To]
  26     if Cc is not None and not isinstance(Cc, list):
  27         Cc = [Cc]
  28     if Bcc is not None and not isinstance(Bcc, list):
  29         Bcc = [Bcc]
  30     if From is None:
  31         From = ("%s Support" % api.config.PLC_NAME,
  32                 api.config.PLC_MAIL_SUPPORT_ADDRESS)
  33
  34     # Create a MIME-encoded UTF-8 message
  35     msg = MIMEText(Body.encode(api.encoding), _charset = api.encoding)
  36
  37     # Unicode subject headers are automatically encoded correctly
  38     msg['Subject'] = Subject
  39
  40     def encode_addresses(addresses, header_name = None):
  41         """
  42         Unicode address headers are automatically encoded by
  43         email.Header, but not correctly. The correct way is to put the
  44         textual name inside quotes and the address inside brackets:
  45
  46         To: "=?utf-8?b?encoded" <recipient@domain>
  47
  48         Each address in addrs may be a tuple of (name, address) or
  49         just an address. Returns a tuple of (header, addrlist)
  50         representing the encoded header text and the list of plain
  51         text addresses.
  52         """
  53
  54         header = []
  55         addrs = []
  56
  57         for addr in addresses:
  58             if isinstance(addr, tuple):
  59                 (name, addr) = addr
  60                 try:
  61                     name = name.encode('ascii')
  62                     header.append('%s <%s>' % (name, addr))
  63                 except:
  64                     h = Header(name, charset = api.encoding, header_name = header_name)
  65                     header.append('"%s" <%s>' % (h.encode(), addr))
  66             else:
  67                 header.append(addr)
  68             addrs.append(addr)
  69
  70         return (", ".join(header), addrs)
  71
  72     (msg['From'], from_addrs) = encode_addresses([From], 'From')
  73     (msg['To'], to_addrs) = encode_addresses(To, 'To')
  74
  75     if Cc is not None:
  76         (msg['Cc'], cc_addrs) = encode_addresses(Cc, 'Cc')
  77         to_addrs += cc_addrs
  78
  79     if Bcc is not None:
  80         (unused, bcc_addrs) = encode_addresses(Bcc, 'Bcc')
  81         to_addrs += bcc_addrs
  82
  83     # Needed to pass some spam filters
  84     msg['Reply-To'] = msg['From']
  85     msg['X-Mailer'] = "Python/" + sys.version.split(" ")[0]
  86
  87     if not api.config.PLC_MAIL_ENABLED:
  88         logger.info("PLC_MAIL_ENABLED not set")
  89         logger.info("From: %(From)s, To: %(To)s, Subject: %(Subject)s" % msg)
  90         return
  91
  92     s = SMTP()
  93     s.connect()
  94     rejected = s.sendmail(from_addrs[0], to_addrs, msg.as_string(), rcpt_options = ["NOTIFY=NEVER"])
  95     s.close()
  96
  97     if rejected:
  98         raise PLCAPIError("Error sending message to " + ", ".join(list(rejected.keys())))