From c6d994731fa01f7278aa89308e11f6f892c4d668 Mon Sep 17 00:00:00 2001 From: "Ivan Mladjenovic (He/Him)" Date: Thu, 28 Nov 2024 14:30:28 +0000 Subject: [PATCH] fix typing issue --- backend/src/session/file_uploads.py | 5 ++++- backend/src/utils/file_utils.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/src/session/file_uploads.py b/backend/src/session/file_uploads.py index 134cf24a..4419441b 100644 --- a/backend/src/session/file_uploads.py +++ b/backend/src/session/file_uploads.py @@ -17,21 +17,24 @@ UPLOADS_KEY_PREFIX = "file_upload_" + class FileUploadMeta(TypedDict): uploadId: str filename: str + class FileUpload(TypedDict): uploadId: str + content: str filename: str | None contentType: str | None size: int | None - content: str | None def get_session_file_uploads_meta() -> list[FileUploadMeta] | None: return get_session(UPLOADS_META_SESSION_KEY, []) + def get_session_file_upload(upload_id) -> FileUpload | None: value = redis_client.get(UPLOADS_KEY_PREFIX + upload_id) if value and isinstance(value, str): diff --git a/backend/src/utils/file_utils.py b/backend/src/utils/file_utils.py index 66e499af..e4fcaf2d 100644 --- a/backend/src/utils/file_utils.py +++ b/backend/src/utils/file_utils.py @@ -12,7 +12,7 @@ MAX_FILE_SIZE = 10*1024*1024 -def handle_file_upload(file:UploadFile) -> FileUpload: +def handle_file_upload(file: UploadFile) -> FileUpload: if (file.size or 0) > MAX_FILE_SIZE: raise HTTPException(status_code=413, detail=f"File upload must be less than {MAX_FILE_SIZE} bytes")