Use with blocks to write to temp files

1 files changed, 5 insertions, 7 deletions

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