Small changes to exception handling introducing additional exception classes.

Author: Emanuel Winblad

vilfo/client.py

@@ -46,10 +46,11 @@ def _request(self, method, endpoint, headers=None, data=None, params=None, timeo
         try:
             response = getattr(requests, method)(url, headers=headers, data=data, params=params, timeout=timeout)
         except requests.exceptions.RequestException as ex:
-            raise ex
+            # Wrap the exception in our own exception class.
+            raise vilfo.exceptions.VilfoRequestException(ex)
 
         if 404 == response.status_code:
-            raise vilfo.exceptions.VilfoException()
+            raise vilfo.exceptions.NotFoundException()
 
         if 403 == response.status_code or response_content_is_login_page(response.content):
             raise vilfo.exceptions.AuthenticationException()

vilfo/exceptions.py

@@ -1,6 +1,14 @@
 """The exceptions used by the Vilfo client."""
+from requests.exceptions import RequestException
+
 class VilfoException(Exception):
     """General Vilfo exception occurred."""
 
 class AuthenticationException(VilfoException):
     """An exception occurred related to authentication."""
+
+class NotFoundException(VilfoException):
+    """An exception indicating an API resource was not found."""
+
+class VilfoRequestException(VilfoException, RequestException):
+    """An exception caught by Vilfo but raised by the requests library."""