diff options
Diffstat (limited to 'Lib/zipfile.py')
| -rw-r--r-- | Lib/zipfile.py | 6 |
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.""" |
