Changeset 101

Timestamp:

10/04/08 08:29:34 (3 months ago)

Author:

fs

Message:

adds support for non ascii characters in recipient names (#68)

Location:

trunk

Files:

• tests/test_message.py (modified) (1 diff)
• turbomail/message.py (modified) (1 diff)
• turbomail/util.py (modified) (2 diffs)

trunk/tests/test_message.py

@@ -59 +59 @@
         self.failUnless(subject_string in msg_string)
     
-    
+    def test_from_with_umlaut(self):
+        from_name = u"Karl MÃŒller"
+        from_email = u"karl.mueller@example.com"
+        
+        self.message.author = [(from_name, from_email)]
+        self.message.encoding = "ISO-8859-1"
+        
+        msg_string = str(self.message)
+        encoded_name = str(Header(from_name, "ISO-8859-1"))
+        from_string = "From: %s <%s>" % (encoded_name, from_email)
+        self.failUnless(from_string in msg_string)

trunk/turbomail/message.py

@@ -219 +219 @@
                 if isinstance(header[1], unicode):
                     header[1] = Header(header[1], self.encoding)
+                elif isinstance(header[1], AddressList):
+                    header[1] = header[1].encode(self.encoding)
                 header[1] = str(header[1])
                 message.add_header(*header)

trunk/turbomail/util.py

@@ -55 +55 @@
     
     def __str__(self):
+        return self.encode()
+    
+    def encode(self, encoding="ascii"):
         assert self.address, "You must specify an address."
-        
-        return formataddr((self.name, self.address)).replace("\n", "")
-    
+        name_string = self.name
+        if isinstance(name_string, unicode):
+            name_string = str(Header(name_string, encoding))
+        return formataddr((name_string, self.address)).replace("\n", "")
 
 
 class AddressList(list):
-    def __init__(self, addresses=[]):
+    def __init__(self, addresses=[], encoding=None):
         super(AddressList, self).__init__()
         
+        self.encoding = encoding
         if isinstance(addresses, (basestring, tuple)):
             addresses = [addresses]
@@ -76 +81 @@
     
     def __str__(self):
-        return ",\n ".join([str(i) for i in self])
+        return self.encode(self.encoding)
+    
+    def encode(self, encoding="ascii"):
+        addresses = []
+        for address in self:
+            addresses.append(address.encode(encoding))
+        return ",\n ".join(addresses)
     
     def __setitem__(self, k, value):