mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-20 01:40:15 +00:00
30 lines
675 B
Python
30 lines
675 B
Python
from dataclasses import asdict
|
|
from app.models import Album
|
|
|
|
|
|
def album_serializer(album: Album, to_remove: set[str]) -> dict:
|
|
album_dict = asdict(album)
|
|
|
|
to_remove.update(key for key in album_dict.keys() if key.startswith("is_"))
|
|
for key in to_remove:
|
|
album_dict.pop(key, None)
|
|
|
|
return album_dict
|
|
|
|
|
|
def serialize_for_card(album: Album):
|
|
props_to_remove = {
|
|
"duration",
|
|
"count",
|
|
"albumartist_hashes",
|
|
"og_title",
|
|
"base_title",
|
|
"genres",
|
|
}
|
|
|
|
return album_serializer(album, props_to_remove)
|
|
|
|
|
|
def serialize_for_card_many(albums: list[Album]):
|
|
return [serialize_for_card(a) for a in albums]
|