初始提交

2026-07-24 09:43:10 +08:00
commit b5b4721456
793 changed files with 116305 additions and 0 deletions
@@ -0,0 +1,99 @@
+import os
+from datetime import datetime
+
+import aiofiles
+from fastapi import BackgroundTasks, Request, UploadFile
+
+from common.vo import CrudResponseModel
+from config.env import UploadConfig
+from exceptions.exception import ServiceException
+from module_admin.entity.vo.common_vo import UploadResponseModel
+from utils.upload_util import UploadUtil
+
+
+class CommonService:
+    """
+    通用模块服务层
+    """
+
+    @classmethod
+    async def upload_service(cls, request: Request, file: UploadFile) -> CrudResponseModel:
+        """
+        通用上传service
+
+        :param request: Request对象
+        :param file: 上传文件对象
+        :return: 上传结果
+        """
+        if not UploadUtil.check_file_extension(file):
+            raise ServiceException(message='文件类型不合法')
+        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)
+        async with aiofiles.open(filepath, 'wb') as f:
+            # 流式写出大型文件，这里的10代表10MB
+            while True:
+                chunk = await file.read(1024 * 1024 * 10)
+                if not chunk:
+                    break
+                await f.write(chunk)
+
+        return CrudResponseModel(
+            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='上传成功',
+        )
+
+    @classmethod
+    async def download_services(
+        cls, background_tasks: BackgroundTasks, file_name: str, delete: bool
+    ) -> CrudResponseModel:
+        """
+        下载下载目录文件service
+
+        :param background_tasks: 后台任务对象
+        :param file_name: 下载的文件名称
+        :param delete: 是否在下载完成后删除文件
+        :return: 上传结果
+        """
+        filepath = os.path.join(UploadConfig.DOWNLOAD_PATH, file_name)
+        if '..' in file_name:
+            raise ServiceException(message='文件名称不合法')
+        if not UploadUtil.check_file_exists(filepath):
+            raise ServiceException(message='文件不存在')
+        if delete:
+            background_tasks.add_task(UploadUtil.delete_file, filepath)
+        return CrudResponseModel(is_success=True, result=UploadUtil.generate_file(filepath), message='下载成功')
+
+    @classmethod
+    async def download_resource_services(cls, resource: str) -> CrudResponseModel:
+        """
+        下载上传目录文件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)
+        ):
+            raise ServiceException(message='文件名称不合法')
+        if not UploadUtil.check_file_exists(filepath):
+            raise ServiceException(message='文件不存在')
+        return CrudResponseModel(is_success=True, result=UploadUtil.generate_file(filepath), message='下载成功')