Handling warnings from server

Author: paulstretenowich

pt_cli/connect.py

@@ -16,12 +16,18 @@
 class Error(Exception):
     """docstring for Error"""
     def __init__(self, msg):
-        self.args = (f"{type(self).__name__}: {msg}",)
+        if isinstance(msg, list):
+            self.args = (f"{type(self).__name__}: \n{chr(10).join(msg)}",)
+        else:
+            self.args = (f"{type(self).__name__}: {msg}",)
         sys.exit(self)
 
 class BadRequestError(Error):
     """docstring for BadRequestError"""
 
+class BadRequestWarning(Error):
+    """docstring for BadRequestError"""
+
 class OAuthNego():
     """
     The base class to create python client that is able to
@@ -87,9 +93,11 @@ def connect(self, r_get):
     def maybe_json(self, data):
         try:
             loads = json.loads(data)
-            # logger.info(f"\n\n{type(data)}\n{type(loads)}\n\n")
+            # logger.info(f"\n\n{loads}\n\n")
             if isinstance(loads, dict) and loads.get("DB_ACTION_ERROR"):
                 raise BadRequestError(loads.get("DB_ACTION_ERROR"))
+            if isinstance(loads, dict) and loads.get("DB_ACTION_WARNING"):
+                raise BadRequestWarning(loads.get("DB_ACTION_WARNING"))
             self.data_type = 'json'
             return loads
         except json.decoder.JSONDecodeError: