Refactors Project for More Declarative Package Name

storage_service/controller/__init__.py

@@ -0,0 +1,19 @@
+from storage_service.config.config_allowed_origins import get_allowed_origins
+from storage_service.controller.storage_controller import s3_router
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+
+app = FastAPI()
+
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=get_allowed_origins(),
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+app.include_router(s3_router)

storage_service/controller/storage_controller.py

@@ -0,0 +1,46 @@
+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.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_utils.cbv import cbv
+from fastapi_utils.inferring_router import InferringRouter
+from rq import Queue
+
+from typing import Annotated
+
+s3_router = InferringRouter()
+
+
+@cbv(s3_router)
+class StorageController:
+    queue: Queue = Depends(dependency_queue, use_cache=True)
+    storage_service: StorageService = Depends(dependency_storage_service, use_cache=True)
+
+    @s3_router.post("/new_file_url/", status_code=200)
+    def new_file_url(
+            self,
+            username: Annotated[str, Body(embed=True)],
+            file_postfix: Annotated[str, Body(embed=True)],
+            file_type: Annotated[FileType, Body(embed=True)],
+    ) -> dict[str, str]:
+        return self.storage_service.get_temp_upload_link(
+            file_name_hash(username, file_postfix), file_type
+        )
+
+    @s3_router.get("/file_url/", status_code=200)
+    def file_url(self, username: str, file_postfix: str) -> dict[str, str]:
+        return self.storage_service.get_temp_read_link(
+            file_name_hash(username, file_postfix)
+        )
+
+    @s3_router.post("/process_file/", 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)