diff --git a/app/controller/add_file_route.py b/app/controller/add_file_route.py new file mode 100644 index 0000000..f7c2152 --- /dev/null +++ b/app/controller/add_file_route.py @@ -0,0 +1,106 @@ +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.db_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") + ): + print("file", file) + print("head_id", head_id) + print("attr1", attr1) + print("attr2", attr2) + print("attr3", attr3) + + 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 diff --git a/app/model/FileModel.py b/app/model/FileModel.py new file mode 100644 index 0000000..780ecc8 --- /dev/null +++ b/app/model/FileModel.py @@ -0,0 +1,70 @@ +import datetime +from pathlib import Path + +from fastapi import UploadFile +from sqlmodel import Field + +from app.config.env import env +from app.model.BasicModel import BasicModel +from app.utils.create_module_service import create_model_service +from app.utils.db_utils import AsyncSessionDep +from app.utils.next_id import next_id +from app.utils.path_join import path_join + + +class FileModel(BasicModel, table=True): + __tablename__ = "pl_upload" + + name: str = Field(default=None, description='文件名称') + path: str = Field(default=None, description='文件路径') + head_id: str = Field(default=None, description='父对象id') + attr1: str = Field(default=None, description='扩展属性1') + attr2: str = Field(default=None, description='扩展属性2') + attr3: str = Field(default=None, description='扩展属性3') + + +FileService = create_model_service(Cls=FileModel) + + +# 文件保存服务 +class FileSaveService: + # 将文件保存到服务本地目录 + # 并且往附件表中插入对应的文件记录 + @staticmethod + async def saveFile( + session: AsyncSessionDep, + file: UploadFile, + filename: str, + id: str = None, + file_record: dict = None, + ): + if not id: + id = await next_id() + + datetime_string = datetime.datetime.now().strftime("%Y%m%d%H%M%S") + file_id = f"{datetime_string}_{id}" + + save_path = path_join(env.file_save_path, file_id).replace('\\', '/') + print("save_path", save_path) + + public_path = path_join(env.file_public_path, file_id).replace('\\', '/') + print("public_path", public_path) + + # parents=True 表示创建所有不存在的父目录 + # exist_ok=True 表示如果目录已存在不抛出异常 + Path(save_path).mkdir(parents=True, exist_ok=True) + + file_save_path = path_join(save_path, filename) + file_public_path = path_join(public_path, filename) + + with open(file_save_path, 'wb') as f: + f.write(await file.read()) + + file_dict = { + "id": id, + "name": filename, + "path": file_public_path, + **file_record, + } + file_cls = await FileService.item_insert(session=session, row_dict=file_dict) + return {"result": file_cls.model_dump()} diff --git a/app/server.py b/app/server.py index 334a447..9820274 100644 --- a/app/server.py +++ b/app/server.py @@ -6,6 +6,7 @@ from langserve import add_routes from app.config.env import env from app.controller.add_approve_route import add_approve_route +from app.controller.add_file_route import add_file_route from app.controller.add_hotel_route import add_hotel_route from app.controller.add_langgraph_approve_route import add_langgraph_approve_route from app.controller.add_langgraph_chat_route import add_langgraph_chat_route @@ -60,6 +61,7 @@ add_langgraph_chat_route(app) add_approve_route(app) add_reimburse_route(app) add_hotel_route(app) +add_file_route(app) @app.get("/get_env") diff --git a/app/utils/path_join.py b/app/utils/path_join.py new file mode 100644 index 0000000..5f33a3d --- /dev/null +++ b/app/utils/path_join.py @@ -0,0 +1,6 @@ +import os +from typing import List + + +def path_join(*pat_list: List[str]): + return os.path.join(*pat_list).replace('\\', '/')