browser2: More specialized exceptions

and extend common exceptions
This commit is contained in:
Laurent Bachelier 2014-06-02 17:13:24 +02:00
commit c69c5cf5ef
2 changed files with 55 additions and 1 deletions

View file

@ -39,6 +39,7 @@ except ImportError:
from weboob.tools.log import getLogger
from .cookies import WeboobCookieJar
from .exceptions import HTTPNotFound, ClientError, ServerError
class Profile(object):
@ -318,9 +319,29 @@ class BaseBrowser(object):
if allow_redirects:
response = self.handle_refresh(response)
self.raise_for_status(response)
return response
def raise_for_status(self, response):
"""
Like Response.raise_for_status but will use other classes if needed.
"""
http_error_msg = None
if 400 <= response.status_code < 500:
http_error_msg = '%s Client Error: %s' % (response.status_code, response.reason)
cls = ClientError
if response.status_code == 404:
cls = HTTPNotFound
elif 500 <= response.status_code < 600:
http_error_msg = '%s Server Error: %s' % (response.status_code, response.reason)
cls = ServerError
if http_error_msg:
raise cls(http_error_msg, response=response)
# in case we did not catch something that should be
response.raise_for_status()
return response
def build_request(self, url, referrer=None, data_encoding=None, **kwargs):
"""