Fix email parsing after python3 upgrade

1 files changed, 3 insertions, 2 deletions

diff --git a/emailcanary/emailutils.py b/emailcanary/emailutils.py
index 54323a1..2358a3a 100644
--- a/emailcanary/emailutils.py
+++ b/emailcanary/emailutils.py
@@ -1,6 +1,7 @@
 import sys
 import imaplib, email
 
+
 def get_imap(server, username, password):
     '''Connect and login via IMAP'''
     try:
@@ -29,8 +30,8 @@ def get_message(mail, uid):
         if dat0 is None:
             return None
         else:
-            raw_email = str(dat0[1])
-            return email.message_from_string(raw_email)
+            raw_email = dat0[1]
+            return email.message_from_bytes(raw_email)
     else:
         raise Exception("Bad response from server: %s" % (result))