Merge pull request #11 from krasoffski/errors

Improve error message for part 1 issue #10

Author: Alexander.Iljushkin

reportportal_client/errors.py

@@ -0,0 +1,16 @@
+class Error(Exception):
+    """General exception for package."""
+
+
+class EntryCreatedError(Error):
+    """Represents error in case no entry is created.
+
+    No 'id' in the json response.
+    """
+
+
+class OperationCompletionError(Error):
+    """Represents error in case of operation failure.
+
+    No 'msg' in the json response.
+    """

reportportal_client/model/response.py

@@ -1,5 +1,7 @@
 import json
 
+from ..errors import EntryCreatedError, OperationCompletionError
+
 
 class RS(object):
     def __init__(self, raw):
@@ -15,9 +17,8 @@ def __init__(self, raw):
     def id(self):
         try:
             return json.loads(self.raw)["id"]
-        except KeyError as error:
-            error.message += "Raw: {0}".format(self.raw)
-            raise
+        except KeyError:
+            raise EntryCreatedError("raw: {0}".format(self.raw))
 
     def as_dict(self):
         return {"id": self.id}
@@ -31,9 +32,8 @@ def __init__(self, raw):
     def msg(self):
         try:
             return json.loads(self.raw)["msg"]
-        except KeyError as error:
-            error.message += "Raw: {0}".format(self.raw)
-            raise
+        except KeyError:
+            raise OperationCompletionError("raw: {0}".format(self.raw))
 
     def as_dict(self):
         return {"msg": self.msg}