FEAT: Changed recommendation_utils to be able to dynamically request the amount of streams and categories with default values

This commit is contained in:
JustIceO7
2025-02-08 15:50:46 +00:00
parent e2dac21c7a
commit 105fa48f82

View File

@@ -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