Merge pull request #119 from yeralin/fix-read-stream

Check availability before conditionals
This commit is contained in:
碧舞 すみほ 2022-03-10 20:00:40 +09:00 committed by GitHub
commit 287fbbc2d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions

View File

@ -177,8 +177,8 @@ class AbsChunkedInputStream(io.BytesIO, HaltListener):
if chunk_end > self.size():
chunk_end = int(self.size() / (128 * 1024))
chunk_end_off = int(self.size() % (128 * 1024))
self.check_availability(chunk, True, False)
if chunk_off + __size > len(self.buffer()[chunk]):
self.check_availability(chunk, True, False)
buffer.write(self.buffer()[chunk][chunk_off:])
chunk += 1
while chunk <= chunk_end:
@ -189,7 +189,6 @@ class AbsChunkedInputStream(io.BytesIO, HaltListener):
buffer.write(self.buffer()[chunk])
chunk += 1
else:
self.check_availability(chunk, True, False)
buffer.write(self.buffer()[chunk][chunk_off:chunk_off + __size])
buffer.seek(0)
self.__pos += buffer.getbuffer().nbytes