Pārlūkot izejas kodu

fix(file_factory): validate upload_file_id before querying UploadFile (#11937)

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 mēneši atpakaļ
vecāks
revīzija
8f73670925
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 4 1
      api/factories/file_factory.py

+ 4 - 1
api/factories/file_factory.py

@@ -116,8 +116,11 @@ def _build_from_local_file(
     tenant_id: str,
     transfer_method: FileTransferMethod,
 ) -> File:
+    upload_file_id = mapping.get("upload_file_id")
+    if not upload_file_id:
+        raise ValueError("Invalid upload file id")
     stmt = select(UploadFile).where(
-        UploadFile.id == mapping.get("upload_file_id"),
+        UploadFile.id == upload_file_id,
         UploadFile.tenant_id == tenant_id,
     )