Files
swingmusic-extended/app/store/homepage.py
T
2024-10-25 23:26:08 +03:00

31 lines
821 B
Python

from app.models.mix import Mix
from app.store.tracks import TrackStore
from app.utils.auth import get_current_userid
class HomepageStore:
entries = {
"artist_mixes": {},
}
@classmethod
def set_artist_mixes(cls, mixes: list[Mix], userid: int = 1):
idmap = {mix.id: mix for mix in mixes}
cls.entries["artist_mixes"][userid] = idmap
@classmethod
def get_artist_mixes(cls):
return [
{
"type": "mix",
"item": mix.to_dict(),
}
for mix in cls.entries["artist_mixes"]
.get(get_current_userid(), {})
.values()
]
@classmethod
def get_mix(cls, mixtype: str, mixid: str):
return cls.entries[mixtype].get(get_current_userid(), {}).get(mixid).to_full_dict()