1 files changed, 26 insertions, 29 deletions

diff --git a/cogs/vclogs.py b/cogs/vclogs.py
index 404ffe0..e5ca202 100644
--- a/cogs/vclogs.py
+++ b/cogs/vclogs.py
@@ -23,11 +23,10 @@ init_database()
 
 def load_activity_data():
     try:
-        conn = get_connection()
-        c = conn.cursor()
-        c.execute("SELECT user_id, join_time, total_active, total_afk, is_afk, afk_start, active_start, channel_name FROM voice_activity")
-        rows = c.fetchall()
-        conn.close()
+        with get_connection() as conn:
+            c = conn.cursor()
+            c.execute("SELECT user_id, join_time, total_active, total_afk, is_afk, afk_start, active_start, channel_name FROM voice_activity")
+            rows = c.fetchall()
 
         return {
             row[0]: {
@@ -48,30 +47,28 @@ def load_activity_data():
 
 def save_activity_data(data):
     try:
-        conn = get_connection()
-        c = conn.cursor()
-        c.execute("PRAGMA foreign_keys = ON")
-
-        c.execute("DELETE FROM voice_activity")
-
-        for user_id, user_data in data.items():
-            c.execute(
-                """INSERT INTO voice_activity (user_id, join_time, total_active, total_afk, is_afk, afk_start, active_start, channel_name)
-                VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
-                (
-                    user_id,
-                    user_data.get("join_time"),
-                    user_data.get("total_active", 0),
-                    user_data.get("total_afk", 0),
-                    int(user_data.get("is_afk", False)),
-                    user_data.get("afk_start"),
-                    user_data.get("active_start"),
-                    user_data.get("channel_name"),
-                ),
-            )
-
-        conn.commit()
-        conn.close()
+        with get_connection() as conn:
+            c = conn.cursor()
+            c.execute("PRAGMA foreign_keys = ON")
+
+            c.execute("DELETE FROM voice_activity")
+
+            for user_id, user_data in data.items():
+                c.execute(
+                    """INSERT INTO voice_activity (user_id, join_time, total_active, total_afk, is_afk, afk_start, active_start, channel_name)
+                    VALUES (?, ?, ?, ?, ?, ?, ?, ?)""",
+                    (
+                        user_id,
+                        user_data.get("join_time"),
+                        user_data.get("total_active", 0),
+                        user_data.get("total_afk", 0),
+                        int(user_data.get("is_afk", False)),
+                        user_data.get("afk_start"),
+                        user_data.get("active_start"),
+                        user_data.get("channel_name"),
+                    ),
+                )
+            conn.commit()
         logger.info(f"Saved voice activity for {len(data)} users")
     except Exception as e:
         logger.error(f"Error saving voice activity: {e}")