From 105fa48f82cf7ca1446e5b30f7e7a097382210a8 Mon Sep 17 00:00:00 2001 From: JustIceO7 Date: Sat, 8 Feb 2025 15:50:46 +0000 Subject: [PATCH] FEAT: Changed recommendation_utils to be able to dynamically request the amount of streams and categories with default values --- web_server/utils/recommendation_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/web_server/utils/recommendation_utils.py b/web_server/utils/recommendation_utils.py index d863457..ec921bd 100644 --- a/web_server/utils/recommendation_utils.py +++ b/web_server/utils/recommendation_utils.py @@ -81,9 +81,9 @@ def get_highest_view_categories(no_categories: int = 4) -> Optional[List[dict]]: """, (no_categories,)) return categories -def get_user_category_recommendations(user_id: int) -> Optional[List[dict]]: +def get_user_category_recommendations(user_id: int, no_categories: int = 4) -> Optional[List[dict]]: """ - Queries user_preferences database to find users top 5 favourite streaming category and returns the category + Queries user_preferences database to find users top favourite streaming category and returns the category """ with Database() as db: categories = db.fetchall(""" @@ -92,6 +92,6 @@ def get_user_category_recommendations(user_id: int) -> Optional[List[dict]]: JOIN user_preferences ON categories.category_id = user_preferences.category_id WHERE user_id = ? ORDER BY favourability DESC - LIMIT 5 - """, (user_id,)) + LIMIT ? + """, (user_id, no_categories)) return categories \ No newline at end of file