2025-summer/team-1
12
Fork 0
Code Issues Pull requests Projects Releases Activity

add renewwww queueueue

Browse source
This commit is contained in:
Simone Tesini 2025-08-01 22:48:38 +02:00
parent c89d3af268
commit 2ab871f7a5
3 changed files with 37 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

48
backend/src/room.py
View file

@ -9,6 +9,7 @@ TAG_WEIGHT = 0.1
RANDOM_WEIGHT = 0.1
RECENT_PENALTY = 0.5
RECENT_COUNT = 10
QUEUE_SIZE = 3
type UserScoredSong = tuple[Song, int]
@ -34,11 +35,26 @@ class Room:
pin: int | None
tags: set[str]
songs: dict[str, UserScoredSong] # canzoni + voto
history: list[Song] # canzoni riprodotte (in ordine)
playing: list[Song] # canzoni che sono i riproduzione
songs: dict[str, UserScoredSong] # all songs + user score (the playlist)
history: list[Song] # all songs previously played
playing: list[Song] # queue
playing_idx: int
def renew_queue(self):
for song in self.playing:
self.history.append(song)
self.playing.clear()
self.playing_idx = 0
rankings: dict[str, float] = {}
for id, (song, user_score) in self.songs.items():
rankings[id] = self.rank_song(song, user_score).total()
# sort dict items by their values and pick the highest 3
top_items = sorted(rankings.items(), key=lambda item: item[1], reverse=True)[:QUEUE_SIZE] # [:3]
self.playing = list(map(lambda x: self.songs[x[0]][0], top_items)) # remove the ranking and take only the songs
def rank_song_from_id(self, id: str) -> Rank:
scored = self.songs[id]
return self.rank_song(scored[0], scored[1])
@ -109,18 +125,18 @@ def test_algo():
None,
set(["rock", "rap"]),
{
"paulham": (songs[0], 2),
"cisco": (songs[1], 1),
"vpn": (songs[2], 5),
"gang": (songs[3], 1),
"bertha1": (songs[4], 1),
"bertha2": (songs[5], 1),
"bertha3": (songs[6], 1),
"cexx": (songs[7], -123123),
"paulham": (songs[0], 7),
"cisco": (songs[1], 5),
"vpn": (songs[2], 11),
"gang": (songs[3], 10),
"bertha1": (songs[4], 4),
"bertha2": (songs[5], 5),
"bertha3": (songs[6], -4),
"cexx": (songs[7], 12),
},
[songs[4], songs[5], songs[0]],
[],
[songs[2], songs[0], songs[1]],
1,
)
print(room.rank_song_from_id("paulham"), room.rank_song_from_id("paulham").total())
print(room.rank_song_from_id("vpn"), room.rank_song_from_id("vpn").total())
print(room.rank_song_from_id("cexx"), room.rank_song_from_id("cexx").total())
room.renew_queue()
print(room.playing)