use context manager when writing out record file

1 files changed, 5 insertions, 4 deletions

diff --git a/wheel/install.py b/wheel/install.py
index 3af6d0c..9d48efe 100644
--- a/wheel/install.py
+++ b/wheel/install.py
@@ -360,10 +360,11 @@ class WheelFile(object):
                 os.chmod(dest, info.external_attr >> 16)
 
         record_name = os.path.join(root, self.record_name)
-        writer = csv.writer(open_for_csv(record_name, 'w+'))
-        for reldest, digest, length in sorted(record_data):
-            writer.writerow((reldest, digest, length))
-        writer.writerow((self.record_name, '', ''))
+        with open_for_csv(record_name, 'w+') as record_file:
+            writer = csv.writer(record_file)
+            for reldest, digest, length in sorted(record_data):
+                writer.writerow((reldest, digest, length))
+            writer.writerow((self.record_name, '', ''))
 
     def verify(self, zipfile=None):
         """Configure the VerifyingZipFile `zipfile` by verifying its signature