Add validator for image integrity

Author: adamtheturtle

src/mock_vws/_services_validators/__init__.py

@@ -29,6 +29,7 @@
     validate_image_data_type,
     validate_image_encoding,
     validate_image_format,
+    validate_image_integrity,
     validate_image_is_image,
     validate_image_size,
 )
@@ -122,6 +123,7 @@ def run_services_validators(
     validate_image_format(request_body=request_body)
     validate_image_color_space(request_body=request_body)
     validate_image_size(request_body=request_body)
+    validate_image_integrity(request_body=request_body)
 
     validate_name_type(request_body=request_body)
     validate_name_length(request_body=request_body)

src/mock_vws/_services_validators/image_validators.py

@@ -21,6 +21,33 @@
 _LOGGER = logging.getLogger(name=__name__)
 
 
+@beartype
+def validate_image_integrity(*, request_body: bytes) -> None:
+    """Validate the integrity of the image given to a VWS endpoint.
+
+    Args:
+        request_body: The body of the request.
+
+    Raises:
+        BadImageError: The image is given and is not a valid image file.
+    """
+    if not request_body:
+        return
+
+    request_text = request_body.decode()
+    image = json.loads(s=request_text).get("image")
+    decoded = decode_base64(encoded_data=image)
+
+    image_file = io.BytesIO(initial_bytes=decoded)
+    pil_image = Image.open(fp=image_file)
+
+    try:
+        pil_image.verify()
+    except SyntaxError as exc:
+        _LOGGER.warning(msg="The image is not a valid image file.")
+        raise BadImageError from exc
+
+
 @beartype
 def validate_image_format(*, request_body: bytes) -> None:
     """Validate the format of the image given to a VWS endpoint.