reonokiy/swh-lister
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

lister_transports: filter known http response codes

Browse source
This commit is contained in:
Nicolas Dandrimont 2017-09-05 15:21:17 +02:00
parent de01552ac5
commit 5eca777c75
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
swh/lister/core/lister_transports.py
View file

@ -29,6 +29,8 @@ class SWHListerHttpTransport(abc.ABC):
'To be implemented in the API-specific'
' class inheriting this.')
EXPECTED_STATUS_CODES = (200, 429, 403, 404)
def request_headers(self):
"""Returns dictionary of any request headers needed by the server.
@ -75,9 +77,13 @@ class SWHListerHttpTransport(abc.ABC):
if auth:
params['auth'] = (auth['username'], auth['password'])
try:
return self.session.get(self.api_baseurl + path, **params)
response = self.session.get(self.api_baseurl + path, **params)
except requests.exceptions.ConnectionError as e:
raise FetchError(e)
else:
if response.status_code not in self.EXPECTED_STATUS_CODES:
raise FetchError(response)
return response
def transport_response_to_string(self, response):
"""Implements SWHListerBase.transport_response_to_string for HTTP given