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

Catch bad status code errors

This commit is contained in:
Philip Allgaier 2020-10-03 14:53:59 +02:00 committed by GitHub
parent c6472edcd4
commit 94e47e4c5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
speedtest.py
View File

@ -534,14 +534,19 @@ class SpeedtestHTTPHandler(AbstractHTTPHandler):
self.timeout = timeout
def http_open(self, req):
return self.do_open(
_build_connection(
SpeedtestHTTPConnection,
self.source_address,
self.timeout
),
req
)
try:
return self.do_open(
_build_connection(
SpeedtestHTTPConnection,
self.source_address,
self.timeout
),
req
)
except BadStatusLine:
raise SpeedtestHTTPError(
'Server returned bad status code'
)
http_request = AbstractHTTPHandler.do_request_
@ -558,15 +563,20 @@ class SpeedtestHTTPSHandler(AbstractHTTPHandler):
self.timeout = timeout
def https_open(self, req):
return self.do_open(
_build_connection(
SpeedtestHTTPSConnection,
self.source_address,
self.timeout,
context=self._context,
),
req
)
try:
return self.do_open(
_build_connection(
SpeedtestHTTPSConnection,
self.source_address,
self.timeout,
context=self._context,
),
req
)
except BadStatusLine:
raise SpeedtestHTTPError(
'Server returned bad status code'
)
https_request = AbstractHTTPHandler.do_request_