MAJOR: Reworked database to be easier to use and close automatically to prevent resource leaks

web_server/utils/utils.py

@@ -4,36 +4,34 @@ def categories():
     """
     Returns all possible streaming categories
     """
-    db = Database()
-    db.create_connection()
-    all_categories = db.fetchall("SELECT * FROM categories")
+    with Database() as db:
+        all_categories = db.fetchall("SELECT * FROM categories")
+    
     return all_categories
 
 def tags():
     """
     Returns all possible streaming tags
     """
-    db = Database()
-    db.create_connection()
-    all_tags = db.fetchall("SELECT * FROM tags")
+    with Database() as db:
+        all_tags = db.fetchall("SELECT * FROM tags")
+    
     return all_tags
 
 def most_popular_category():
     """
     Returns the most popular category based on live stream viewers
     """
-    db = Database()
-    db.create_connection()
-
-    category = db.fetchone("""
-        SELECT categories.category_id, categories.category_name
-        FROM streams
-        JOIN categories ON streams.category_id = categories.category_id
-        WHERE streams.isLive = 1
-        GROUP BY categories.category_name
-        ORDER BY SUM(streams.num_viewers) DESC
-        LIMIT 1;
-    """)
-
+    with Database() as db:
+        category = db.fetchone("""
+            SELECT categories.category_id, categories.category_name
+            FROM streams
+            JOIN categories ON streams.category_id = categories.category_id
+            WHERE streams.isLive = 1
+            GROUP BY categories.category_name
+            ORDER BY SUM(streams.num_viewers) DESC
+            LIMIT 1;
+        """)
+    
     return category