mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-26 14:23:48 +00:00
zhangtao
73f2823692
1. 重构后端API路由、CRUD与模块结构，整合日志管理，移除废弃demo代码 2. 优化前端组件类型定义、样式与路由配置，修复权限判断逻辑 3. 调整默认排序规则、滚动条样式与工具类函数，更新依赖与配置文件 4. 修复多处类型不匹配与默认值问题，完善表单与菜单验证逻辑
111 lines
3.1 KiB
Python
111 lines
3.1 KiB
Python
import os
|
|
|
|
from fastapi import UploadFile
|
|
|
|
from app.config.setting import settings
|
|
from app.core.base_schema import DownloadFileSchema, UploadResponseSchema
|
|
from app.core.exceptions import CustomException
|
|
from app.core.logger import logger
|
|
from app.utils.upload_util import UploadUtil
|
|
|
|
|
|
class FileService:
|
|
"""
|
|
文件管理服务层
|
|
"""
|
|
|
|
@classmethod
|
|
async def upload_service(
|
|
cls,
|
|
base_url: str,
|
|
file: UploadFile,
|
|
upload_type: str = "file",
|
|
target_path: str | None = None,
|
|
) -> UploadResponseSchema:
|
|
"""
|
|
上传文件。
|
|
|
|
参数:
|
|
- base_url (str): 基础访问 URL。
|
|
- file (UploadFile): 上传文件对象。
|
|
- upload_type (str): 上传类型，'file'、'avatar'、'param'、'resource'，默认 'file'。
|
|
- target_path (str | None): 目标目录路径，仅 resource 类型支持。
|
|
|
|
返回:
|
|
- Dict: 上传响应字典。
|
|
|
|
异常:
|
|
- CustomException: 当未选择文件或上传类型错误时抛出。
|
|
"""
|
|
filename, filepath, file_url = await UploadUtil.upload_file(
|
|
file=file,
|
|
base_url=base_url,
|
|
upload_type=upload_type,
|
|
target_path=target_path,
|
|
)
|
|
|
|
return UploadResponseSchema(
|
|
file_path=f"{filepath}",
|
|
file_name=filename,
|
|
origin_name=file.filename,
|
|
file_url=f"{file_url}",
|
|
)
|
|
|
|
@staticmethod
|
|
def _validate_download_path(file_path: str) -> str:
|
|
"""
|
|
验证下载路径是否安全。
|
|
|
|
参数:
|
|
- file_path (str): 文件路径。
|
|
|
|
返回:
|
|
- str: 安全的绝对路径。
|
|
|
|
异常:
|
|
- CustomException: 当路径不安全时抛出。
|
|
"""
|
|
if not file_path:
|
|
raise CustomException(msg="请选择要下载的文件")
|
|
|
|
dangerous_patterns = ["../", "..\\", "\0"]
|
|
for pattern in dangerous_patterns:
|
|
if pattern in file_path:
|
|
logger.error(f"检测到路径穿越攻击: {file_path}")
|
|
raise CustomException(msg="非法的文件路径")
|
|
|
|
upload_root = settings.UPLOAD_FILE_PATH.resolve()
|
|
abs_path = os.path.normpath(os.path.abspath(file_path))
|
|
|
|
if not abs_path.startswith(str(upload_root)):
|
|
logger.error(f"路径不在上传目录内: {file_path}")
|
|
raise CustomException(msg="非法的文件路径")
|
|
|
|
return abs_path
|
|
|
|
@classmethod
|
|
async def download_service(cls, file_path: str) -> DownloadFileSchema:
|
|
"""
|
|
下载文件。
|
|
|
|
参数:
|
|
- file_path (str): 文件路径。
|
|
|
|
返回:
|
|
- DownloadFileSchema: 下载文件响应对象。
|
|
|
|
异常:
|
|
- CustomException: 当未选择文件或文件不存在时抛出。
|
|
"""
|
|
safe_path = cls._validate_download_path(file_path)
|
|
|
|
if not UploadUtil.check_file_exists(safe_path):
|
|
raise CustomException(msg="文件不存在")
|
|
|
|
file_name = UploadUtil.download_file(safe_path)
|
|
|
|
return DownloadFileSchema(
|
|
file_path=safe_path,
|
|
file_name=str(file_name),
|
|
)
|