Make lowercasing album preserialization a hook (#327)

Author: joostlek

src/spotifyaio/models.py

@@ -12,18 +12,6 @@
 from mashumaro.types import Discriminator, SerializationStrategy
 
 
-class LowercaseAlbumTypeSerializationStrategy(SerializationStrategy):
-    """Serialization strategy for objects encapsulated in items."""
-
-    def serialize(self, value: AlbumType) -> str:
-        """Serialize optional string."""
-        return value
-
-    def deserialize(self, value: str) -> AlbumType:
-        """Deserialize optional string."""
-        return AlbumType(value.lower())
-
-
 class DeviceType(StrEnum):
     """Device type."""
 
@@ -132,11 +120,7 @@ def deserialize(self, value: dict[str, Any]) -> list[Any]:
 class SimplifiedAlbum(DataClassORJSONMixin):
     """Album model."""
 
-    album_type: AlbumType = field(
-        metadata=field_options(
-            serialization_strategy=LowercaseAlbumTypeSerializationStrategy()
-        )
-    )
+    album_type: AlbumType
     total_tracks: int
     album_id: str = field(metadata=field_options(alias="id"))
     images: list[Image]
@@ -146,6 +130,14 @@ class SimplifiedAlbum(DataClassORJSONMixin):
     uri: str
     artists: list[SimplifiedArtist]
 
+    @classmethod
+    def __pre_deserialize__(cls, d: dict[str, Any]) -> dict[str, Any]:
+        """Pre deserialize hook."""
+        return {
+            **d,
+            "album_type": d["album_type"].lower(),
+        }
+
 
 @dataclass
 class Album(SimplifiedAlbum):

src/spotifyaio/spotify.py

@@ -143,7 +143,8 @@ async def _put(
 
     async def get_album(self, album_id: str) -> Album:
         """Get album."""
-        response = await self._get(f"v1/albums/{album_id}")
+        identifier = album_id.split(":")[-1]
+        response = await self._get(f"v1/albums/{identifier}")
         return Album.from_json(response)
 
     async def get_artist(self, artist_id: str) -> Artist: