6586506c97
Feat: Added a dev test account to users for expedited login; Refactor: Improve socket connection handling and add logging for API request duration & update to ListRow key generation for improved uniqueness; Feat: Made it so streams with no set thumbnail use their category's thumbnail; Minor Fix: Corrections to db recommendation methods;
83 lines
3.3 KiB
Python
83 lines
3.3 KiB
Python
from database.database import Database
|
|
from typing import Optional, List, Tuple
|
|
|
|
def user_recommendation_category(user_id: int) -> Optional[int]:
|
|
"""
|
|
Queries user_preferences database to find users favourite streaming category and returns the category
|
|
"""
|
|
with Database() as db:
|
|
data = db.fetchone("""
|
|
SELECT category_id
|
|
FROM user_preferences
|
|
WHERE user_id = ?
|
|
ORDER BY favourability DESC
|
|
LIMIT 1
|
|
""", (user_id,))
|
|
return data
|
|
|
|
#TODO Needs to be reworked to get categories instead of streams of categories (below can be done in another function - get_streams_by_category)
|
|
def followed_categories_recommendations(user_id : int):
|
|
"""
|
|
Returns top 25 streams given a users category following
|
|
"""
|
|
with Database() as db:
|
|
categories = db.fetchall("""
|
|
SELECT user_id, title, num_viewers, categories.category_name
|
|
FROM streams
|
|
JOIN categories ON streams.category_id = categories.category_id
|
|
WHERE category_id IN (SELECT category_id FROM categories WHERE user_id = ?)
|
|
ORDER BY num_viewers DESC
|
|
LIMIT 25; """, (user_id,))
|
|
return categories
|
|
|
|
#TODO Needs to be reworked to get categories instead of streams of 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)
|
|
"""
|
|
with Database() as db:
|
|
data = db.fetchall("""
|
|
SELECT streams.category_id, streams.user_id, streams.title, users.username, streams.num_viewers, categories.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)
|
|
"""
|
|
with Database() as db:
|
|
data = db.fetchall("""
|
|
SELECT stream_id, 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
|
|
|
|
def category_recommendations():
|
|
"""
|
|
Returns a list of the top 5 most popular categories
|
|
"""
|
|
with Database() as db:
|
|
categories = db.fetchall("""
|
|
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 5;
|
|
""")
|
|
return categories
|
|
|