From 3f95b35accc1320ad3365c090b1807e72d1d45ba Mon Sep 17 00:00:00 2001
From: JustIceO7 <oscarcao777@gmail.com>
Date: Thu, 27 Feb 2025 00:37:19 +0000
Subject: [PATCH] CHANGE: Modified admin utils to use with clause

---
 .../BrightnessControl.tsx                     |  0
 .../Screenshot.tsx                            |  0
 web_server/utils/admin_utils.py               | 63 +++++++++----------
 3 files changed, 31 insertions(+), 32 deletions(-)
 rename frontend/src/components/{functionality => Functionality}/BrightnessControl.tsx (100%)
 rename frontend/src/components/{functionality => Functionality}/Screenshot.tsx (100%)

diff --git a/frontend/src/components/functionality/BrightnessControl.tsx b/frontend/src/components/Functionality/BrightnessControl.tsx
similarity index 100%
rename from frontend/src/components/functionality/BrightnessControl.tsx
rename to frontend/src/components/Functionality/BrightnessControl.tsx
diff --git a/frontend/src/components/functionality/Screenshot.tsx b/frontend/src/components/Functionality/Screenshot.tsx
similarity index 100%
rename from frontend/src/components/functionality/Screenshot.tsx
rename to frontend/src/components/Functionality/Screenshot.tsx
diff --git a/web_server/utils/admin_utils.py b/web_server/utils/admin_utils.py
index 61c1d42..1e6999a 100644
--- a/web_server/utils/admin_utils.py
+++ b/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()
-    db.create_connection()
-
-    is_admin = db.fetchone("""
-                SELECT is_admin
-                FROM users
-                WHERE username = ?;
-                """, (username,))
-    
-    return is_admin
+    """
+   Returns whether user is admin 
+    """
+    with Database() as db:
+        is_admin = db.fetchone("""
+                    SELECT is_admin
+                    FROM users
+                    WHERE username = ?;
+                    """, (username,))
+        
+        return bool(is_admin)
 
 def check_if_user_exists(banned_user):
-    # Create a connection to the database
-    db = Database()
-    db.create_connection()
-
-    user_exists = db.fetchone("""
-                SELECT user_id 
-                FROM users
-                WHERE username = ?;""",
-            (banned_user,))
-    
-    return user_exists
+    """
+    Returns whether user exists
+    """
+    with Database() as db:
+        user_exists = db.fetchone("""
+                    SELECT user_id 
+                    FROM users
+                    WHERE username = ?;""",
+                (banned_user,))
+        
+        return bool(user_exists)
 
 def ban_user(banned_user):
-    """Ban a user."""
-    # Create a connection to the database
-    db = Database()
-    db.create_connection()
-
-    db.execute("""
-                DELETE FROM users
-                WHERE username = ?;""", 
-                (banned_user)
-            )
\ No newline at end of file
+    """
+    Bans a user
+    """
+    with Database() as db:
+        db.execute("""
+                    DELETE FROM users
+                    WHERE username = ?;""", 
+                    (banned_user)
+                )
\ No newline at end of file