martsforever
103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
import asyncio
|
|
import os.path
|
|
from typing import List
|
|
|
|
from fastapi import FastAPI, UploadFile, Form
|
|
|
|
from app.config.env import env
|
|
from app.model.FileModel import FileSaveService
|
|
from app.utils.mysql_utils import AsyncSessionDep
|
|
from app.utils.path_join import path_join
|
|
|
|
|
|
def add_file_route(app: FastAPI):
|
|
async def _save_file(
|
|
session: AsyncSessionDep,
|
|
file: UploadFile,
|
|
body: dict,
|
|
):
|
|
filename = file.filename
|
|
return await FileSaveService.saveFile(
|
|
session=session,
|
|
file=file,
|
|
filename=filename,
|
|
id=body.get('id', None),
|
|
file_record=body,
|
|
)
|
|
|
|
async def _save_file_list(
|
|
session: AsyncSessionDep,
|
|
file_list: List[UploadFile],
|
|
body: dict,
|
|
):
|
|
async_task_list = [asyncio.create_task(
|
|
FileSaveService.saveFile(
|
|
session=session,
|
|
file=item,
|
|
filename=item.filename,
|
|
file_record=body,
|
|
)
|
|
) for item in file_list]
|
|
result_list = await asyncio.gather(*async_task_list)
|
|
return {"result": result_list}
|
|
|
|
async def _delete_file(row_dict: dict):
|
|
file_public_path = row_dict.get('path')
|
|
path_list: List[str] = file_public_path.split('/')
|
|
original_name = path_list.pop()
|
|
file_id = path_list.pop()
|
|
save_file_path = path_join(env.file_save_path, file_id, original_name)
|
|
|
|
# 删除文件
|
|
try:
|
|
os.remove(save_file_path)
|
|
except FileNotFoundError:
|
|
print("文件不存在:" + save_file_path)
|
|
|
|
# 删除文件夹
|
|
save_dir_path = path_join(env.file_save_path, file_id)
|
|
try:
|
|
os.rmdir(save_dir_path)
|
|
except FileNotFoundError:
|
|
print("文件夹不存在:" + save_dir_path)
|
|
|
|
return {"result": True}
|
|
|
|
# 上传文件接口，文件会持久化保存
|
|
@app.post('/save_file')
|
|
async def save_file(
|
|
file: UploadFile,
|
|
session: AsyncSessionDep,
|
|
head_id: str = Form(default=None, description="父对象id"),
|
|
attr1: str = Form(default=None, description="扩展属性1"),
|
|
attr2: str = Form(default=None, description="扩展属性2"),
|
|
attr3: str = Form(default=None, description="扩展属性3")
|
|
):
|
|
return await _save_file(session, file, {
|
|
"head_id": head_id,
|
|
"attr1": attr1,
|
|
"attr2": attr2,
|
|
"attr3": attr3
|
|
})
|
|
|
|
# 上传文件接口，文件不会持久化保存，仅用于临时保存，只是为了验证
|
|
@app.post('/upload_file')
|
|
async def save_file(
|
|
file: UploadFile,
|
|
session: AsyncSessionDep,
|
|
head_id: str = Form(default=None, description="父对象id"),
|
|
attr1: str = Form(default=None, description="扩展属性1"),
|
|
attr2: str = Form(default=None, description="扩展属性2"),
|
|
attr3: str = Form(default=None, description="扩展属性3")
|
|
):
|
|
result = await _save_file(session, file, {
|
|
"head_id": head_id,
|
|
"attr1": attr1,
|
|
"attr2": attr2,
|
|
"attr3": attr3
|
|
})
|
|
if 'result' in result:
|
|
print(f"upload_file:自动删除文件「{result['result']['path']}」")
|
|
await _delete_file(result['result'])
|
|
return result
|