Improved exception-classes.

mjmaenpaa · mjmaenpaa · commit 8351b2d5fcb6 · 2014-11-01T15:56:43.000+02:00
- Added http status-code and gitlab error message to GitlabError-class.
- Added new GitlabOperationError-class to help separate connection and
authentication errors from other gitlab errors.
diff --git a/gitlab.py b/gitlab.py
@@ -42,39 +42,62 @@ def default(self, obj):
         return json.JSONEncoder.default(self, obj)
 
 
-class GitlabConnectionError(Exception):
+class GitlabError(Exception):
+    def __init__(self, error_message="", response_code=None,
+                 response_body=None):
+
+        Exception.__init__(self, error_message)
+        # Http status code
+        self.response_code = response_code
+        # Full http response
+        self.response_body = response_body
+        # Parsed error message from gitlab
+        self.error_message = error_message
+
+    def __str__(self):
+        if self.response_code is not None:
+            return "{0}: {1}".format(self.response_code, self.error_message)
+        else:
+            return "{0}".format(self.error_message)
+
+
+class GitlabAuthenticationError(GitlabError):
+    pass
+
+
+class GitlabConnectionError(GitlabError):
     pass
 
 
-class GitlabListError(Exception):
+class GitlabOperationError(GitlabError):
     pass
 
 
-class GitlabGetError(Exception):
+class GitlabListError(GitlabOperationError):
     pass
 
 
-class GitlabCreateError(Exception):
+class GitlabGetError(GitlabOperationError):
     pass
 
 
-class GitlabUpdateError(Exception):
+class GitlabCreateError(GitlabOperationError):
     pass
 
 
-class GitlabDeleteError(Exception):
+class GitlabUpdateError(GitlabOperationError):
     pass
 
 
-class GitlabProtectError(Exception):
+class GitlabDeleteError(GitlabOperationError):
     pass
 
 
-class GitlabTransferProjectError(Exception):
+class GitlabProtectError(GitlabOperationError):
     pass
 
 
-class GitlabAuthenticationError(Exception):
+class GitlabTransferProjectError(GitlabOperationError):
     pass
 
 
