From 95827ccf9565bc041e0db6b1bc1d9487b2b3b6b0 Mon Sep 17 00:00:00 2001
From: JustIceO7 <oscarcao777@gmail.com>
Date: Wed, 29 Jan 2025 03:04:56 +0000
Subject: [PATCH] UPDATE: Added util functions for handling streaming routes

---
 web_server/utils/recommendation_utils.py | 56 ++++++++++++++++++++----
 web_server/utils/user_utils.py           |  4 +-
 web_server/utils/utils.py                | 22 +++++++++-
 3 files changed, 70 insertions(+), 12 deletions(-)

diff --git a/web_server/utils/recommendation_utils.py b/web_server/utils/recommendation_utils.py
index 61351fe..a772256 100644
--- a/web_server/utils/recommendation_utils.py
+++ b/web_server/utils/recommendation_utils.py
@@ -1,26 +1,64 @@
 from database.database import Database
-from user_utils import get_user_id
 from typing import Optional, List, Tuple
 
-def user_recommendation_category(username: str) -> Optional[int]:
+def user_recommendation_category(user_id: int) -> Optional[int]:
     """
     Queries user_preferences database to find users favourite streaming category and returns the category
     """
     db = Database()
     cursor = db.create_connection()
-    user_id = get_user_id(username)
 
     data = cursor.execute(
         "SELECT category_id FROM user_preferences WHERE user_id = ? ORDER BY favourability DESC LIMIT 1", (user_id,)).fetchone()
     return data[0]
 
-def recommendations_based_on_category(category_id: int) -> Optional[List[Tuple[int, str, int]]]:
+def followed_categories_recommendations(user_id: int):
     """
-    Queries stream database to get top 10 most viewed streams based on given category and returns (stream_id, title, num_viewers)
+    Returns top 25 streams given a users category following
     """
     db = Database()
-    cursor = db.create_connection()
+    db.create_connection()
+
+    categories = db.fetchall("""
+                             SELECT users.user_id, title, username, num_viewers, category_name 
+                             FROM streams 
+                             WHERE category_id IN (SELECT category_id FROM categories WHERE user_id = ?)
+                             ORDER BY num_viewers DESC
+                             LIMIT 25;  """, (user_id,))
+    return categories
+
+def recommendations_based_on_category(category_id: int) -> Optional[List[Tuple[int, str, int]]]:
+    """
+    Queries stream database to get top 25 most viewed streams based on given category and returns 
+    (user_id, title, username, num_viewers, category_name)
+    """
+    db = Database()
+    db.create_connection()
+
+    data = db.fetchall("""
+        SELECT users.user_id, title, username, num_viewers, category_name
+        FROM streams 
+        JOIN users ON users.user_id = streams.user_id
+        JOIN categories ON streams.category_id = categories.category_id
+        WHERE categories.category_id = ? 
+        ORDER BY num_viewers DESC 
+        LIMIT 25""", (category_id,))
+    return data
+
+def default_recommendations():
+    """
+    Return a list of 25 recommended live streams by number of viewers
+    (user_id, title, username, num_viewers, category_name)
+    """
+    db = Database()
+    db.create_connection()
+    data = db.fetchall("""
+        SELECT users.user_id, title, username, num_viewers, category_name
+        FROM streams 
+        JOIN users ON users.user_id = streams.user_id 
+        JOIN categories ON streams.category_id = categories.category_id
+        ORDER BY num_viewers DESC 
+        LIMIT 25
+    """)
+    return data
 
-    data = cursor.execute(
-        "SELECT user_id, stream_id, title, num_viewers FROM streams WHERE category_id = ? ORDER BY num_viwers DESC LIMIT 10", (category_id,)).fetchall()
-    return data
\ No newline at end of file
diff --git a/web_server/utils/user_utils.py b/web_server/utils/user_utils.py
index 5c8ff4d..277dec0 100644
--- a/web_server/utils/user_utils.py
+++ b/web_server/utils/user_utils.py
@@ -126,10 +126,10 @@ def reset_password(new_password: str, email: str):
     Given email and new password reset the password for a given user
     """
     db = Database()
-    cursor = db.create_connection()
+    db.create_connection()
 
     try:
-        cursor.execute("UPDATE users SET password = ? WHERE email = ?", (generate_password_hash(new_password), email))
+        db.execute("UPDATE users SET password = ? WHERE email = ?", (generate_password_hash(new_password), email))
         db.commit()
         return True
     except Exception as e:
diff --git a/web_server/utils/utils.py b/web_server/utils/utils.py
index c0d33dc..2fe6fa1 100644
--- a/web_server/utils/utils.py
+++ b/web_server/utils/utils.py
@@ -16,4 +16,24 @@ def tags():
     db = Database()
     cursor = db.create_connection()
     all_tags = cursor.execute("SELECT * FROM tags").fetchall()
-    return all_tags
\ No newline at end of file
+    return all_tags
+
+def most_popular_category():
+    """
+    Returns the most popular category based on live stream viewers
+    """
+    db = Database()
+    cursor = db.create_connection()
+
+    category = cursor.execute("""
+        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;
+    """).fetchone()
+
+    return category
+