Переглянути джерело

fix(file_factory): validate upload_file_id format as UUID (#12084)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 3 місяців тому
батько
коміт
754791efd3
1 змінених файлів з 6 додано та 0 видалено
  1. 6 0
      api/factories/file_factory.py

+ 6 - 0
api/factories/file_factory.py

@@ -1,4 +1,5 @@
 import mimetypes
+import uuid
 from collections.abc import Callable, Mapping, Sequence
 from typing import Any, cast
 
@@ -119,6 +120,11 @@ def _build_from_local_file(
     upload_file_id = mapping.get("upload_file_id")
     if not upload_file_id:
         raise ValueError("Invalid upload file id")
+    # check if upload_file_id is a valid uuid
+    try:
+        uuid.UUID(upload_file_id)
+    except ValueError:
+        raise ValueError("Invalid upload file id format")
     stmt = select(UploadFile).where(
         UploadFile.id == upload_file_id,
         UploadFile.tenant_id == tenant_id,