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

Reformats Endpoints Configuration and Start Script

Browse Source
This commit is contained in:
Vitor Hideyoshi
2024-05-21 20:36:16 -03:00
parent 000cb08e37
commit c59dc14810
10 changed files with 92 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
storage_service/controller/storage_controller.py
View File

@@ -20,17 +20,17 @@ from fastapi import APIRouter, Depends, HTTPException
from fastapi_utils.cbv import cbv
from rq import Queue
s3_router = APIRouter(tags=["storage"])
router = APIRouter(tags=["storage"])
@cbv(s3_router)
@cbv(router)
class StorageController:
queue: Queue = Depends(dependency_queue, use_cache=True)
storage_service: StorageService = Depends(
dependency_storage_service, use_cache=True
)
@s3_router.post("/file", status_code=200)
@router.post("/file", status_code=200)
def new_file_url(self, new_file_request: NewFileURLRequest) -> SignedUrlResponse:
hashed_file_name = generate_file_hash(
new_file_request.file_key, new_file_request.file_postfix
@@ -40,7 +40,7 @@ class StorageController:
hashed_file_name, new_file_request.file_type
)
@s3_router.get("/file", status_code=200)
@router.get("/file", status_code=200)
def file_url(self, file_key: str, file_postfix: str) -> SignedUrlResponse:
try:
return self.storage_service.get_temp_read_link(
@@ -49,13 +49,13 @@ class StorageController:
except Exception as _:
raise FileNotFoundException("File not found")
@s3_router.delete("/file", status_code=204)
@router.delete("/file", status_code=204)
def delete_file(self, file_key: str, file_postfix: str):
return self.storage_service.delete_file(
generate_file_hash(file_key, file_postfix)
)
@s3_router.post("/file/process", status_code=200)
@router.post("/file/process", status_code=200)
def process_file(self, process_file_request: ProcessFileRequest):
self.queue.enqueue(
storage_file_worker,