CHANGE: Modified admin utils to use with clause

web_server/utils/admin_utils.py

@@ -1,39 +1,38 @@
 from database.database import Database
 
 def check_if_admin(username):
-    # Create a connection to the database
+    """
-    db = Database()
+   Returns whether user is admin 
-    db.create_connection()
+    """
+    with Database() as db:
+        is_admin = db.fetchone("""
+                    SELECT is_admin
+                    FROM users
+                    WHERE username = ?;
+                    """, (username,))
         
-    is_admin = db.fetchone("""
+        return bool(is_admin)
-                SELECT is_admin
-                FROM users
-                WHERE username = ?;
-                """, (username,))
-    
-    return is_admin
 
 def check_if_user_exists(banned_user):
-    # Create a connection to the database
+    """
-    db = Database()
+    Returns whether user exists
-    db.create_connection()
+    """
+    with Database() as db:
+        user_exists = db.fetchone("""
+                    SELECT user_id 
+                    FROM users
+                    WHERE username = ?;""",
+                (banned_user,))
         
-    user_exists = db.fetchone("""
+        return bool(user_exists)
-                SELECT user_id 
-                FROM users
-                WHERE username = ?;""",
-            (banned_user,))
-    
-    return user_exists
 
 def ban_user(banned_user):
-    """Ban a user."""
+    """
-    # Create a connection to the database
+    Bans a user
-    db = Database()
+    """
-    db.create_connection()
+    with Database() as db:
-
+        db.execute("""
-    db.execute("""
+                    DELETE FROM users
-                DELETE FROM users
+                    WHERE username = ?;""", 
-                WHERE username = ?;""", 
+                    (banned_user)
-                (banned_user)
+                )
-            )