feat: 文件上传接口

2026-04-02 19:55:28 +08:00
parent acc9a849a4
commit 81c73b86ac
3 changed files with 220 additions and 0 deletions
@@ -0,0 +1,102 @@
+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