add OperationNotAllowedException to catch limits

Author: Matt Sokoloff

labelbox/client.py

@@ -201,6 +201,7 @@ def check_errors(keywords, *path):
             return None
 
         def get_error_status_code(error):
+            print(error)
             return error["extensions"]["exception"]["status"]
 
         if check_errors(["AUTHENTICATION_ERROR"], "extensions",
@@ -255,6 +256,12 @@ def get_error_status_code(error):
             else:
                 raise labelbox.exceptions.InternalServerError(message)
 
+        not_allowed_error = check_errors(["OPERATION_NOT_ALLOWED"],
+                                         "extensions", "code")
+        if not_allowed_error is not None:
+            message = not_allowed_error.get("message")
+            raise labelbox.exceptions.OperationNotAllowedException(message)
+
         if len(errors) > 0:
             logger.warning("Unparsed errors on query execution: %r", errors)
             raise labelbox.exceptions.LabelboxError("Unknown error: %s" %

labelbox/exceptions.py

@@ -112,4 +112,9 @@ class InconsistentOntologyException(Exception):
 
 class MALValidationError(LabelboxError):
     """Raised when user input is invalid for MAL imports."""
-    ...
+    pass
+
+
+class OperationNotAllowedException(Exception):
+    """Raised when user does not have permissions to a resource or has exceeded usage limit"""
+    pass