From: W. Trevor King <wking@tremily.us>
Date: Thu, 24 Jan 2013 03:59:55 +0000 (-0500)
Subject: email: Factor message-to-bytes formatting out into _flatten()
X-Git-Tag: v3.0~6
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=8aa81e5ac6d8167951b735fd93511f101c3c2185;p=rss2email.git

email: Factor message-to-bytes formatting out into _flatten()

Signed-off-by: W. Trevor King <wking@tremily.us>
---

diff --git a/rss2email/email.py b/rss2email/email.py
index 90d245d..3b9f983 100644
--- a/rss2email/email.py
+++ b/rss2email/email.py
@@ -159,9 +159,59 @@ def smtp_send(sender, recipient, message, config=None, section='DEFAULT'):
     smtp.send_message(message, sender, [recipient])
     smtp.quit()
 
+def _flatten(message):
+    r"""Flatten an email.message.Message to bytes
+
+    >>> import rss2email.config
+    >>> config = rss2email.config.Config()
+    >>> config.read_dict(rss2email.config.CONFIG)
+
+    Here's a 7-bit, base64 version:
+
+    >>> message = get_message(
+    ...     sender='John <jdoe@a.com>', recipient='Ζεύς <z@olympus.org>',
+    ...     subject='Homage',
+    ...     body="You're great, Ζεύς!\\n",
+    ...     content_type='plain',
+    ...     config=config)
+    >>> for line in _flatten(message).split(b'\n'):
+    ...     print(line)
+    b'MIME-Version: 1.0'
+    b'Content-Type: text/plain; charset="utf-8"'
+    b'Content-Transfer-Encoding: base64'
+    b'From: John <jdoe@a.com>'
+    b'To: =?utf-8?b?zpbOtc+Nz4I=?= <z@olympus.org>'
+    b'Subject: Homage'
+    b''
+    b'WW91J3JlIGdyZWF0LCDOls61z43PgiFcbg=='
+    b''
+
+    Here's an 8-bit version:
+
+    >>> config.set('DEFAULT', 'use-8bit', str(True))
+    >>> message = get_message(
+    ...     sender='John <jdoe@a.com>', recipient='Ζεύς <z@olympus.org>',
+    ...     subject='Homage',
+    ...     body="You're great, Ζεύς!\\n",
+    ...     content_type='plain',
+    ...     config=config)
+    >>> for line in _flatten(message).split(b'\n'):
+    ...     print(line)
+    b'MIME-Version: 1.0'
+    b'Content-Type: text/plain; charset="utf-8"'
+    b'From: John <jdoe@a.com>'
+    b'To: =?utf-8?b?zpbOtc+Nz4I=?= <z@olympus.org>'
+    b'Subject: Homage'
+    b'Content-Transfer-Encoding: 8bit'
+    b''
+    b"You're great, \xce\x96\xce\xb5\xcf\x8d\xcf\x82!\\n"
+    """
+    return message.as_string().encode(str(message.get_charset()))
+
 def sendmail_send(sender, recipient, message, config=None, section='DEFAULT'):
     if config is None:
         config = _config.CONFIG
+    message_bytes = _flatten(message)
     _LOG.debug(
         'sending message to {} via /usr/sbin/sendmail'.format(recipient))
     try:
@@ -169,8 +219,7 @@ def sendmail_send(sender, recipient, message, config=None, section='DEFAULT'):
             ['/usr/sbin/sendmail', '-f', sender, recipient],
             stdin=_subprocess.PIPE, stdout=_subprocess.PIPE,
             stderr=_subprocess.PIPE)
-        stdout,stderr = p.communicate(message.as_string()
-                                      .encode(str(message.get_charset())))
+        stdout,stderr = p.communicate(message_bytes)
         status = p.wait()
         if status:
             raise _error.SendmailError(