Refactor: Email parser use with to handle buffer (#11496)

### What problem does this PR solve?
 Email parser use with to handle buffer

### Type of change

- [x] Refactoring
This commit is contained in:
Stephen Hu
2025-11-25 10:03:37 +08:00
committed by GitHub
parent d1744aaaf3
commit 41665b0865

View File

@ -51,9 +51,11 @@ def chunk(
attachment_res = []
if binary:
msg = BytesParser(policy=policy.default).parse(io.BytesIO(binary))
with io.BytesIO(binary) as buffer:
msg = BytesParser(policy=policy.default).parse(buffer)
else:
msg = BytesParser(policy=policy.default).parse(open(filename, "rb"))
with open(filename, "rb") as buffer:
msg = BytesParser(policy=policy.default).parse(buffer)
text_txt, html_txt = [], []
# get the email header info