Handle entity-too-large HTTP errors (#81)

CERT-Polska · Aug 25, 2022 · 33ace40 · 33ace40
1 parent 7177597
commit 33ace40
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/mwdblib/__version__.py b/mwdblib/__version__.py
@@ -1 +1 @@
-__version__ = "4.2.1"
+__version__ = "4.3.0"
diff --git a/mwdblib/exc.py b/mwdblib/exc.py
@@ -186,6 +186,16 @@ class VersionMismatchError(MWDBError):
     pass
 
 
+class ObjectTooLargeError(MWDBError):
+    """
+    Server signaled that the uploaded object is too large.
+
+    .. versionadded:: 4.3.0
+    """
+
+    pass
+
+
 def get_http_error_message(http_error: requests.exceptions.HTTPError) -> str:
     import json
 
@@ -230,6 +240,8 @@ def map_http_error(http_error: requests.exceptions.HTTPError) -> MWDBError:
         requests.codes.gateway_timeout,
     ]:
         return GatewayError(http_error=http_error)
+    elif http_error.response.status_code == requests.codes.request_entity_too_large:
+        return ObjectTooLargeError(http_error=http_error)
     else:
         # Other codes are classified as internal error
         return InternalError(http_error=http_error)