From f9508d6b662f9d35ca44d3c781947206e0e6c5c0 Mon Sep 17 00:00:00 2001 From: Kyle Stewart <4b796c65+bitbucket@gmail.com> Date: Thu, 17 Mar 2016 05:31:33 -0700 Subject: use context manager when writing out record file --- wheel/install.py | 9 +++++---- 1 file 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 -- cgit v1.2.1