瀏覽代碼

handle case msg.get_payload(decode=True) is None

Son NK 4 年之前
父節點
當前提交
756e8080ab
共有 1 個文件被更改,包括 6 次插入1 次删除
  1. 6 1
      email_handler.py

+ 6 - 1
email_handler.py

@@ -338,7 +338,12 @@ def replace_header_when_reply(msg: Message, alias: Alias, header: str):
 def replace_str_in_msg(msg: Message, fr: str, to: str):
     if msg.get_content_maintype() != "text":
         return msg
-    new_body = msg.get_payload(decode=True).replace(fr.encode(), to.encode())
+
+    msg_payload = msg.get_payload(decode=True)
+    if not msg_payload:
+        return msg
+
+    new_body = msg_payload.replace(fr.encode(), to.encode())
 
     # If utf-8 decoding fails, do not touch message part
     try: