Get list of albums (#349)

Author: joostlek

src/spotifyaio/models.py

@@ -519,3 +519,10 @@ def __pre_deserialize__(cls, d: dict[str, Any]) -> dict[str, Any]:
         """Pre deserialize hook."""
         episodes = d.get("episodes", {}).pop("items", [])
         return {**d, "episodes": episodes}
+
+
+@dataclass
+class AlbumsResponse(DataClassORJSONMixin):
+    """Albums response model."""
+
+    albums: list[Album]

src/spotifyaio/spotify.py

@@ -14,6 +14,7 @@
 from spotifyaio.exceptions import SpotifyConnectionError
 from spotifyaio.models import (
     Album,
+    AlbumsResponse,
     Artist,
     ArtistResponse,
     BasePlaylist,
@@ -47,6 +48,7 @@
     TopTracksResponse,
     UserProfile,
 )
+from spotifyaio.util import get_identifier
 
 if TYPE_CHECKING:
     from spotifyaio import SimplifiedAlbum, Track
@@ -143,11 +145,22 @@ async def _put(
 
     async def get_album(self, album_id: str) -> Album:
         """Get album."""
-        identifier = album_id.split(":")[-1]
+        identifier = get_identifier(album_id)
         response = await self._get(f"v1/albums/{identifier}")
         return Album.from_json(response)
 
-    # Get a list of albums
+    async def get_albums(self, album_ids: list[str]) -> list[Album]:
+        """Get albums."""
+        if not album_ids:
+            return []
+        if len(album_ids) > 20:
+            msg = "Maximum of 20 albums can be requested at once"
+            raise ValueError(msg)
+        params: dict[str, Any] = {
+            "ids": ",".join([get_identifier(i) for i in album_ids])
+        }
+        response = await self._get("v1/albums", params=params)
+        return AlbumsResponse.from_json(response).albums
 
     # Get an album's tracks
 

src/spotifyaio/util.py

@@ -0,0 +1,6 @@
+"""Utility functions for the SpotifyAIO package."""
+
+
+def get_identifier(identifier: str) -> str:
+    """Get the identifier from a Spotify URI."""
+    return identifier.split(":")[-1]