Add Pydantic to IIB web

Author: xDaile

iib/web/api_v1.py

@@ -37,6 +37,7 @@
     User,
 )
 from iib.web.s3_utils import get_object_from_s3_bucket
+from iib.web.pydantic_models import AddPydanticModel, RmPydanticModel
 from botocore.response import StreamingBody
 from iib.web.utils import pagination_metadata, str_to_bool
 from iib.workers.tasks.build import (
@@ -576,10 +577,16 @@ def add_bundles() -> Tuple[flask.Response, int]:
     :rtype: flask.Response
     :raise ValidationError: if required parameters are not supplied
     """
-    payload: AddRequestPayload = cast(AddRequestPayload, flask.request.get_json())
-    if not isinstance(payload, dict):
-        raise ValidationError('The input data must be a JSON object')
 
+    try:
+        add_request_payload = AddRequestPydanticModel.model_validate(
+            flask.request.data, strict=True,
+        )
+    except ValidationError as e:
+        # If the JSON data doesn't match the Pydantic model, return a 400 Bad Request response
+        return flask.jsonify({'Error parsing data': str(e)}), 400
+
+    payload = add_request_payload.model_dump_json()
     # Only run `_get_unique_bundles` if it is a list. If it's not, `from_json`
     # will raise an error to the user.
     if payload.get('bundles') and isinstance(payload['bundles'], list):
@@ -807,7 +814,15 @@ def rm_operators() -> Tuple[flask.Response, int]:
     :rtype: flask.Response
     :raise ValidationError: if required parameters are not supplied
     """
-    payload: RmRequestPayload = cast(RmRequestPayload, flask.request.get_json())
+    try:
+        rm_request_payload = RmRequestPydanticModel.model_validate(
+            flask.request.data, strict=True,
+        )
+    except ValidationError as e:
+        # If the JSON data doesn't match the Pydantic model, return a 400 Bad Request response
+        return flask.jsonify({'Error parsing data': str(e)}), 400
+
+    payload: RmRequestPayload = cast(RmRequestPayload, rm_request_payload.model_dump_json())
     if not isinstance(payload, dict):
         raise ValidationError('The input data must be a JSON object')