spacefreak/pyquarantine-milter
1
0
Fork 0
Code Issues Pull Requests Releases Activity

fix html code in text body when html part is injected

Browse Source
This commit is contained in:
Thomas Oettli
2022-08-12 15:09:30 +02:00
parent a4c2ec3952
commit 91144643f3
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pyquarantine/base.py
View File

@@ -162,8 +162,9 @@ class MilterMessage(MIMEPart):
def inject_body_part(part, content, subtype="plain"): def inject_body_part(part, content, subtype="plain"):
parts = [] parts = []
text_body = None text_body = None
text_content = None
if subtype == "html": if subtype == "html":
text_body = part.get_body(preferencelist=("plain")) text_body, text_content = part.get_body_content("plain")
for p in part.iter_parts(): for p in part.iter_parts():
if text_body and p == text_body: if text_body and p == text_body:
@@ -173,8 +174,8 @@ def inject_body_part(part, content, subtype="plain"):
boundary = part.get_boundary() boundary = part.get_boundary()
p_subtype = part.get_content_subtype() p_subtype = part.get_content_subtype()
part.clear_content() part.clear_content()
if text_body: if text_content != None:
part.set_content(content) part.set_content(text_content)
part.add_alternative(content, subtype=subtype) part.add_alternative(content, subtype=subtype)
else: else:
part.set_content(content, subtype=subtype) part.set_content(content, subtype=subtype)