Просмотр исходного кода

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.15.0
-LAN- 10 месяцев назад
Родитель
Сommit
754791efd3
Аккаунт пользователя с таким Email не найден
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,

Загрузка…
Отмена
Сохранить