update:文件大小限制

apps/admin/depends.py

@@ -10,6 +10,10 @@ from core.settings import settings
 
 
 async def admin_required(authorization: Union[str, None] = Header(default=None), request: Request = None):
-    if authorization != settings.admin_token:
-        raise HTTPException(status_code=401, detail='未授权或授权校验失败')
-    return True
+    is_admin = authorization == settings.admin_token
+    if request.url.path.startswith('/share/'):
+        if not settings.openUpload and not is_admin:
+            raise HTTPException(status_code=403, detail='本站未开启游客上传，如需上传请先登录后台')
+    else:
+        if not is_admin:
+            raise HTTPException(status_code=401, detail='未授权或授权校验失败')

apps/base/views.py

@@ -2,13 +2,15 @@
 # @Author  : Lan
 # @File    : views.py
 # @Software: PyCharm
-from fastapi import APIRouter, Form, UploadFile, File, Depends
+from fastapi import APIRouter, Form, UploadFile, File, Depends, HTTPException
 from starlette.responses import FileResponse
 
+from apps.admin.depends import admin_required
 from apps.base.models import FileCodes
 from apps.base.pydantics import SelectFileModel
 from apps.base.utils import get_expire_info, get_file_path_name, error_ip_limit
 from core.response import APIResponse
+from core.settings import settings
 from core.storage import file_storage
 
 share_api = APIRouter(
@@ -17,7 +19,7 @@ share_api = APIRouter(
 )
 
 
-@share_api.post('/text/')
+@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')):
     expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
     await FileCodes.create(
@@ -34,8 +36,10 @@ async def share_text(text: str = Form(...), expire_value: int = Form(default=1,
     })
 
 
-@share_api.post('/file/')
+@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(...)):
+    if file.size > settings.uploadSize:
+        raise HTTPException(status_code=403, detail=f'文件大小超过限制，最大为{settings.uploadSize}字节')
     expired_at, expired_count, used_count, code = await get_expire_info(expire_value, expire_style)
     path, suffix, prefix, uuid_file_name, save_path = await get_file_path_name(file)
     await file_storage.save_file(file, save_path)