From bb898f11dae5cc5ca3ec037cc3e441e9aab0fa87 Mon Sep 17 00:00:00 2001 From: UTSAV OJHA Date: Fri, 24 Oct 2025 17:00:00 +0530 Subject: [PATCH] Update validation.py Added MIME-type validation --- optifit backend/validation.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/optifit backend/validation.py b/optifit backend/validation.py index 1740dc9..1d3cf6e 100644 --- a/optifit backend/validation.py +++ b/optifit backend/validation.py @@ -1,6 +1,6 @@ import os import uuid - +import mimetypes class APIError(Exception): def __init__(self, message, status_code=500): self.message = message @@ -88,6 +88,14 @@ def validate_video_file(file): f"Unsupported file format '{file_ext}'. " f"Allowed formats: {', '.join(ALLOWED_EXTENSIONS)}" ) + #mime type check + mime_type, _ = mimetypes.guess_type(file.filename) + if not mime_type or not mime_type.startswith('video/'): + raise UnsupportedMediaTypeError(f"Invalid MIME type: {mime_type}") + + #reset point for later processing + file.seek(0) + return True # Validating the job request for getting the results