mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-07 03:35:35 +00:00
36 lines
813 B
Python
36 lines
813 B
Python
from dataclasses import asdict
|
|
|
|
from app.models.track import Track
|
|
|
|
|
|
def track_serializer(track: Track, to_remove: set = {}, remove_disc=True) -> dict:
|
|
album_dict = asdict(track)
|
|
props = {
|
|
"date",
|
|
"genre",
|
|
"last_mod",
|
|
"og_title",
|
|
"og_album",
|
|
"copyright",
|
|
"disc",
|
|
"track",
|
|
}.union(to_remove)
|
|
|
|
if not remove_disc:
|
|
props.remove("disc")
|
|
props.remove("track")
|
|
|
|
props.update(key for key in album_dict.keys() if key.startswith("is_"))
|
|
props.remove("is_favorite")
|
|
|
|
for key in props:
|
|
album_dict.pop(key, None)
|
|
|
|
return album_dict
|
|
|
|
|
|
def serialize_tracks(
|
|
tracks: list[Track], _remove: set = {}, remove_disc=True
|
|
) -> list[dict]:
|
|
return [track_serializer(t, _remove, remove_disc) for t in tracks]
|