feat: 新增通用上传接口和通用下载接口

ruoyi-fastapi-backend/module_admin/service/common_service.py

@@ -1,5 +1,10 @@
+from fastapi import Request, BackgroundTasks
 import os
 from fastapi import UploadFile
+from datetime import datetime
+from config.env import UploadConfig
+from module_admin.entity.vo.common_vo import *
+from utils.upload_util import UploadUtil
 
 
 class CommonService:
@@ -8,11 +13,75 @@ class CommonService:
     """
 
     @classmethod
-    def upload_service(cls, path: str, task_path: str, upload_id: str, file: UploadFile):
+    def upload_service(cls, request: Request, file: UploadFile):
+        """
+        通用上传service
+        :param request: Request对象
+        :param file: 上传文件对象
+        :return: 上传结果
+        """
+        if not UploadUtil.check_file_extension(file):
+            result = dict(is_success=False, message='文件类型不合法')
+        else:
+            relative_path = f'upload/{datetime.now().strftime("%Y")}/{datetime.now().strftime("%m")}/{datetime.now().strftime("%d")}'
+            dir_path = os.path.join(UploadConfig.UPLOAD_PATH, relative_path)
+            try:
+                os.makedirs(dir_path)
+            except FileExistsError:
+                pass
+            filename = f'{file.filename.rsplit(".", 1)[0]}_{datetime.now().strftime("%Y%m%d%H%M%S")}{UploadConfig.UPLOAD_MACHINE}{UploadUtil.generate_random_number()}.{file.filename.rsplit(".")[-1]}'
+            filepath = os.path.join(dir_path, filename)
+            with open(filepath, 'wb') as f:
+                # 流式写出大型文件，这里的10代表10MB
+                for chunk in iter(lambda: file.file.read(1024 * 1024 * 10), b''):
+                    f.write(chunk)
 
-        filepath = os.path.join(path, task_path, upload_id, f'{file.filename}')
-        with open(filepath, 'wb') as f:
-            # 流式写出大型文件，这里的10代表10MB
-            for chunk in iter(lambda: file.file.read(1024 * 1024 * 10), b''):
-                f.write(chunk)
+            result = dict(
+                is_success=True,
+                result=UploadResponseModel(
+                    fileName=f'{UploadConfig.UPLOAD_PREFIX}/{relative_path}/{filename}',
+                    newFileName=filename,
+                    originalFilename=file.filename,
+                    url=f'{request.base_url}{UploadConfig.UPLOAD_PREFIX[1:]}/{relative_path}/{filename}'
+                ),
+                message='上传成功'
+            )
 
+        return CrudResponseModel(**result)
+
+    @classmethod
+    def download_services(cls, background_tasks: BackgroundTasks, file_name, delete: bool):
+        """
+        下载下载目录文件service
+        :param background_tasks: 后台任务对象
+        :param file_name: 下载的文件名称
+        :param delete: 是否在下载完成后删除文件
+        :return: 上传结果
+        """
+        filepath = os.path.join(UploadConfig.DOWNLOAD_PATH, file_name)
+        if '..' in file_name:
+            result = dict(is_success=False, message='文件名称不合法')
+        elif not UploadUtil.check_file_exists(filepath):
+            result = dict(is_success=False, message='文件不存在')
+        else:
+            result = dict(is_success=True, result=UploadUtil.generate_file(filepath), message='下载成功')
+            if delete:
+                background_tasks.add_task(UploadUtil.delete_file, filepath)
+        return CrudResponseModel(**result)
+
+    @classmethod
+    def download_resource_services(cls, resource: str):
+        """
+        下载上传目录文件service
+        :param resource: 下载的文件名称
+        :return: 上传结果
+        """
+        filepath = os.path.join(resource.replace(UploadConfig.UPLOAD_PREFIX, UploadConfig.UPLOAD_PATH))
+        filename = resource.rsplit("/", 1)[-1]
+        if '..' in filename or not UploadUtil.check_file_timestamp(filename) or not UploadUtil.check_file_machine(filename) or not UploadUtil.check_file_random_code(filename):
+            result = dict(is_success=False, message='文件名称不合法')
+        elif not UploadUtil.check_file_exists(filepath):
+            result = dict(is_success=False, message='文件不存在')
+        else:
+            result = dict(is_success=True, result=UploadUtil.generate_file(filepath), message='下载成功')
+        return CrudResponseModel(**result)