diff options
| -rw-r--r-- | dedup/compression.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/dedup/compression.py b/dedup/compression.py index 161eda2..ea921c4 100644 --- a/dedup/compression.py +++ b/dedup/compression.py @@ -103,10 +103,9 @@ class DecompressedStream: self.decompressor = decompressor self.buff = b"" self.pos = 0 - self.closed = False def _fill_buff_until(self, predicate): - assert not self.closed + assert self.fileobj is not None data = True while True: if predicate(self.buff) or not data: @@ -143,12 +142,12 @@ class DecompressedStream: return iter(self.readline, b'') def tell(self): - assert not self.closed + assert self.fileobj is not None return self.pos def seek(self, pos): """Forward seeks by absolute position only.""" - assert not self.closed + assert self.fileobj is not None if pos < self.pos: raise ValueError("negative seek not allowed on decompressed stream") while True: @@ -162,12 +161,11 @@ class DecompressedStream: return def close(self): - if not self.closed: + if self.fileobj is not None: self.fileobj.close() self.fileobj = None self.decompressor = None self.buff = b"" - self.closed = True decompressors = { '.gz': GzipDecompressor, |