Browse Source

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 tháng trước cách đây
mục cha
commit
8f73670925
1 tập tin đã thay đổi với 4 bổ sung1 xóa
  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,
     )