No such thing as bytes in Python 2

TK-IT · May 1, 2015 · 39c1ef6 · 39c1ef6
1 parent 911f831
commit 39c1ef6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/emailtunnel/__init__.py b/emailtunnel/__init__.py
@@ -83,7 +83,10 @@ def __init__(self, message=None):
         if message:
             assert isinstance(message, bytes)
 
-            self.message = email.message_from_bytes(message)
+            if six.PY3:
+                self.message = email.message_from_bytes(message)
+            else:
+                self.message = email.message_from_string(message)
 
             if not self._sanity_check(message):
                 self._sanity_log_invalid(message)