first commit

swingmusic/serializers/album.py

@@ -0,0 +1,50 @@
+from dataclasses import asdict
+
+from swingmusic.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 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]