dbaxa
diff --git a/‎cmdexec/clients/python/src/databricks/sql/exc.py‎
Lines changed: 49 additions & 2 deletions b/‎cmdexec/clients/python/src/databricks/sql/exc.py‎
Lines changed: 49 additions & 2 deletions
@@ -1,6 +1,26 @@
+import json
+import logging
+
+logger = logging.getLogger(__name__)
+
+
 ### PEP-249 Mandated ###
 class Error(Exception):
-    pass
+    """Base class for DB-API2.0 exceptions.
+    `message`: An optional user-friendly error message. It should be short, actionable and stable
+    `context`: Optional extra context about the error. MUST be JSON serializable
+    """
+
+    def __init__(self, message=None, context=None, *args, **kwargs):
+        super().__init__(message, *args, **kwargs)
+        self.message = message
+        self.context = context or {}
+
+    def __str__(self):
+        return self.message
+
+    def message_with_context(self):
+        return self.message + ": " + json.dumps(self.context, default=str)
 
 
 class Warning(Exception):
@@ -41,5 +61,32 @@ class NotSupportedError(DatabaseError):
 
 ### Custom error classes ###
 class InvalidServerResponseError(OperationalError):
-    """ Thrown if the server does not set the initial namespace correctly"""
+    """Thrown if the server does not set the initial namespace correctly"""
+    pass
+
+
+class ServerOperationError(DatabaseError):
+    """Thrown if the operation moved to an error state, if for example there was a syntax
+    error.
+    Its context will have the following keys:
+    "diagnostic-info": The full Spark stack trace (if available)
+    "operation-id": The Thrift ID of the operation
+    """
+    pass
+
+
+class RequestError(OperationalError):
+    """Thrown if there was a error during request to the server.
+    Its context will have the following keys:
+    "method": The RPC method name that failed
+    "session-id": The Thrift session guid
+    "query-id": The Thrift query guid (if available)
+    "http-code": HTTP response code to RPC request (if available)
+    "error-message": Error message from the HTTP headers (if available)
+    "original-exception": The Python level original exception
+    "no-retry-reason": Why the request wasn't retried (if available)
+    "bounded-retry-delay": The maximum amount of time an error will be retried before giving up
+    "attempt": current retry number / maximum number of retries
+    "elapsed-seconds": time that has elapsed since first attempting the RPC request
+    """
     pass