From b4b0a098d1b2b589364b3aade2f67a0431bb8376 Mon Sep 17 00:00:00 2001
From: Christian Hagenest <mail@hagenest.dev>
Date: Mon, 8 Jul 2024 14:31:12 +0200
Subject: [PATCH] use with db.write_transaction()

---
 chatmaild/src/chatmaild/rm_accounts.py | 15 ++++-----------
 1 file changed, 4 insertions(+), 11 deletions(-)

diff --git a/chatmaild/src/chatmaild/rm_accounts.py b/chatmaild/src/chatmaild/rm_accounts.py
index 7534f3c0..c8c870fd 100644
--- a/chatmaild/src/chatmaild/rm_accounts.py
+++ b/chatmaild/src/chatmaild/rm_accounts.py
@@ -17,22 +17,18 @@ from .database import Database
 
 
 def remove_users(db: Database, cutoff_date: int):
-    db.connect()
-    try:
+    with db.write_transaction() as conn:
         delete_query = "DELETE FROM users WHERE last_login <?"
-        db.execute_query(delete_query, (cutoff_date))
-        db.commit_changes()
-    finally:
-        db.close()
+        conn.execute(delete_query, (cutoff_date))
 
 
 def remove_user_data(db: Database, cutoff_date: int, dir: Path):
     """Collects all users where last_login < cutoff_date and deletes corresponding directories."""
     db.connect()
 
-    try:
+    with db.write_transaction() as conn:
         select_query = "SELECT user FROM users WHERE last_login <?"
-        cursor = db.execute_query(select_query, (cutoff_date,))
+        cursor = conn.execute(select_query, (cutoff_date,))
         usernames = cursor.fetchall()
 
         for username in usernames:
@@ -41,9 +37,6 @@ def remove_user_data(db: Database, cutoff_date: int, dir: Path):
                 shutil.rmtree(user_dir)
                 print(f"Deleted directory: {user_dir}")
 
-    finally:
-        db.close()
-
 
 def main():
     db = Database(sys.argv[2])