summaryrefslogtreecommitdiff
path: root/Lib/zipfile.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/zipfile.py')
-rw-r--r--Lib/zipfile.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/zipfile.py b/Lib/zipfile.py
index 79ca1527f9..f70cf63d7e 100644
--- a/Lib/zipfile.py
+++ b/Lib/zipfile.py
@@ -719,6 +719,12 @@ class ZipFile:
self.fp = None
raise RuntimeError('Mode must be "r", "w" or "a"')
+ def __enter__(self):
+ return self
+
+ def __exit__(self, type, value, traceback):
+ self.close()
+
def _GetContents(self):
"""Read the directory, making sure we close the file if the format
is bad."""