Commit 8aaeb2f4 authored by Antoine Pitrou's avatar Antoine Pitrou

Merge

parents 84869872 8280b4ba
...@@ -49,18 +49,6 @@ import socket ...@@ -49,18 +49,6 @@ import socket
import asyncore import asyncore
from collections import deque from collections import deque
def buffer(obj, start=None, stop=None):
# if memoryview objects gain slicing semantics,
# this function will change for the better
# memoryview used for the TypeError
memoryview(obj)
if start == None:
start = 0
if stop == None:
stop = len(obj)
x = obj[start:stop]
## print("buffer type is: %s"%(type(x),))
return x
class async_chat (asyncore.dispatcher): class async_chat (asyncore.dispatcher):
"""This is an abstract class. You must derive from this class, and add """This is an abstract class. You must derive from this class, and add
...@@ -240,7 +228,7 @@ class async_chat (asyncore.dispatcher): ...@@ -240,7 +228,7 @@ class async_chat (asyncore.dispatcher):
# handle classic producer behavior # handle classic producer behavior
obs = self.ac_out_buffer_size obs = self.ac_out_buffer_size
try: try:
data = buffer(first, 0, obs) data = first[:obs]
except TypeError: except TypeError:
data = first.more() data = first.more()
if data: if data:
......
...@@ -174,29 +174,31 @@ class BZ2File(io.BufferedIOBase): ...@@ -174,29 +174,31 @@ class BZ2File(io.BufferedIOBase):
# Fill the readahead buffer if it is empty. Returns False on EOF. # Fill the readahead buffer if it is empty. Returns False on EOF.
def _fill_buffer(self): def _fill_buffer(self):
if self._buffer: # Depending on the input data, our call to the decompressor may not
return True # return any data. In this case, try again after reading another block.
while True:
if self._decompressor.unused_data: if self._buffer:
rawblock = self._decompressor.unused_data return True
else:
rawblock = self._fp.read(_BUFFER_SIZE) if self._decompressor.unused_data:
rawblock = self._decompressor.unused_data
if not rawblock:
if self._decompressor.eof:
self._mode = _MODE_READ_EOF
self._size = self._pos
return False
else: else:
raise EOFError("Compressed file ended before the " rawblock = self._fp.read(_BUFFER_SIZE)
"end-of-stream marker was reached")
if not rawblock:
# Continue to next stream. if self._decompressor.eof:
if self._decompressor.eof: self._mode = _MODE_READ_EOF
self._decompressor = BZ2Decompressor() self._size = self._pos
return False
else:
raise EOFError("Compressed file ended before the "
"end-of-stream marker was reached")
# Continue to next stream.
if self._decompressor.eof:
self._decompressor = BZ2Decompressor()
self._buffer = self._decompressor.decompress(rawblock) self._buffer = self._decompressor.decompress(rawblock)
return True
# Read data until EOF. # Read data until EOF.
# If return_data is false, consume the data without returning it. # If return_data is false, consume the data without returning it.
...@@ -256,11 +258,14 @@ class BZ2File(io.BufferedIOBase): ...@@ -256,11 +258,14 @@ class BZ2File(io.BufferedIOBase):
return self._read_block(size) return self._read_block(size)
def read1(self, size=-1): def read1(self, size=-1):
"""Read up to size uncompressed bytes with at most one read """Read up to size uncompressed bytes, while trying to avoid
from the underlying stream. making multiple reads from the underlying stream.
Returns b'' if the file is at EOF. Returns b'' if the file is at EOF.
""" """
# Usually, read1() calls _fp.read() at most once. However, sometimes
# this does not give enough data for the decompressor to make progress.
# In this case we make multiple reads, to avoid returning b"".
with self._lock: with self._lock:
self._check_can_read() self._check_can_read()
if (size == 0 or self._mode == _MODE_READ_EOF or if (size == 0 or self._mode == _MODE_READ_EOF or
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment