Commit f61869d9 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix: cascade shared chat delete

parent 95505bc4
......@@ -230,10 +230,13 @@ class ChatTable:
def delete_chats_by_user_id(self, user_id: str) -> bool:
try:
self.delete_shared_chats_by_user_id(user_id)
query = Chat.delete().where(Chat.user_id == user_id)
query.execute() # Remove the rows, return number of rows removed.
return True and self.delete_shared_chats_by_user_id(user_id)
return True
except:
return False
......@@ -244,6 +247,8 @@ class ChatTable:
for chat in Chat.select().where(Chat.user_id == user_id)
]
print(shared_chat_ids)
query = Chat.delete().where(Chat.user_id << shared_chat_ids)
query.execute() # Remove the rows, return number of rows removed.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment