mirror of
https://github.com/open-webui/open-webui
synced 2025-04-25 00:39:36 +00:00
Rewrite logic
This commit is contained in:
parent
8609ca3657
commit
1c60b8d543
@ -2,7 +2,6 @@ import logging
|
|||||||
import os
|
import os
|
||||||
import uuid
|
import uuid
|
||||||
from fnmatch import fnmatch
|
from fnmatch import fnmatch
|
||||||
from functools import lru_cache
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
from urllib.parse import quote
|
from urllib.parse import quote
|
||||||
@ -74,19 +73,6 @@ def has_access_to_file(
|
|||||||
return has_access
|
return has_access
|
||||||
|
|
||||||
|
|
||||||
############################
|
|
||||||
# Get all files for user, with 1 cache
|
|
||||||
############################
|
|
||||||
|
|
||||||
|
|
||||||
@lru_cache(maxsize=1)
|
|
||||||
def get_all_files_for_user(user_id: str, admin: bool):
|
|
||||||
if admin:
|
|
||||||
return Files.get_files()
|
|
||||||
else:
|
|
||||||
return Files.get_files_by_user_id(user_id)
|
|
||||||
|
|
||||||
|
|
||||||
############################
|
############################
|
||||||
# Upload File
|
# Upload File
|
||||||
############################
|
############################
|
||||||
@ -205,8 +191,14 @@ async def search_files(
|
|||||||
),
|
),
|
||||||
user=Depends(get_verified_user),
|
user=Depends(get_verified_user),
|
||||||
):
|
):
|
||||||
# Retrieve files from cache
|
"""
|
||||||
files = get_all_files_for_user(user.id, user.role == "admin")
|
Search for files by filename with support for wildcard patterns.
|
||||||
|
"""
|
||||||
|
# Get files according to user role
|
||||||
|
if user.role == "admin":
|
||||||
|
files = Files.get_files()
|
||||||
|
else:
|
||||||
|
files = Files.get_files_by_user_id(user.id)
|
||||||
|
|
||||||
# Get matching files
|
# Get matching files
|
||||||
matching_files = [
|
matching_files = [
|
||||||
|
Loading…
Reference in New Issue
Block a user