Add error reporting callbacks to client

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -9,7 +9,7 @@
 import urllib.parse
 from collections import defaultdict
 from datetime import datetime, timezone
-from typing import Any, List, Dict, Union, Optional, overload
+from typing import Any, List, Dict, Union, Optional, overload, Callable
 
 import requests
 import requests.exceptions
@@ -138,15 +138,19 @@ def _default_headers(self):
     @retry.Retry(predicate=retry.if_exception_type(
         labelbox.exceptions.InternalServerError,
         labelbox.exceptions.TimeoutError))
-    def execute(self,
-                query=None,
-                params=None,
-                data=None,
-                files=None,
-                timeout=60.0,
-                experimental=False,
-                error_log_key="message",
-                raise_return_resource_not_found=False):
+    def execute(
+        self,
+        query=None,
+        params=None,
+        data=None,
+        files=None,
+        timeout=60.0,
+        experimental=False,
+        error_log_key="message",
+        raise_return_resource_not_found=False,
+        error_handlers: Optional[Dict[str, Callable[[Dict[str, Any]],
+                                                    None]]] = None
+    ) -> Dict[str, Any]:
         """ Sends a request to the server for the execution of the
         given query.
 
@@ -323,7 +327,12 @@ def get_error_status_code(error: dict) -> int:
         # TODO: fix this in the server API
         internal_server_error = check_errors(["INTERNAL_SERVER_ERROR"],
                                              "extensions", "code")
+        error_code = "INTERNAL_SERVER_ERROR"
+
         if internal_server_error is not None:
+            if error_handlers and error_code in error_handlers:
+                handler = error_handlers[error_code]
+                handler(response)
             message = internal_server_error.get("message")
             error_status_code = get_error_status_code(internal_server_error)
             if error_status_code == 400: