Fix: treat first text/plain alternative as plaintext body

Improve handling of alternative parts and `content_subtype`
to match how Django's SMTP backend handles some unusual cases.

Change Test backend to support (and record) text/* alternative
parts. (But still reject other types of alternatives.)

Fixes #252
This commit is contained in:
medmunds
2022-01-10 15:27:12 -08:00
committed by Mike Edmunds
parent 28247ec042
commit 60fbe1e896
4 changed files with 111 additions and 9 deletions

View File

@@ -25,6 +25,20 @@ Release history
^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^
.. This extra heading level keeps the ToC from becoming unmanageably long .. This extra heading level keeps the ToC from becoming unmanageably long
vNext
-----
*Unreleased changes on main branch*
Fixes
~~~~~
* Allow `attach_alternative("content", "text/plain")` in place of setting
an EmailMessage's `body`, and generally improve alternative part
handling for consistency with Django's SMTP EmailBackend.
(Thanks to `@cjsoftuk`_ for reporting the issue.)
v8.4 v8.4
---- ----
@@ -1241,6 +1255,7 @@ Features
.. _@anstosa: https://github.com/anstosa .. _@anstosa: https://github.com/anstosa
.. _@calvin: https://github.com/calvin .. _@calvin: https://github.com/calvin
.. _@chrisgrande: https://github.com/chrisgrande .. _@chrisgrande: https://github.com/chrisgrande
.. _@cjsoftuk: https://github.com/cjsoftuk
.. _@costela: https://github.com/costela .. _@costela: https://github.com/costela
.. _@decibyte: https://github.com/decibyte .. _@decibyte: https://github.com/decibyte
.. _@dominik-lekse: https://github.com/dominik-lekse .. _@dominik-lekse: https://github.com/dominik-lekse

View File

@@ -281,9 +281,7 @@ class BasePayload:
else: else:
value = converter(value) value = converter(value)
if value is not UNSET: if value is not UNSET:
if attr == 'body': if attr == 'from_email':
setter = self.set_html_body if message.content_subtype == 'html' else self.set_text_body
elif attr == 'from_email':
setter = self.set_from_email_list setter = self.set_from_email_list
elif attr == 'extra_headers': elif attr == 'extra_headers':
setter = self.process_extra_headers setter = self.process_extra_headers
@@ -452,6 +450,18 @@ class BasePayload:
# headers is a CaseInsensitiveDict, and is a copy (so is safe to modify) # headers is a CaseInsensitiveDict, and is a copy (so is safe to modify)
self.unsupported_feature('extra_headers') self.unsupported_feature('extra_headers')
def set_body(self, body):
# Interpret message.body depending on message.content_subtype.
# (Subclasses should generally implement set_text_body and set_html_body
# rather than overriding this.)
content_subtype = self.message.content_subtype
if content_subtype == "plain":
self.set_text_body(body)
elif content_subtype == "html":
self.set_html_body(body)
else:
self.add_alternative(body, "text/%s" % content_subtype)
def set_text_body(self, body): def set_text_body(self, body):
raise NotImplementedError("%s.%s must implement set_text_body" % raise NotImplementedError("%s.%s must implement set_text_body" %
(self.__class__.__module__, self.__class__.__name__)) (self.__class__.__module__, self.__class__.__name__))
@@ -461,17 +471,28 @@ class BasePayload:
(self.__class__.__module__, self.__class__.__name__)) (self.__class__.__module__, self.__class__.__name__))
def set_alternatives(self, alternatives): def set_alternatives(self, alternatives):
# Handle treating first text/{plain,html} alternatives as bodies.
# (Subclasses should generally implement add_alternative
# rather than overriding this.)
has_plain_body = self.message.content_subtype == "plain" and self.message.body
has_html_body = self.message.content_subtype == "html" and self.message.body
for content, mimetype in alternatives: for content, mimetype in alternatives:
if mimetype == "text/html": if mimetype == "text/plain" and not has_plain_body:
# This assumes that there's at most one html alternative, self.set_text_body(content)
# and so it should be the html body. (Most ESPs don't has_plain_body = True
# support multiple html alternative parts anyway.) elif mimetype == "text/html" and not has_html_body:
self.set_html_body(content) self.set_html_body(content)
has_html_body = True
else: else:
self.add_alternative(content, mimetype) self.add_alternative(content, mimetype)
def add_alternative(self, content, mimetype): def add_alternative(self, content, mimetype):
self.unsupported_feature("alternative part with type '%s'" % mimetype) if mimetype == "text/plain":
self.unsupported_feature("multiple plaintext parts")
elif mimetype == "text/html":
self.unsupported_feature("multiple html parts")
else:
self.unsupported_feature("alternative part with type '%s'" % mimetype)
def set_attachments(self, attachments): def set_attachments(self, attachments):
for attachment in attachments: for attachment in attachments:

View File

@@ -107,7 +107,12 @@ class TestPayload(BasePayload):
self.params['html_body'] = body self.params['html_body'] = body
def add_alternative(self, content, mimetype): def add_alternative(self, content, mimetype):
self.unsupported_feature("alternative part with type '%s'" % mimetype) # For testing purposes, we allow all "text/*" alternatives,
# but not any other mimetypes.
if mimetype.startswith('text'):
self.params.setdefault('alternatives', []).append((content, mimetype))
else:
self.unsupported_feature("alternative part with type '%s'" % mimetype)
def add_attachment(self, attachment): def add_attachment(self, attachment):
self.params.setdefault('attachments', []).append(attachment) self.params.setdefault('attachments', []).append(attachment)

View File

@@ -426,6 +426,67 @@ class SpecialHeaderTests(TestBackendTestCase):
self.message.send() self.message.send()
class AlternativePartsTests(TestBackendTestCase):
"""Anymail should handle alternative parts consistently with Django's SMTP backend"""
def test_default_usage(self):
"""Body defaults to text/plain, use alternative for html"""
self.message.body = "plain body"
self.message.attach_alternative("html body", "text/html")
self.message.send()
params = self.get_send_params()
self.assertEqual(params['text_body'], "plain body")
self.assertEqual(params['html_body'], "html body")
self.assertNotIn('alternatives', params)
def test_content_subtype_html(self):
"""Change body to text/html, use alternative for plain"""
self.message.content_subtype = "html"
self.message.body = "html body"
self.message.attach_alternative("plain body", "text/plain")
self.message.send()
params = self.get_send_params()
self.assertEqual(params['text_body'], "plain body")
self.assertEqual(params['html_body'], "html body")
self.assertNotIn('alternatives', params)
def test_attach_plain_and_html(self):
"""Use alternatives for both bodies"""
message = AnymailMessage(subject="Subject", from_email="from@example.com", to=["to@example.com"])
message.attach_alternative("plain body", "text/plain")
message.attach_alternative("html body", "text/html")
message.send()
params = self.get_send_params()
self.assertEqual(params['text_body'], "plain body")
self.assertEqual(params['html_body'], "html body")
self.assertNotIn('alternatives', params)
def test_additional_plain_part(self):
"""Two plaintext bodies"""
# In theory this is supported (e.g., for different languages or charsets),
# though MUAs are unlikely to display anything after the first.
self.message.body = "plain body"
self.message.attach_alternative("second plain body", "text/plain")
self.message.send()
params = self.get_send_params()
self.assertEqual(params['text_body'], "plain body")
self.assertEqual(params['alternatives'], [("second plain body", "text/plain")])
def test_exotic_content_subtype(self):
"""Change body to text/calendar, use alternatives for plain and html"""
# This is unlikely to work with most ESPs, but we can try to communicate the intent...
# (You probably want an attachment rather than an alternative part.)
self.message.content_subtype = "calendar"
self.message.body = "BEGIN:VCALENDAR..."
self.message.attach_alternative("plain body", "text/plain")
self.message.attach_alternative("html body", "text/html")
self.message.send()
params = self.get_send_params()
self.assertEqual(params['text_body'], "plain body")
self.assertEqual(params['html_body'], "html body")
self.assertEqual(params['alternatives'], [("BEGIN:VCALENDAR...", "text/calendar")])
class BatchSendDetectionTestCase(TestBackendTestCase): class BatchSendDetectionTestCase(TestBackendTestCase):
"""Tests shared code to consistently determine whether to use batch send""" """Tests shared code to consistently determine whether to use batch send"""