From 308705e4fab1fe784218bde5cc8c8d2daddf6700 Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Sat, 10 Jan 2009 16:22:51 +0000 Subject: Merged revisions 68484-68485 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r68484 | antoine.pitrou | 2009-01-10 17:13:45 +0100 (sam., 10 janv. 2009) | 3 lines Issue #3860: GzipFile and BZ2File now support the context manager protocol. ........ r68485 | antoine.pitrou | 2009-01-10 17:15:24 +0100 (sam., 10 janv. 2009) | 1 line Add NEWS entry for r68484. ........ --- Lib/gzip.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'Lib/gzip.py') diff --git a/Lib/gzip.py b/Lib/gzip.py index 560a722bae9..019c3e2df1f 100644 --- a/Lib/gzip.py +++ b/Lib/gzip.py @@ -472,6 +472,14 @@ class GzipFile: else: raise StopIteration + def __enter__(self): + if self.fileobj is None: + raise ValueError("I/O operation on closed GzipFile object") + return self + + def __exit__(self, *args): + self.close() + def _test(): # Act like gzip; with -d, act like gunzip. -- cgit v1.2.3