mirror of
https://github.com/Dvorinka/swingmusic-extended.git
synced 2026-06-03 20:13:02 +00:00
cwilvx
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
from dataclasses import asdict
|
|
from app.models import Album
|
|
|
|
|
|
def album_serializer(album: Album, to_remove: set[str]) -> dict:
|
|
try:
|
|
album_dict = asdict(album)
|
|
except TypeError:
|
|
return {}
|
|
|
|
to_remove.update(key for key in album_dict.keys() if key.startswith("is_"))
|
|
for key in to_remove:
|
|
album_dict.pop(key, None)
|
|
|
|
# remove artist images
|
|
for artist in album_dict["albumartists"]:
|
|
artist.pop("image", None)
|
|
|
|
album_dict["type"] = "album"
|
|
return album_dict
|
|
|
|
|
|
def serialize_for_card(album: Album):
|
|
props_to_remove = {
|
|
"duration",
|
|
"count",
|
|
"artisthashes",
|
|
"albumartists_hashes",
|
|
"created_date",
|
|
"og_title",
|
|
"base_title",
|
|
"genres",
|
|
"playcount",
|
|
"trackcount",
|
|
"type",
|
|
"playduration",
|
|
"genrehashes",
|
|
"fav_userids",
|
|
"extra",
|
|
"id",
|
|
"lastplayed",
|
|
"weakhash",
|
|
}
|
|
|
|
return album_serializer(album, props_to_remove)
|
|
|
|
|
|
def serialize_for_card_many(albums: list[Album]):
|
|
return [serialize_for_card(a) for a in albums]
|