Adding custom ParameterValidator to Connexion setup

CLOUDBLD-12516

Signed-off-by: Felipe de Almeida <[email protected]>

cachito/web/app.py

@@ -26,7 +26,7 @@
 from cachito.web.docs import docs
 from cachito.web.errors import json_error, validation_error
 from cachito.web.metrics import init_metrics
-from cachito.web.validation import RequestBodyValidator
+from cachito.web.validation import ParameterValidator, RequestBodyValidator
 
 
 def healthcheck():
@@ -118,7 +118,7 @@ def create_app(config_obj=None):
     connexion_app.add_api(
         f"{path}/static/api_v1.yaml",
         strict_validation=True,
-        validator_map={"body": RequestBodyValidator},
+        validator_map={"body": RequestBodyValidator, "parameter": ParameterValidator},
     )
 
     app.add_url_rule("/healthcheck", view_func=healthcheck)

cachito/web/validation.py

@@ -4,6 +4,7 @@
 
 import jsonschema
 from connexion import decorators
+from connexion.exceptions import BadRequestProblem, ExtraParameterProblem
 
 from cachito.errors import ValidationError
 
@@ -70,3 +71,26 @@ def validate_schema(self, data, url):
             raise ValidationError(exception.message)
 
         return None
+
+
+class ParameterValidator(decorators.validation.ParameterValidator):
+    """
+    Changes the default Connexion exception to Cachito's ValidationError.
+
+    For more information about custom validation error handling:
+        - https://github.com/zalando/connexion/issues/558
+        - https://connexion.readthedocs.io/en/latest/request.html
+    """
+
+    def __call__(self, function):
+        """Throw cachito.ValidationError."""
+        wrapper = super().__call__(function)
+
+        def handle_wrapper(request):
+            """Handle original wrapper."""
+            try:
+                return wrapper(request)
+            except (BadRequestProblem, ExtraParameterProblem) as exception:
+                raise ValidationError(exception.detail)
+
+        return handle_wrapper