From: Oleg Broytman <phd@phdru.name>
Date: Sun, 27 Aug 2017 17:04:33 +0000 (+0300)
Subject: Fix: Recode text from mailcap command from default encoding to charset
X-Git-Tag: 2.8.0~31
X-Git-Url: https://git.phdru.name/?p=mimedecode.git;a=commitdiff_plain;h=5e0a0832c3c713d57cdc01b91b9e6a1c4ccc15f3

Fix: Recode text from mailcap command from default encoding to charset
---

diff --git a/mimedecode.py b/mimedecode.py
index 9433b0d..5459cbf 100755
--- a/mimedecode.py
+++ b/mimedecode.py
@@ -256,13 +256,15 @@ def decode_body(msg, s):
     pipe.stdout.close()
     if pipe.wait() == 0: # result=0, Ok
         s = new_s
-    os.remove(filename)
-
-    set_content_type(msg, "text/plain")
-    if s is new_s:
+        if bytes is not str and isinstance(s, bytes):  # Python3
+            s = s.decode(g.default_encoding, "replace")
+        if charset and not isinstance(s, bytes):
+            s = s.encode(charset, "replace")
+        set_content_type(msg, "text/plain")
         msg["X-MIME-Autoconverted"] = "from %s to text/plain by %s id %s" % (content_type, g.host_name, command.split()[0])
     else:
         msg["X-MIME-Autoconverted"] = "failed conversion from %s to text/plain by %s id %s" % (content_type, g.host_name, command.split()[0])
+    os.remove(filename)
 
     return s