MIMEPart does not exist in Python 2

TK-IT · May 1, 2015 · 1592c8e · 1592c8e
1 parent 161fda0
commit 1592c8e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/emailtunnel/__init__.py b/emailtunnel/__init__.py
@@ -187,7 +187,10 @@ def set_unique_header(self, key, value):
             self.message[key] = value
 
     def set_body_text(self, body, encoding):
-        body_part = email.message.MIMEPart()
+        if six.PY3:
+            body_part = email.message.MIMEPart()
+        else:
+            body_part = email.message.Message()
         if encoding:
             encoded = body.encode(encoding)
             body_part.set_payload(encoded)