feat: add helper functions to handle track, playlist and album

spotbee/spotify.py

@@ -0,0 +1,44 @@
+import spotipy
+from spotipy.oauth2 import SpotifyClientCredentials
+
+
+class SpotifyHandler:
+    def __init__(self, client_id: str, client_secret: str) -> None:
+        self.credentials = SpotifyClientCredentials(
+            client_id,
+            client_secret,
+        )
+        self.spotify = spotipy.Spotify(client_credentials_manager=self.credentials)
+
+    async def get_tracks_from_playlist(self, playlist_url: str) -> list:
+        results = self.spotify.playlist_tracks(playlist_url, additional_types=["track"])
+
+        tracklist = []
+        for item in results["items"]:
+            tracklist.append(
+                item["track"]["name"] + " - " + item["track"]["artists"][0]["name"]
+            )
+
+        return tracklist
+
+    async def get_tracks_from_album(self, playlist_url: str) -> list:
+        results = self.spotify.album_tracks(playlist_url)
+        tracklist = []
+        for item in results["items"]:
+            tracklist.append(item["name"] + " - " + item["artists"][0]["name"])
+
+        return list(set(tracklist))
+
+    async def get_track_from_url(self, track_url: str) -> list:
+        result = self.spotify.track(track_url)
+        return ["{} - {}".format(result["name"], result["artists"][0]["name"])]
+
+    async def handle_spotify_url(self, url: str) -> list | None:
+        if url.__contains__("playlist"):
+            return await self.get_tracks_from_playlist(url)
+        elif url.__contains__("album"):
+            return await self.get_tracks_from_album(url)
+        elif url.__contains__("track"):
+            return await self.get_track_from_url(url)
+        else:
+            return None