mirror of
https://github.com/Dvorinka/SpotifyRecAlg.git
synced 2026-06-03 20:13:03 +00:00
Tomas Dvorak
44 lines
914 B
Python
44 lines
914 B
Python
from dataclasses import asdict
|
|
|
|
from swingmusic.models.artist import Artist
|
|
|
|
|
|
def serialize_for_card(artist: Artist, include: set[str] = None):
|
|
if include is None:
|
|
include = set()
|
|
try:
|
|
artist_dict = asdict(artist)
|
|
except TypeError:
|
|
return {}
|
|
|
|
props_to_remove = {
|
|
"is_favorite",
|
|
"trackcount",
|
|
"duration",
|
|
"albumcount",
|
|
"playcount",
|
|
"playduration",
|
|
"lastplayed",
|
|
"id",
|
|
"genres",
|
|
"genrehashes",
|
|
"extra",
|
|
"created_date",
|
|
"date",
|
|
"fav_userids",
|
|
"_score",
|
|
}
|
|
|
|
if include:
|
|
props_to_remove = props_to_remove - include
|
|
|
|
for key in props_to_remove:
|
|
artist_dict.pop(key, None)
|
|
|
|
artist_dict["type"] = "artist"
|
|
return artist_dict
|
|
|
|
|
|
def serialize_for_cards(artists: list[Artist]):
|
|
return [serialize_for_card(a) for a in artists]
|