diff options
author | Kyle Stewart <4b796c65+bitbucket@gmail.com> | 2016-03-17 05:31:33 -0700 |
---|---|---|
committer | Kyle Stewart <4b796c65+bitbucket@gmail.com> | 2016-03-17 05:31:33 -0700 |
commit | f9508d6b662f9d35ca44d3c781947206e0e6c5c0 (patch) | |
tree | c7a36c7871bed3990ac8a3d64bf77ba2f555fc4c | |
parent | e992a5bb8ed77fe6f9e46da1e2add3c5f12038f8 (diff) | |
download | wheel-f9508d6b662f9d35ca44d3c781947206e0e6c5c0.tar.gz |
use context manager when writing out record file
-rw-r--r-- | wheel/install.py | 9 |
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 |