diff --git a/apps/base/views.py b/apps/base/views.py index c838e2b..8e553eb 100644 --- a/apps/base/views.py +++ b/apps/base/views.py @@ -23,6 +23,12 @@ share_api = APIRouter( # 分享文本的API @share_api.post('/text/', dependencies=[Depends(admin_required)]) async def share_text(text: str = Form(...), expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), ip: str = Depends(ip_limit['upload'])): + # 获取大小 + text_size = len(text.encode('utf-8')) + # 限制 222KB + max_txt_size = 222 * 1024 # 转换为字节 + if text_size > max_txt_size: + raise HTTPException(status_code=403, detail=f'内容过多,建议采用文件形式') # 获取过期信息 expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style) # 创建一个新的FileCodes实例 @@ -47,9 +53,10 @@ async def share_text(text: str = Form(...), expire_value: int = Form(default=1, @share_api.post('/file/', dependencies=[Depends(admin_required)]) async def share_file(expire_value: int = Form(default=1, gt=0), expire_style: str = Form(default='day'), file: UploadFile = File(...), ip: str = Depends(ip_limit['upload'])): - # 检查文件大小是否超过限制 - if file.size > int(settings.uploadSize): - raise HTTPException(status_code=403, detail=f'文件大小超过限制,最大为{settings.uploadSize}字节') + if file.size > settings.uploadSize: + # 转换为 MB 并格式化输出 + max_size_mb = settings.uploadSize / (1024 * 1024) + raise HTTPException(status_code=403, detail=f'大小超过限制,最大为{max_size_mb:.2f} MB') # 获取过期信息 if expire_style not in settings.expireStyle: raise HTTPException(status_code=400, detail='过期时间类型错误')