diff options
Diffstat (limited to 'Lib/packaging/command/upload_docs.py')
| -rw-r--r-- | Lib/packaging/command/upload_docs.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/Lib/packaging/command/upload_docs.py b/Lib/packaging/command/upload_docs.py index 29ea6e9..47e6217 100644 --- a/Lib/packaging/command/upload_docs.py +++ b/Lib/packaging/command/upload_docs.py @@ -18,14 +18,13 @@ from packaging.command.cmd import Command def zip_dir(directory): """Compresses recursively contents of directory into a BytesIO object""" destination = BytesIO() - zip_file = zipfile.ZipFile(destination, "w") - for root, dirs, files in os.walk(directory): - for name in files: - full = os.path.join(root, name) - relative = root[len(directory):].lstrip(os.path.sep) - dest = os.path.join(relative, name) - zip_file.write(full, dest) - zip_file.close() + with zipfile.ZipFile(destination, "w") as zip_file: + for root, dirs, files in os.walk(directory): + for name in files: + full = os.path.join(root, name) + relative = root[len(directory):].lstrip(os.path.sep) + dest = os.path.join(relative, name) + zip_file.write(full, dest) return destination |