From 5c7284e584315d67a77e4dc69c6f88216fae1266 Mon Sep 17 00:00:00 2001 From: WorldTeacher <41587052+WorldTeacher@users.noreply.github.com> Date: Mon, 7 Oct 2024 08:40:26 +0200 Subject: [PATCH] add function to check if to be deleted user has loans --- src/logic/database.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/src/logic/database.py b/src/logic/database.py index 16beeaa..7ba1d0d 100644 --- a/src/logic/database.py +++ b/src/logic/database.py @@ -273,7 +273,18 @@ class Database: if len(result) == 0: log.info(f"Deleting {len(result)} inactive users") for user in result: - self.deleteUser(user) + hasLoans = self.hasLoans(user[0]) + if not hasLoans: + self.deleteUser(user) + + def hasLoans(self, userid)->bool: + query = f"SELECT * FROM loans WHERE user_id = '{userid}' AND returned = 0" + conn = self.connect() + cursor = conn.cursor() + cursor.execute(query) + result = cursor.fetchall() + self.close_connection(conn) + return False if len(result) == 0 else True def deleteUser(self, userid): log.debug(f"Deleting User {userid}")