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