commit 569e61f35196cb765eb9940f75b27bb765e73073
author Ezio Melotti <ezio.melotti@gmail.com> Wed Dec 30 06:14:51 2009 +0000
committer Ezio Melotti <ezio.melotti@gmail.com> Wed Dec 30 06:14:51 2009 +0000
tree 3bf8941cf86c15a9a6f9c4afbeb68211436e826b
parent eb74da8e67bed4beef1bfc97384b3aa6e0f9836b

#5511: Added the ability to use ZipFile as a context manager. Patch by Brian Curtin.

Lib/zipfile.py

diff --git a/Lib/zipfile.py b/Lib/zipfile.py
index 2aa4bff..89b4166 100644
--- a/Lib/zipfile.py
+++ b/Lib/zipfile.py
@@ -721,6 +721,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."""