Forráskód Böngészése

chore: delete user backend refac

Timothy J. Baek 1 éve
szülő
commit
48cc2c5053
2 módosított fájl, 17 hozzáadás és 6 törlés
  1. 9 0
      backend/apps/web/models/chats.py
  2. 8 6
      backend/apps/web/models/users.py

+ 9 - 0
backend/apps/web/models/chats.py

@@ -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)

+ 8 - 6
backend/apps/web/models/users.py

@@ -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)
 
-            # Delete User
-            query = User.delete().where(User.id == id)
-            query.execute()  # Remove the rows, return number of rows removed.
+            if result:
+                # Delete User
+                query = User.delete().where(User.id == id)
+                query.execute()  # Remove the rows, return number of rows removed.
 
-            return True
+                return True
+            else:
+                return False
         except:
             return False