Update streaming.py's handle_stream
This is an attempt to resolve an upstream issue with unexpected blank streams. The issue appears to be in urllib3, and causes fatal errors when the stream returned is `b''`. I believe this simple workaround will bypass the error and continue monitoring the stream.
This commit is contained in:
pare
0cd284bc2f
commit
793f0b207f
S'han modificat 1 arxius amb 23 adicions i 20 eliminacions
|
@ -41,26 +41,29 @@ class StreamListener(object):
|
|||
|
||||
response; a requests response object with the open stream for reading.
|
||||
"""
|
||||
event = {}
|
||||
line_buffer = bytearray()
|
||||
for chunk in response.iter_content(chunk_size = 1):
|
||||
if chunk:
|
||||
if chunk == b'\n':
|
||||
try:
|
||||
line = line_buffer.decode('utf-8')
|
||||
except UnicodeDecodeError as err:
|
||||
six.raise_from(
|
||||
MastodonMalformedEventError("Malformed UTF-8"),
|
||||
err
|
||||
)
|
||||
if line == '':
|
||||
self._dispatch(event)
|
||||
event = {}
|
||||
try:
|
||||
event = {}
|
||||
line_buffer = bytearray()
|
||||
for chunk in response.iter_content(chunk_size = 1):
|
||||
if chunk:
|
||||
if chunk == b'\n':
|
||||
try:
|
||||
line = line_buffer.decode('utf-8')
|
||||
except UnicodeDecodeError as err:
|
||||
six.raise_from(
|
||||
MastodonMalformedEventError("Malformed UTF-8"),
|
||||
err
|
||||
)
|
||||
if line == '':
|
||||
self._dispatch(event)
|
||||
event = {}
|
||||
else:
|
||||
event = self._parse_line(line, event)
|
||||
line_buffer = bytearray()
|
||||
else:
|
||||
event = self._parse_line(line, event)
|
||||
line_buffer = bytearray()
|
||||
else:
|
||||
line_buffer.extend(chunk)
|
||||
line_buffer.extend(chunk)
|
||||
except:
|
||||
pass
|
||||
|
||||
def _parse_line(self, line, event):
|
||||
if line.startswith(':'):
|
||||
|
|
Loading…
Referencia en una nova incidència