old-portfolio/storage-hideyoshi.com
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adds New Delete File Endpoint and Refactors Endpoints, Adds VirusChecker

Browse Source
This commit is contained in:
Vitor Hideyoshi
2023-09-10 02:21:41 -03:00
parent ab65be9710
commit c826000954
13 changed files with 149 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
storage_service/controller/storage_controller.py
View File

@@ -4,12 +4,12 @@ from storage_service.depends.depend_queue import dependency_queue
from storage_service.depends.depend_s3_service import (
dependency_storage_service,
)
from storage_service.service.storage_service import StorageService
from storage_service.service.storage.storage_service import StorageService
from storage_service.utils.enums.file_type import FileType
from storage_service.utils.file_name_hash import file_name_hash
from storage_service.worker.storage_file_worker import storage_file_worker
from fastapi import Body, Depends, Form
from fastapi import Body, Depends
from fastapi_utils.cbv import cbv
from fastapi_utils.inferring_router import InferringRouter
from rq import Queue
@@ -26,7 +26,7 @@ class StorageController:
dependency_storage_service, use_cache=True
)
@s3_router.post("/new_file_url/", status_code=200)
@s3_router.post("/file/", status_code=200)
def new_file_url(
self,
username: Annotated[str, Body(embed=True)],
@@ -37,16 +37,24 @@ class StorageController:
file_name_hash(username, file_postfix), file_type
)
@s3_router.get("/file_url/", status_code=200)
@s3_router.get("/file/", status_code=200)
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)
)
@s3_router.post("/process_file/", status_code=200)
@s3_router.delete("/file/", status_code=204)
def delete_file(self, username: str, file_postfix: str):
return self.storage_service.delete_file(
file_name_hash(username, file_postfix)
)
@s3_router.post("/file/process", status_code=200)
def process_file(
self,
username: Annotated[str, Body(embed=True)],
file_postfix: Annotated[str, Body(embed=True)],
):
self.queue.enqueue(storage_file_worker, username, file_postfix)