mirror of
https://github.com/open-webui/open-webui
synced 2024-11-06 00:32:05 +00:00
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
from fastapi import Response
|
|
from fastapi import Depends, FastAPI, HTTPException, status
|
|
from datetime import datetime, timedelta
|
|
from typing import List, Union, Optional
|
|
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
import time
|
|
import uuid
|
|
|
|
from apps.web.models.users import UserModel, UserRoleUpdateForm, Users
|
|
|
|
from utils.utils import (
|
|
get_password_hash,
|
|
bearer_scheme,
|
|
create_token,
|
|
)
|
|
from constants import ERROR_MESSAGES
|
|
|
|
router = APIRouter()
|
|
|
|
############################
|
|
# GetUsers
|
|
############################
|
|
|
|
|
|
@router.get("/", response_model=List[UserModel])
|
|
async def get_users(skip: int = 0, limit: int = 50, cred=Depends(bearer_scheme)):
|
|
token = cred.credentials
|
|
user = Users.get_user_by_token(token)
|
|
|
|
if user:
|
|
if user.role == "admin":
|
|
return Users.get_users(skip, limit)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.INVALID_TOKEN,
|
|
)
|
|
|
|
|
|
############################
|
|
# UpdateUserRole
|
|
############################
|
|
|
|
|
|
@router.post("/update/role", response_model=Optional[UserModel])
|
|
async def update_user_role(form_data: UserRoleUpdateForm, cred=Depends(bearer_scheme)):
|
|
token = cred.credentials
|
|
user = Users.get_user_by_token(token)
|
|
|
|
if user:
|
|
if user.role == "admin":
|
|
if user.id != form_data.id:
|
|
return Users.update_user_role_by_id(form_data.id, form_data.role)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.INVALID_TOKEN,
|
|
)
|
|
|
|
|
|
############################
|
|
# DeleteUser
|
|
############################
|
|
|
|
|
|
@router.delete("/{user_id}", response_model=bool)
|
|
async def delete_user_by_id(user_id: str, cred=Depends(bearer_scheme)):
|
|
token = cred.credentials
|
|
user = Users.get_user_by_token(token)
|
|
|
|
if user:
|
|
if user.role == "admin":
|
|
if user.id != user_id:
|
|
result = Users.delete_user_by_id(user_id)
|
|
|
|
if result:
|
|
return True
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
detail=ERROR_MESSAGES.DELETE_USER_ERROR,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail=ERROR_MESSAGES.ACTION_PROHIBITED,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_403_FORBIDDEN,
|
|
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
|
|
)
|
|
else:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_401_UNAUTHORIZED,
|
|
detail=ERROR_MESSAGES.INVALID_TOKEN,
|
|
)
|