Simplify admin preview of emails

mhagander · mhagander · commit 74d9a03f3fce · 2020-07-13T14:57:17.000+02:00
Use the python3 function to get the plaintext body of the email, instead
of our own very limited one we had before.
diff --git a/pgweb/mailqueue/admin.py b/pgweb/mailqueue/admin.py
@@ -1,6 +1,7 @@
 from django.contrib import admin
 
 from email.parser import Parser
+from email import policy
 
 from .models import QueuedMail
 
@@ -13,18 +14,9 @@ def parsed_content(self, obj):
         # We only try to parse the *first* piece, because we assume
         # all our emails are trivial.
         try:
-            parser = Parser()
+            parser = Parser(policy=policy.default)
             msg = parser.parsestr(obj.fullmsg)
-            b = msg.get_payload(decode=True)
-            if b:
-                return b.decode('utf8')
-
-            pl = msg.get_payload()
-            for p in pl:
-                b = p.get_payload(decode=True)
-                if b:
-                    return b.decode('utf8')
-            return "Could not find body"
+            return msg.get_body(preferencelist=('plain', )).get_payload(decode=True).decode('utf8')
         except Exception as e:
             return "Failed to get body: %s" % e
 
