From d8d2f3529f291ad1ac78b41e0f54192eed668b28 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Thu, 22 Aug 2024 16:08:03 +0200 Subject: [PATCH] refac --- backend/apps/webui/routers/files.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/backend/apps/webui/routers/files.py b/backend/apps/webui/routers/files.py index ae01df1ae..ba571fc71 100644 --- a/backend/apps/webui/routers/files.py +++ b/backend/apps/webui/routers/files.py @@ -26,7 +26,7 @@ from apps.webui.models.files import ( FileModel, FileModelResponse, ) -from utils.utils import get_current_user, get_admin_user +from utils.utils import get_verified_user, get_admin_user from constants import ERROR_MESSAGES from importlib import util @@ -50,7 +50,7 @@ router = APIRouter() @router.post("/") -def upload_file(file: UploadFile = File(...), user=Depends(get_current_user)): +def upload_file(file: UploadFile = File(...), user=Depends(get_verified_user)): log.info(f"file.content_type: {file.content_type}") try: unsanitized_filename = file.filename @@ -105,7 +105,7 @@ def upload_file(file: UploadFile = File(...), user=Depends(get_current_user)): @router.get("/", response_model=list[FileModel]) -async def list_files(user=Depends(get_current_user)): +async def list_files(user=Depends(get_verified_user)): files = Files.get_files() return files @@ -153,7 +153,7 @@ async def delete_all_files(user=Depends(get_admin_user)): @router.get("/{id}", response_model=Optional[FileModel]) -async def get_file_by_id(id: str, user=Depends(get_current_user)): +async def get_file_by_id(id: str, user=Depends(get_verified_user)): file = Files.get_file_by_id(id) if file: @@ -171,7 +171,7 @@ async def get_file_by_id(id: str, user=Depends(get_current_user)): @router.get("/{id}/content", response_model=Optional[FileModel]) -async def get_file_content_by_id(id: str, user=Depends(get_current_user)): +async def get_file_content_by_id(id: str, user=Depends(get_verified_user)): file = Files.get_file_by_id(id) if file: @@ -194,7 +194,7 @@ async def get_file_content_by_id(id: str, user=Depends(get_current_user)): @router.get("/{id}/content/{file_name}", response_model=Optional[FileModel]) -async def get_file_content_by_id(id: str, user=Depends(get_current_user)): +async def get_file_content_by_id(id: str, user=Depends(get_verified_user)): file = Files.get_file_by_id(id) if file: @@ -222,7 +222,7 @@ async def get_file_content_by_id(id: str, user=Depends(get_current_user)): @router.delete("/{id}") -async def delete_file_by_id(id: str, user=Depends(get_current_user)): +async def delete_file_by_id(id: str, user=Depends(get_verified_user)): file = Files.get_file_by_id(id) if file: