tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

feat: delete user backend support

Browse source
This commit is contained in:
Timothy J. Baek 2023-12-28 23:02:49 -08:00
parent 8370465796
commit b61bb77950
3 changed files with 51 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
backend/apps/web/models/users.py
View file

@ -8,6 +8,8 @@ from utils.utils import decode_token
from utils.misc import get_gravatar_url
from apps.web.internal.db import DB
from apps.web.models.chats import Chat
####################
# User DB Schema
@ -110,5 +112,19 @@ class UsersTable:
except:
return None
def delete_user_by_id(self, id: str) -> bool:
try:
# Delete User Chats
query = Chat.delete().where(Chat.user_id == id)
query.execute() # Remove the rows, return number of rows removed.
# Delete User
query = User.delete().where(User.id == id)
query.execute() # Remove the rows, return number of rows removed.
return True
except:
return False
Users = UsersTable(DB)