From 7980eaa98d7a4448d3e0a6f336a0dd55f0346b8f Mon Sep 17 00:00:00 2001 From: Antoine Pitrou Date: Wed, 6 Oct 2010 21:21:18 +0000 Subject: Issue #9759: GzipFile now raises ValueError when an operation is attempted after the file is closed. Patch by Jeffrey Finkelstein. --- Lib/gzip.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'Lib/gzip.py') diff --git a/Lib/gzip.py b/Lib/gzip.py index 7c3fd5157e..21b140498f 100644 --- a/Lib/gzip.py +++ b/Lib/gzip.py @@ -210,6 +210,13 @@ class GzipFile(io.BufferedIOBase): s = repr(fileobj) return '' + def _check_closed(self): + """Raises a ValueError if the underlying file object has been closed. + + """ + if self.closed: + raise ValueError('I/O operation on closed file.') + def _init_write(self, filename): self.name = filename self.crc = zlib.crc32(b"") & 0xffffffff @@ -288,6 +295,7 @@ class GzipFile(io.BufferedIOBase): self._add_read_data(uncompress) def write(self,data): + self._check_closed() if self.mode != WRITE: import errno raise IOError(errno.EBADF, "write() on read-only GzipFile object") @@ -308,6 +316,7 @@ class GzipFile(io.BufferedIOBase): return len(data) def read(self, size=-1): + self._check_closed() if self.mode != READ: import errno raise IOError(errno.EBADF, "read() on write-only GzipFile object") @@ -457,6 +466,7 @@ class GzipFile(io.BufferedIOBase): self.myfileobj = None def flush(self,zlib_mode=zlib.Z_SYNC_FLUSH): + self._check_closed() if self.mode == WRITE: # Ensure the compressor's buffer is flushed self.fileobj.write(self.compress.flush(zlib_mode)) -- cgit v1.2.1