mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-26 06:19:04 +00:00
zhangtao
3b04534b32
- 修改alembic配置以使用异步数据库URI和更新Base类引用 - 新增数据库Schema优化脚本，统一字段长度，添加索引，规范外键策略 - 重构API路由管理，按模块类型分组并统一前缀 - 删除示例、监控及系统各子模块的模型定义，减少冗余代码 - 将mcp_server相关代码迁移到module_ai模块下，规范模块目录结构 - 迁移example控制器至module_application.application模块，并重命名相关服务和参数名 - 优化示例控制器中的依赖和响应结构，统一命名规范
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Dict
|
|
from fastapi import UploadFile, BackgroundTasks
|
|
|
|
from app.core.exceptions import CustomException
|
|
from app.core.base_schema import UploadResponseSchema, DownloadFileSchema
|
|
from app.utils.upload_util import UploadUtil
|
|
|
|
class FileService:
|
|
"""
|
|
文件管理服务层
|
|
"""
|
|
|
|
@classmethod
|
|
async def upload_service(cls, base_url: str, file: UploadFile, upload_type: str = 'local') -> Dict:
|
|
""" 上传文件"""
|
|
if not file:
|
|
raise CustomException(msg="请选择要上传的文件")
|
|
if upload_type == 'local':
|
|
filename, filepath, file_url = await UploadUtil.upload_file(file=file, base_url=base_url)
|
|
elif upload_type == 'oss':
|
|
filename, filepath, file_url = await UploadUtil.upload_file_oss(file=file, oss_folder=file.filename)
|
|
else:
|
|
raise CustomException(msg="上传类型错误")
|
|
|
|
return UploadResponseSchema(
|
|
file_path=f'{filepath}',
|
|
file_name=filename,
|
|
origin_name=file.filename,
|
|
file_url=f'{file_url}',
|
|
).model_dump()
|
|
|
|
|
|
@classmethod
|
|
async def download_service(cls, background_tasks: BackgroundTasks, file_path: str, delete: bool) -> DownloadFileSchema:
|
|
"""
|
|
下载文件
|
|
:param background_tasks: 后台任务
|
|
:param file_path: 文件路径
|
|
:param delete: 是否在下载完成后删除文件
|
|
:return: 结果
|
|
"""
|
|
if not file_path:
|
|
raise CustomException(msg="请选择要下载的文件")
|
|
if not UploadUtil.check_file_exists(file_path):
|
|
raise CustomException(msg="文件不存在")
|
|
file_name = UploadUtil.download_file(file_path)
|
|
if delete:
|
|
background_tasks.add_task(UploadUtil.delete_file, file_path)
|
|
return DownloadFileSchema(
|
|
file_path=file_path,
|
|
file_name=file_name,
|
|
) |