Allow creation of Mail from EmailMessage

Author: cmccandless

sendgrid/helpers/mail/mail.py

@@ -141,3 +141,23 @@ def get(self):
 
         return {key: value for key, value in mail.items()
                 if value is not None and value != [] and value != {}}
+
+    @classmethod
+    def from_EmailMessage(cls, message):
+        """Create a Mail object from an instance of
+        email.message.EmailMessage.
+        :type message: email.message.EmailMessage
+        :rtype: Mail
+        """
+        mail = cls(
+            from_email=Email(message.get('From')),
+            subject=message.get('Subject'),
+            to_email=Email(message.get('To')),
+        )
+        mail.add_content(Content(
+            message.get_content_type(),
+            message.get_content()
+        ))
+        for k, v in message.items():
+            mail.add_header(Header(k, v))
+        return mail

test/test_mail.py

@@ -2,6 +2,8 @@
 import json
 import unittest
 
+from email.message import EmailMessage
+
 from sendgrid.helpers.mail import (
     ASM,
     APIKeyIncludedException,
@@ -557,3 +559,21 @@ def test_disable_tracking(self):
     def test_directly_setting_substitutions(self):
         personalization = Personalization()
         personalization.substitutions = [{'a': 0}]
+
+    def test_from_emailmessage(self):
+        message = EmailMessage()
+        message.set_content('message that is not urgent')
+        message.set_default_type('text/plain')
+        message['Subject'] = 'URGENT TITLE'
+        message['From'] = '[email protected]'
+        message['To'] = '[email protected]'
+        mail = Mail.from_EmailMessage(message)
+        self.assertEqual(mail.subject, 'URGENT TITLE')
+        self.assertEqual(mail.from_email.email, '[email protected]')
+        self.assertEqual(len(mail.personalizations), 1)
+        self.assertEqual(len(mail.personalizations[0].tos), 1)
+        self.assertDictEqual(mail.personalizations[0].tos[0], {'email': '[email protected]'})
+        self.assertEqual(len(mail.contents), 1)
+        content = mail.contents[0]
+        self.assertEqual(content.type, 'text/plain')
+        self.assertEqual(content.value, 'message that is not urgent\n')