7fbdcb3b0f
1. 新增：微信小程序端新增微信手机号登录功能（必须为企业认证小程序） 2. 新增：加入动态更新常见问题 3. 新增：新增小程序分享功能 4. 新增：小程序新增第一次登录需要修改密码 5. 新增：新增接口权限控制 6. 新增：用户新增is_staff用来判断是否为工作人员 7. 新增：软删除新增is_delete字段来判断，delete_datetime当前主要来记录时间 8. 更新：部分接口删除功能已更新，需要试用软删除的才会试用软删除 9. 更新：更新系统配置缓存功能 10. 更新：接口认证依赖项更新 11. 更新：获取系统基础配置信息与用户协议与隐私协议更新 12. 优化：优化接口与数据库操作
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Creaet Time : 2022/12/12 14:31
|
||
# @File : file_base.py
|
||
# @IDE : PyCharm
|
||
# @desc : 简要说明
|
||
|
||
import datetime
|
||
import os
|
||
import uuid
|
||
from fastapi import UploadFile
|
||
from core.exception import CustomException
|
||
from utils import status
|
||
|
||
|
||
class FileBase:
|
||
|
||
IMAGE_ACCEPT = ["image/png", "image/jpeg", "image/gif", "image/x-icon"]
|
||
VIDEO_ACCEPT = ["audio/mp4", "video/mp4", "video/mpeg"]
|
||
ALL_ACCEPT = [*IMAGE_ACCEPT, *VIDEO_ACCEPT]
|
||
|
||
@classmethod
|
||
def generate_path(cls, path: str, filename):
|
||
"""
|
||
生成文件路径
|
||
"""
|
||
if path[0] == "/":
|
||
path = path[1:]
|
||
if path[-1] == "/":
|
||
path = path[:-1]
|
||
full_date = datetime.datetime.now().date()
|
||
_filename = str(int(datetime.datetime.now().timestamp())) + str(uuid.uuid4())[:8]
|
||
return f"{path}/{full_date}/{_filename}{os.path.splitext(filename)[-1]}"
|
||
|
||
@classmethod
|
||
async def validate_file(cls, file: UploadFile, max_size: int = None, mime_types: list = None) -> bool:
|
||
"""
|
||
验证文件是否符合格式
|
||
|
||
:params max_size: 文件最大值，单位 MB
|
||
"""
|
||
if max_size:
|
||
size = len(await file.read()) / 1024 / 1024
|
||
if size > max_size:
|
||
raise CustomException(f"上传文件过大，不能超过{max_size}MB", status.HTTP_ERROR)
|
||
await file.seek(0)
|
||
if mime_types:
|
||
if file.content_type not in mime_types:
|
||
raise CustomException(f"上传文件格式错误，只支持 {'/'.join(mime_types)} 格式!", status.HTTP_ERROR)
|
||
return True
|
||
|
||
@classmethod
|
||
def get_file_type(cls, content_type: str) -> str | None:
|
||
"""
|
||
获取文件类型
|
||
|
||
0: 图片
|
||
1：视频
|
||
"""
|
||
if content_type in cls.IMAGE_ACCEPT:
|
||
return "0"
|
||
elif content_type in cls.VIDEO_ACCEPT:
|
||
return "1"
|
||
else:
|
||
return None
|