forked from open-webui/open-webui
chore: delete user backend refac
This commit is contained in:
parent
7fade0bb2f
commit
48cc2c5053
2 changed files with 17 additions and 6 deletions
|
@ -153,5 +153,14 @@ class ChatTable:
|
|||
except:
|
||||
return False
|
||||
|
||||
def delete_chats_by_user_id(self, user_id: str) -> bool:
|
||||
try:
|
||||
query = Chat.delete().where(Chat.user_id == user_id)
|
||||
query.execute() # Remove the rows, return number of rows removed.
|
||||
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
|
||||
Chats = ChatTable(DB)
|
||||
|
|
|
@ -115,14 +115,16 @@ class UsersTable:
|
|||
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.
|
||||
result = Chat.delete_chats_by_user_id(id)
|
||||
|
||||
if result:
|
||||
# Delete User
|
||||
query = User.delete().where(User.id == id)
|
||||
query.execute() # Remove the rows, return number of rows removed.
|
||||
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
except:
|
||||
return False
|
||||
|
||||
|
|
Loading…
Reference in a new issue