mirror/speedtest-cli
2
0
Fork 0
mirror of https://github.com/sivel/speedtest-cli.git synced 2024-11-21 01:13:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Catch OSError and EOFError while reading from gzip stream

Browse Source
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