Returns NULL If File Doesn't Exist

2023-08-24 01:26:29 -03:00
parent 4d0ad4b104
commit 1a3b4cfaa6
3 changed files with 15 additions and 9 deletions
--- a/storage_service/controller/storage_controller.py
+++ b/storage_service/controller/storage_controller.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from storage_service.depends.depend_queue import dependency_queue
 from storage_service.depends.depend_s3_service import (
    dependency_storage_service,
@@ -36,7 +38,7 @@ class StorageController:
        )

    @s3_router.get("/file_url/", status_code=200)
-    def file_url(self, username: str, file_postfix: str) -> dict[str, str]:
+    def file_url(self, username: str, file_postfix: str) -> dict[str, str | None]:
        return self.storage_service.get_temp_read_link(
            file_name_hash(username, file_postfix)
        )