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:
|
except:
|
||||||
return False
|
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)
|
Chats = ChatTable(DB)
|
||||||
|
|
|
@ -115,14 +115,16 @@ class UsersTable:
|
||||||
def delete_user_by_id(self, id: str) -> bool:
|
def delete_user_by_id(self, id: str) -> bool:
|
||||||
try:
|
try:
|
||||||
# Delete User Chats
|
# Delete User Chats
|
||||||
query = Chat.delete().where(Chat.user_id == id)
|
result = Chat.delete_chats_by_user_id(id)
|
||||||
query.execute() # Remove the rows, return number of rows removed.
|
|
||||||
|
|
||||||
|
if result:
|
||||||
# Delete User
|
# Delete User
|
||||||
query = User.delete().where(User.id == id)
|
query = User.delete().where(User.id == id)
|
||||||
query.execute() # Remove the rows, return number of rows removed.
|
query.execute() # Remove the rows, return number of rows removed.
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
except:
|
except:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue