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

ruoyi-fastapi-backend/utils/upload_util.py

@@ -0,0 +1,83 @@
+import random
+import os
+from fastapi import UploadFile
+from datetime import datetime
+from config.env import UploadConfig
+
+
+class UploadUtil:
+    """
+    上传工具类
+    """
+
+    @classmethod
+    def generate_random_number(cls):
+        """
+        生成3位数字构成的字符串
+        """
+        random_number = random.randint(1, 999)
+
+        return f'{random_number:03}'
+
+    @classmethod
+    def check_file_exists(cls, filepath):
+        """
+        检查文件是否存在
+        """
+        return os.path.exists(filepath)
+
+    @classmethod
+    def check_file_extension(cls, file: UploadFile):
+        """
+        检查文件后缀是否合法
+        """
+        file_extension = file.filename.rsplit('.', 1)[-1]
+        if file_extension in UploadConfig.DEFAULT_ALLOWED_EXTENSION:
+            return True
+        return False
+
+    @classmethod
+    def check_file_timestamp(cls, filename):
+        """
+        校验文件时间戳是否合法
+        """
+        timestamp = filename.rsplit('.', 1)[0].split('_')[-1].split(UploadConfig.UPLOAD_MACHINE)[0]
+        try:
+            datetime.strptime(timestamp, '%Y%m%d%H%M%S')
+            return True
+        except ValueError:
+            return False
+
+    @classmethod
+    def check_file_machine(cls, filename):
+        """
+        校验文件机器码是否合法
+        """
+        if filename.rsplit('.', 1)[0][-4] == UploadConfig.UPLOAD_MACHINE:
+            return True
+        return False
+
+    @classmethod
+    def check_file_random_code(cls, filename):
+        """
+        校验文件随机码是否合法
+        """
+        valid_code_list = [f"{i:03}" for i in range(1, 999)]
+        if filename.rsplit('.', 1)[0][-3:] in valid_code_list:
+            return True
+        return False
+
+    @classmethod
+    def generate_file(cls, filepath):
+        """
+        根据文件生成二进制数据
+        """
+        with open(filepath, 'rb') as response_file:
+            yield from response_file
+
+    @classmethod
+    def delete_file(cls, filepath: str):
+        """
+        根据文件路径删除对应文件
+        """
+        os.remove(filepath)