|
| 1 | +import logging |
| 2 | + |
| 3 | +from fastapi import Request, Depends, File, UploadFile |
| 4 | +from fastapi.concurrency import run_in_threadpool |
| 5 | +from fastapi_utils.cbv import cbv |
| 6 | +from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials |
| 7 | +from fastapi_utils.inferring_router import InferringRouter |
| 8 | +from spaceone.core.fastapi.api import BaseAPI, exception_handler |
| 9 | +from spaceone.file_manager.manager.file_connector_manager import FileConnectorManager |
| 10 | + |
| 11 | +from spaceone.file_manager.service.file_service import FileService |
| 12 | + |
| 13 | +_LOGGER = logging.getLogger(__name__) |
| 14 | +_AUTH_SCHEME = HTTPBearer(auto_error=False) |
| 15 | + |
| 16 | +router = InferringRouter(include_in_schema=False) |
| 17 | + |
| 18 | + |
| 19 | +@cbv(router) |
| 20 | +class Files(BaseAPI): |
| 21 | + token: HTTPAuthorizationCredentials = Depends(_AUTH_SCHEME) |
| 22 | + service = "file-manager" |
| 23 | + |
| 24 | + @router.post("/public/upload") |
| 25 | + @exception_handler |
| 26 | + async def upload_public_file(self, request: Request, file: UploadFile = File(...)): |
| 27 | + metadata = { |
| 28 | + "token": self.token.credentials, |
| 29 | + } |
| 30 | + |
| 31 | + params = { |
| 32 | + "name": file.filename, |
| 33 | + "resource_group": "SYSTEM", |
| 34 | + } |
| 35 | + |
| 36 | + file_svc = FileService(metadata) |
| 37 | + response: dict = await run_in_threadpool(file_svc.add, params) |
| 38 | + |
| 39 | + file_id = response.get("file_id") |
| 40 | + |
| 41 | + file_conn_mgr = FileConnectorManager() |
| 42 | + # Update File |
| 43 | + # file_id = file_vo.file_id |
| 44 | + |
| 45 | + return response |
| 46 | + |
| 47 | + @router.get("/public/{file_id}") |
| 48 | + @exception_handler |
| 49 | + async def download_public_file(self, request: Request, file_id: str): |
| 50 | + metadata = { |
| 51 | + "token": self.token.credentials, |
| 52 | + } |
| 53 | + |
| 54 | + params = { |
| 55 | + "file_id": file_id, |
| 56 | + } |
| 57 | + |
| 58 | + file_svc = FileService(metadata) |
| 59 | + response: dict = await run_in_threadpool(file_svc.get, params) |
| 60 | + |
| 61 | + file_id = response.get("file_id") |
| 62 | + |
| 63 | + file_conn_mgr = FileConnectorManager() |
| 64 | + # Update File |
| 65 | + # file_id = file_vo.file_id |
| 66 | + |
| 67 | + return response |
0 commit comments