mirror of
https://github.com/open-webui/open-webui
synced 2024-11-26 05:51:12 +00:00
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
from typing import Optional, Union, List, Dict
|
|
from open_webui.apps.webui.models.groups import Groups
|
|
|
|
|
|
def get_permissions(
|
|
user_id: str,
|
|
default_permissions: Dict[str, bool] = {},
|
|
) -> dict:
|
|
"""
|
|
Get all permissions for a user by combining the permissions of all groups the user is a member of.
|
|
If a permission is defined in multiple groups, the most permissive value is used.
|
|
"""
|
|
|
|
def merge_permissions(
|
|
permissions: Dict[str, bool], new_permissions: Dict[str, bool]
|
|
) -> Dict[str, bool]:
|
|
"""Merge two permission dictionaries, keeping the most permissive value."""
|
|
for key, value in new_permissions.items():
|
|
if key not in permissions:
|
|
permissions[key] = value
|
|
else:
|
|
permissions[key] = (
|
|
permissions[key] or value
|
|
) # Use the most permissive value
|
|
|
|
return permissions
|
|
|
|
user_groups = Groups.get_groups_by_member_id(user_id)
|
|
user_permissions = default_permissions.copy()
|
|
|
|
for group in user_groups:
|
|
user_permissions = merge_permissions(user_permissions, group.permissions)
|
|
|
|
return user_permissions
|
|
|
|
|
|
def has_permission(
|
|
user_id: str,
|
|
permission_key: str,
|
|
default_permissions: Dict[str, bool] = {},
|
|
) -> bool:
|
|
"""
|
|
Check if a user has a specific permission by checking the group permissions
|
|
and falls back to default permissions if not found in any group.
|
|
|
|
Permission keys can be hierarchical and separated by dots ('.').
|
|
"""
|
|
|
|
def get_permission(permissions: Dict[str, bool], keys: List[str]) -> bool:
|
|
"""Traverse permissions dict using a list of keys (from dot-split permission_key)."""
|
|
for key in keys:
|
|
if key not in permissions:
|
|
return False # If any part of the hierarchy is missing, deny access
|
|
permissions = permissions[key] # Go one level deeper
|
|
|
|
return bool(permissions) # Return the boolean at the final level
|
|
|
|
permission_hierarchy = permission_key.split(".")
|
|
|
|
# Retrieve user group permissions
|
|
user_groups = Groups.get_groups_by_member_id(user_id)
|
|
|
|
for group in user_groups:
|
|
group_permissions = group.permissions
|
|
if get_permission(group_permissions, permission_hierarchy):
|
|
return True
|
|
|
|
# Check default permissions afterwards if the group permissions don't allow it
|
|
return get_permission(default_permissions, permission_hierarchy)
|
|
|
|
|
|
def has_access(
|
|
user_id: str,
|
|
type: str = "write",
|
|
access_control: Optional[dict] = None,
|
|
) -> bool:
|
|
if access_control is None:
|
|
return type == "read"
|
|
|
|
user_groups = Groups.get_groups_by_member_id(user_id)
|
|
user_group_ids = [group.id for group in user_groups]
|
|
permission_access = access_control.get(type, {})
|
|
permitted_group_ids = permission_access.get("group_ids", [])
|
|
permitted_user_ids = permission_access.get("user_ids", [])
|
|
|
|
return user_id in permitted_user_ids or any(
|
|
group_id in permitted_group_ids for group_id in user_group_ids
|
|
)
|