From 7cb7b65922b592e7d8a47bffcaaa9e0cbf9320b4 Mon Sep 17 00:00:00 2001 From: Bruce Flynn <brucef@ssec.wisc.edu> Date: Thu, 15 Oct 2015 15:22:51 +0000 Subject: [PATCH] Remove offset from PacketTooShort to fix fobj.tell() err --- edosl0util/stream.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/edosl0util/stream.py b/edosl0util/stream.py index f06ac84..2c7c075 100644 --- a/edosl0util/stream.py +++ b/edosl0util/stream.py @@ -31,10 +31,9 @@ class PacketTooShort(Error): After this error occurs the stream is no longer usable because data offsets are no longer reliable. """ - def __init__(self, header=None, offset=None): - self.offset = offset + def __init__(self, header=None): self.primary_header = header - self.args = (self.offset, self.primary_header) + self.args = (self.primary_header,) class NonConsecutiveSeqId(Error): @@ -49,7 +48,6 @@ class SimpleStream(object): """ def __init__(self, fobj): self.fobj = fobj - self.offset = fobj.tell() def __iter__(self): return self @@ -63,14 +61,13 @@ class SimpleStream(object): if not buf: raise StopIteration() if len(buf) < psize: - raise PacketTooShort(header=None, offset=self.offset) + raise PacketTooShort(header=None) h1 = PrimaryHeader.from_buffer_copy(buf) # read user data size = h1.data_length_minus1 + 1 buf = self.fobj.read(size) if len(buf) < size: - raise PacketTooShort(header=h1, offset=self.offset) - self.offset += (psize + size) + raise PacketTooShort(header=h1) return h1, buf -- GitLab