|
|
|
@ -2,10 +2,9 @@ import io |
|
|
|
|
import os |
|
|
|
|
|
|
|
|
|
class CappedBufferedReader(io.BufferedReader): |
|
|
|
|
def __init__(self, raw, buffer_size=io.DEFAULT_BUFFER_SIZE, capped_size=None, fake_start=None): |
|
|
|
|
def __init__(self, raw, buffer_size=io.DEFAULT_BUFFER_SIZE, capped_size=None): |
|
|
|
|
super().__init__(raw, buffer_size) |
|
|
|
|
self.capped_size = capped_size |
|
|
|
|
self.fake_start = fake_start |
|
|
|
|
|
|
|
|
|
def read(self, size=None): |
|
|
|
|
if self.capped_size is not None: |
|
|
|
@ -25,8 +24,6 @@ class CappedBufferedReader(io.BufferedReader): |
|
|
|
|
else: |
|
|
|
|
offset = self.capped_size |
|
|
|
|
whence = os.SEEK_SET |
|
|
|
|
elif whence == os.SEEK_SET and self.fake_start is not None: |
|
|
|
|
offset += self.fake_start |
|
|
|
|
return super().seek(offset, whence) |
|
|
|
|
|
|
|
|
|
@property |
|
|
|
@ -40,5 +37,5 @@ class CappedBufferedReader(io.BufferedReader): |
|
|
|
|
def name(self, new_name): |
|
|
|
|
self._name = new_name |
|
|
|
|
|
|
|
|
|
def bopen(file, capped_size, fake_start): |
|
|
|
|
return CappedBufferedReader(open(file, 'rb', buffering=0), capped_size=capped_size, fake_start=fake_start) |
|
|
|
|
def bopen(file, capped_size): |
|
|
|
|
return CappedBufferedReader(open(file, 'rb', buffering=0), capped_size=capped_size) |
|
|
|
|