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. |  | ||||||
| 
 | 
 | ||||||
|             # Delete User |             if result: | ||||||
|             query = User.delete().where(User.id == id) |                 # Delete User | ||||||
|             query.execute()  # Remove the rows, return number of rows removed. |                 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: |         except: | ||||||
|             return False |             return False | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue
	
	 Timothy J. Baek
						Timothy J. Baek