walkero
/
speedtest-cli
1
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

Catch OSError and EOFError while reading from gzip stream

This commit is contained in:
Matt Martz 2018-01-02 18:32:03 -06:00
parent 9848481d06
commit c16ffd4ae7
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
speedtest.py
View File

@ -1041,7 +1041,10 @@ class Speedtest(object):
stream = get_response_stream(uh)
while 1:
configxml.append(stream.read(1024))
try:
configxml.append(stream.read(1024))
except (OSError, EOFError):
raise ConfigRetrievalError(get_exception())
if len(configxml[-1]) == 0:
break
stream.close()
@ -1166,7 +1169,10 @@ class Speedtest(object):
serversxml = []
while 1:
serversxml.append(stream.read(1024))
try:
serversxml.append(stream.read(1024))
except (OSError, EOFError):
raise ServersRetrievalError(get_exception())
if len(serversxml[-1]) == 0:
break