49 lines
2.4 KiB
Python
49 lines
2.4 KiB
Python
|
from pathlib import Path
|
||
|
|
from typing import Annotated, Literal
|
||
|
|
|
||
|
|
from fastapi import APIRouter, BackgroundTasks, Body, Depends, File, Form, Query, Request, Security, UploadFile
|
||
|
|
from fastapi.responses import FileResponse, JSONResponse
|
||
|
|
|
||
|
|
from app.common.response import ResponseSchema, SuccessResponse, UploadFileResponse
|
||
|
|
from app.core.base_schema import AuthSchema, UploadResponseSchema
|
||
|
|
from app.core.dependencies import AuthPermission, get_current_user
|
||
|
|
from app.core.router_class import OperationLogRoute
|
||
|
|
from app.utils.upload_util import UploadUtil
|
||
|
|
|
||
|
|
from .service import FileService
|
||
|
|
|
||
|
|
FileRouter = APIRouter(route_class=OperationLogRoute, prefix="/file", tags=["文件管理"])
|
||
|
|
|
||
|
|
|
||
|
|
@FileRouter.post("/upload", summary="上传文件", response_model=ResponseSchema[UploadResponseSchema], dependencies=[Security(AuthPermission(["module_common:file:upload"]))])
|
||
|
|
async def upload_controller(
|
||
|
|
request: Request,
|
||
|
|
auth: Annotated[AuthSchema, Depends(get_current_user)],
|
||
|
|
file: Annotated[UploadFile, File(description="上传文件")],
|
||
|
|
upload_type: Annotated[
|
||
|
|
Literal["file", "avatar", "param", "resource"] | None,
|
||
|
|
Query(description="上传类型: file=通用文件, avatar=头像, param=参数配置, resource=监控资源"),
|
||
|
|
] = "file",
|
||
|
|
target_path: Annotated[str | None, Form(description="目标目录路径（仅 resource 类型支持）")] = None,
|
||
|
|
) -> JSONResponse:
|
||
|
|
result = await FileService.upload_service(
|
||
|
|
base_url="/", # 返回 /static 开头的相对路径，去掉域名端口，便于环境迁移（dev 由 vite proxy /static 转发到后端）
|
||
|
|
file=file,
|
||
|
|
upload_type=upload_type or "file",
|
||
|
|
target_path=target_path,
|
||
|
|
)
|
||
|
|
return SuccessResponse(data=result, msg="上传文件成功")
|
||
|
|
|
||
|
|
|
||
|
|
@FileRouter.post("/download", summary="下载文件", dependencies=[Security(AuthPermission(["module_common:file:download"]))])
|
||
|
|
async def download_controller(
|
||
|
|
auth: Annotated[AuthSchema, Depends(get_current_user)],
|
||
|
|
background_tasks: BackgroundTasks,
|
||
|
|
file_path: Annotated[str, Body(description="文件路径")],
|
||
|
|
delete: Annotated[bool, Body(description="是否删除文件")] = False,
|
||
|
|
) -> FileResponse:
|
||
|
|
result = await FileService.download_service(file_path=file_path)
|
||
|
|
if delete:
|
||
|
|
background_tasks.add_task(UploadUtil.delete_file, Path(result.file_path))
|
||
|
|
return UploadFileResponse(file_path=result.file_path, filename=result.file_name)
|