blankie
/
autoytarchive
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

hope i fixed the bug

This commit is contained in:
blank X 2021-04-08 11:11:14 +07:00
parent d0d7009886
commit d05e04e73a
Signed by: blankie
GPG Key ID: CC15FC822C7F61F5
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
autoytarchive/cappedio.py
View File

@ -14,16 +14,22 @@ class CappedBufferedReader(io.BufferedReader):
return super().read(size) return super().read(size)
def seek(self, offset, whence=os.SEEK_SET): def seek(self, offset, whence=os.SEEK_SET):
if offset == 0 and whence == os.SEEK_END and self.capped_size is not None: if self.capped_size is not NOne:
if self.capped_size < 0: if offset == 0 and whence == os.SEEK_END:
offset = 0 if self.capped_size < 0:
else: offset = 0
real_end = super().seek(0, os.SEEK_END)
if self.capped_size > real_end:
offset = real_end
else: else:
offset = self.capped_size real_end = super().seek(0, os.SEEK_END)
whence = os.SEEK_SET if self.capped_size > real_end:
offset = real_end
else:
offset = self.capped_size
self.capped_size -= offset
whence = os.SEEK_SET
elif whence == os.SEEK_SET:
current_pos = self.tell()
if current_pos > offset:
self.capped_size += current_pos - offset
return super().seek(offset, whence) return super().seek(offset, whence)
@property @property