- new message_from_string function

- fixed get_payload encoding bug

1 files changed, 8 insertions, 1 deletions

diff --git a/umessage.py b/umessage.py
index a42cb40..b28663f 100644
--- a/umessage.py
+++ b/umessage.py
@@ -21,6 +21,12 @@ def message_from_file(fd):
     except email.Errors.MessageParseError:
         return ''
     
+def message_from_string(string):
+    try:
+        return UMessage(email.message_from_string(string))
+    except email.Errors.MessageParseError:
+        return ''
+    
 class UMessage:
     """Encapsulates an email.Message instance and returns only unicode objects"""
 
@@ -47,7 +53,8 @@ class UMessage:
                 return [UMessage(msg) for msg in payload]
             if message.get_content_maintype() != 'text':
                 return payload
-            return unicode(payload or '', message.get_charset() or 'iso-8859-15')
+            return unicode(payload or '', message.get_content_charset()
+                           or 'iso-8859-15')
         else:
             payload = UMessage(message.get_payload(index, decode))
         return payload