Merged revisions 77136 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r77136 | ezio.melotti | 2009-12-30 08:14:51 +0200 (Wed, 30 Dec 2009) | 1 line
#5511: Added the ability to use ZipFile as a context manager. Patch by Brian Curtin.
........
diff --git a/Lib/zipfile.py b/Lib/zipfile.py
index 79ca152..f70cf63 100644
--- a/Lib/zipfile.py
+++ b/Lib/zipfile.py
@@ -719,6 +719,12 @@
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."""