Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-03-17 17:58:01 -05:00
parent fcf7208352
commit de3317e26b
220 changed files with 17200 additions and 22836 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/routers/folders.py
+44 -76
View File
@@ -44,7 +44,7 @@ router = APIRouter()
############################
@router.get("/", response_model=list[FolderNameIdResponse])
@router.get('/', response_model=list[FolderNameIdResponse])
async def get_folders(
request: Request,
user=Depends(get_verified_user),
@@ -56,9 +56,9 @@ async def get_folders(
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
if user.role != "admin" and not has_permission(
if user.role != 'admin' and not has_permission(
user.id,
"features.folders",
'features.folders',
request.app.state.config.USER_PERMISSIONS,
db=db,
):
@@ -72,35 +72,24 @@ async def get_folders(
# Verify folder data integrity
folder_list = []
for folder in folders:
if folder.parent_id and not Folders.get_folder_by_id_and_user_id(
folder.parent_id, user.id, db=db
):
folder = Folders.update_folder_parent_id_by_id_and_user_id(
folder.id, user.id, None, db=db
)
if folder.parent_id and not Folders.get_folder_by_id_and_user_id(folder.parent_id, user.id, db=db):
folder = Folders.update_folder_parent_id_by_id_and_user_id(folder.id, user.id, None, db=db)
if folder.data:
if "files" in folder.data:
if 'files' in folder.data:
valid_files = []
for file in folder.data["files"]:
if file.get("type") == "file":
if Files.check_access_by_user_id(
file.get("id"), user.id, "read", db=db
):
for file in folder.data['files']:
if file.get('type') == 'file':
if Files.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
valid_files.append(file)
elif file.get("type") == "collection":
if Knowledges.check_access_by_user_id(
file.get("id"), user.id, "read", db=db
):
elif file.get('type') == 'collection':
if Knowledges.check_access_by_user_id(file.get('id'), user.id, 'read', db=db):
valid_files.append(file)
else:
valid_files.append(file)
folder.data["files"] = valid_files
Folders.update_folder_by_id_and_user_id(
folder.id, user.id, FolderUpdateForm(data=folder.data), db=db
)
folder.data['files'] = valid_files
Folders.update_folder_by_id_and_user_id(folder.id, user.id, FolderUpdateForm(data=folder.data), db=db)
folder_list.append(FolderNameIdResponse(**folder.model_dump()))
@@ -112,33 +101,29 @@ async def get_folders(
############################
@router.post("/")
@router.post('/')
def create_folder(
form_data: FolderForm,
user=Depends(get_verified_user),
db: Session = Depends(get_session),
):
folder = Folders.get_folder_by_parent_id_and_user_id_and_name(
form_data.parent_id, user.id, form_data.name, db=db
)
folder = Folders.get_folder_by_parent_id_and_user_id_and_name(form_data.parent_id, user.id, form_data.name, db=db)
if folder:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Folder already exists"),
detail=ERROR_MESSAGES.DEFAULT('Folder already exists'),
)
try:
folder = Folders.insert_new_folder(
user.id, form_data, form_data.parent_id, db=db
)
folder = Folders.insert_new_folder(user.id, form_data, form_data.parent_id, db=db)
return folder
except Exception as e:
log.exception(e)
log.error("Error creating folder")
log.error('Error creating folder')
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Error creating folder"),
detail=ERROR_MESSAGES.DEFAULT('Error creating folder'),
)
@@ -147,10 +132,8 @@ def create_folder(
############################
@router.get("/{id}", response_model=Optional[FolderModel])
async def get_folder_by_id(
id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)
):
@router.get('/{id}', response_model=Optional[FolderModel])
async def get_folder_by_id(id: str, user=Depends(get_verified_user), db: Session = Depends(get_session)):
folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
return folder
@@ -166,7 +149,7 @@ async def get_folder_by_id(
############################
@router.post("/{id}/update")
@router.post('/{id}/update')
async def update_folder_name_by_id(
id: str,
form_data: FolderUpdateForm,
@@ -175,7 +158,6 @@ async def update_folder_name_by_id(
):
folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
if form_data.name is not None:
# Check if folder with same name exists
existing_folder = Folders.get_folder_by_parent_id_and_user_id_and_name(
@@ -184,20 +166,18 @@ async def update_folder_name_by_id(
if existing_folder and existing_folder.id != id:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Folder already exists"),
detail=ERROR_MESSAGES.DEFAULT('Folder already exists'),
)
try:
folder = Folders.update_folder_by_id_and_user_id(
id, user.id, form_data, db=db
)
folder = Folders.update_folder_by_id_and_user_id(id, user.id, form_data, db=db)
return folder
except Exception as e:
log.exception(e)
log.error(f"Error updating folder: {id}")
log.error(f'Error updating folder: {id}')
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Error updating folder"),
detail=ERROR_MESSAGES.DEFAULT('Error updating folder'),
)
else:
raise HTTPException(
@@ -215,7 +195,7 @@ class FolderParentIdForm(BaseModel):
parent_id: Optional[str] = None
@router.post("/{id}/update/parent")
@router.post('/{id}/update/parent')
async def update_folder_parent_id_by_id(
id: str,
form_data: FolderParentIdForm,
@@ -231,20 +211,18 @@ async def update_folder_parent_id_by_id(
if existing_folder:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Folder already exists"),
detail=ERROR_MESSAGES.DEFAULT('Folder already exists'),
)
try:
folder = Folders.update_folder_parent_id_by_id_and_user_id(
id, user.id, form_data.parent_id, db=db
)
folder = Folders.update_folder_parent_id_by_id_and_user_id(id, user.id, form_data.parent_id, db=db)
return folder
except Exception as e:
log.exception(e)
log.error(f"Error updating folder: {id}")
log.error(f'Error updating folder: {id}')
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Error updating folder"),
detail=ERROR_MESSAGES.DEFAULT('Error updating folder'),
)
else:
raise HTTPException(
@@ -262,7 +240,7 @@ class FolderIsExpandedForm(BaseModel):
is_expanded: bool
@router.post("/{id}/update/expanded")
@router.post('/{id}/update/expanded')
async def update_folder_is_expanded_by_id(
id: str,
form_data: FolderIsExpandedForm,
@@ -272,16 +250,14 @@ async def update_folder_is_expanded_by_id(
folder = Folders.get_folder_by_id_and_user_id(id, user.id, db=db)
if folder:
try:
folder = Folders.update_folder_is_expanded_by_id_and_user_id(
id, user.id, form_data.is_expanded, db=db
)
folder = Folders.update_folder_is_expanded_by_id_and_user_id(id, user.id, form_data.is_expanded, db=db)
return folder
except Exception as e:
log.exception(e)
log.error(f"Error updating folder: {id}")
log.error(f'Error updating folder: {id}')
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Error updating folder"),
detail=ERROR_MESSAGES.DEFAULT('Error updating folder'),
)
else:
raise HTTPException(
@@ -295,7 +271,7 @@ async def update_folder_is_expanded_by_id(
############################
@router.delete("/{id}")
@router.delete('/{id}')
async def delete_folder_by_id(
request: Request,
id: str,
@@ -305,9 +281,9 @@ async def delete_folder_by_id(
):
if Chats.count_chats_by_folder_id_and_user_id(id, user.id, db=db):
chat_delete_permission = has_permission(
user.id, "chat.delete", request.app.state.config.USER_PERMISSIONS, db=db
user.id, 'chat.delete', request.app.state.config.USER_PERMISSIONS, db=db
)
if user.role != "admin" and not chat_delete_permission:
if user.role != 'admin' and not chat_delete_permission:
raise HTTPException(
status_code=status.HTTP_403_FORBIDDEN,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
@@ -319,33 +295,25 @@ async def delete_folder_by_id(
folder = folders.pop()
if folder:
try:
folder_ids = Folders.delete_folder_by_id_and_user_id(
folder.id, user.id, db=db
)
folder_ids = Folders.delete_folder_by_id_and_user_id(folder.id, user.id, db=db)
for folder_id in folder_ids:
if delete_contents:
Chats.delete_chats_by_user_id_and_folder_id(
user.id, folder_id, db=db
)
Chats.delete_chats_by_user_id_and_folder_id(user.id, folder_id, db=db)
else:
Chats.move_chats_by_user_id_and_folder_id(
user.id, folder_id, None, db=db
)
Chats.move_chats_by_user_id_and_folder_id(user.id, folder_id, None, db=db)
return True
except Exception as e:
log.exception(e)
log.error(f"Error deleting folder: {id}")
log.error(f'Error deleting folder: {id}')
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT("Error deleting folder"),
detail=ERROR_MESSAGES.DEFAULT('Error deleting folder'),
)
finally:
# Get all subfolders
subfolders = Folders.get_folders_by_parent_id_and_user_id(
folder.id, user.id, db=db
)
subfolders = Folders.get_folders_by_parent_id_and_user_id(folder.id, user.id, db=db)
folders.extend(subfolders)
else: