From fb8a43fd5d4bb87a90ab7b6e4fa4812df3dffba6 Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Fri, 18 Dec 2020 10:44:42 +0100
Subject: [PATCH] use decode_text() in add_header()

---
 app/email_utils.py | 30 ++++++++++++------------------
 1 file changed, 12 insertions(+), 18 deletions(-)

diff --git a/app/email_utils.py b/app/email_utils.py
index 3855c915..795d46a7 100644
--- a/app/email_utils.py
+++ b/app/email_utils.py
@@ -763,36 +763,30 @@ def add_header(msg: Message, text_header, html_header) -> Message:
         payload = msg.get_payload()
         if type(payload) is str:
             clone_msg = copy(msg)
-            to_append = encode_text(f"{text_header}\n---\n", encoding)
-            payload = f"{to_append}{payload}"
-            clone_msg.set_payload(payload)
+            new_payload = f"""{text_header}
+---
+{decode_text(payload, encoding)}"""
+            clone_msg.set_payload(encode_text(new_payload, encoding))
             return clone_msg
     elif msg.get_content_type() == "text/html":
         encoding = get_encoding(msg)
         payload = msg.get_payload()
         if type(payload) is str:
-            new_payload = (
-                encode_text(
-                    f"""
-<table width="100%" style="width: 100%; -premailer-width: 100%; -premailer-cellpadding: 0; -premailer-cellspacing: 0; margin: 0; padding: 0;">
+            new_payload = f"""<table width="100%" style="width: 100%; -premailer-width: 100%; -premailer-cellpadding: 0;
+  -premailer-cellspacing: 0; margin: 0; padding: 0;">
     <tr>
         <td style="border-bottom:1px dashed #5675E2; padding: 10px 0px">{html_header}</td>
     </tr>
     <tr>
-        <td>""",
-                    encoding,
-                )
-                + payload
-                + encode_text(
-                    """</td>
+        <td>
+        {decode_text(payload, encoding)}
+        </td>
     </tr>
 </table>
-            """,
-                    encoding,
-                )
-            )
+"""
+
             clone_msg = copy(msg)
-            clone_msg.set_payload(new_payload)
+            clone_msg.set_payload(encode_text(new_payload, encoding))
             return clone_msg
     elif msg.get_content_type() in ("multipart/alternative", "multipart/related"):
         new_parts = []