From f8635602757502fe27d63794c3a3511fc312f9ad Mon Sep 17 00:00:00 2001 From: Lucas Hoffmann Date: Sat, 17 Dec 2016 13:07:59 +0100 Subject: Use with blocks to write to temp files --- alot/db/utils.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) (limited to 'alot/db') diff --git a/alot/db/utils.py b/alot/db/utils.py index 188a4b3a..ab20e26b 100644 --- a/alot/db/utils.py +++ b/alot/db/utils.py @@ -314,13 +314,11 @@ def extract_body(mail, types=None, field_key='copiousoutput'): # open tempfile, respect mailcaps nametemplate nametemplate = entry.get('nametemplate', '%s') prefix, suffix = parse_mailcap_nametemplate(nametemplate) - tmpfile = tempfile.NamedTemporaryFile(delete=False, - prefix=prefix, - suffix=suffix) - # write payload to tmpfile - tmpfile.write(raw_payload) - tmpfile.close() - tempfile_name = tmpfile.name + with tempfile.NamedTemporaryFile( + delete=False, prefix=prefix, suffix=suffix) \ + as tmpfile: + tmpfile.write(raw_payload) + tempfile_name = tmpfile.name else: stdin = raw_payload -- cgit v1.2.3