Last active
March 1, 2023 05:37
-
-
Save julian-west/b465261758198308a5262a40ed2cb78e to your computer and use it in GitHub Desktop.
Extract track name and artist from a Spotify playlist to a csv file
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """Get song titles and artists from Spotify playlist""" | |
| import csv | |
| import os | |
| import re | |
| import spotipy | |
| from dotenv import load_dotenv | |
| from spotipy.oauth2 import SpotifyClientCredentials | |
| # load credentials from .env file | |
| load_dotenv() | |
| CLIENT_ID = os.getenv("CLIENT_ID", "") | |
| CLIENT_SECRET = os.getenv("CLIENT_SECRET", "") | |
| OUTPUT_FILE_NAME = "track_info.csv" | |
| # change for your target playlist | |
| PLAYLIST_LINK = "https://open.spotify.com/playlist/6jAarBZaMmBLnSIeltPzkz?si=d42be5c6ec194bb9" | |
| # authenticate | |
| client_credentials_manager = SpotifyClientCredentials( | |
| client_id=CLIENT_ID, client_secret=CLIENT_SECRET | |
| ) | |
| # create spotify session object | |
| session = spotipy.Spotify(client_credentials_manager=client_credentials_manager) | |
| # get uri from https link | |
| if match := re.match(r"https://open.spotify.com/playlist/(.*)\?", PLAYLIST_LINK): | |
| playlist_uri = match.groups()[0] | |
| else: | |
| raise ValueError("Expected format: https://open.spotify.com/playlist/...") | |
| # get list of tracks in a given playlist (note: max playlist length 100) | |
| tracks = session.playlist_tracks(playlist_uri)["items"] | |
| # create csv file | |
| with open(OUTPUT_FILE_NAME, "w", encoding="utf-8") as file: | |
| writer = csv.writer(file) | |
| # write header column names | |
| writer.writerow(["track", "artist"]) | |
| # extract name and artist | |
| for track in tracks: | |
| name = track["track"]["name"] | |
| artists = ", ".join( | |
| [artist["name"] for artist in track["track"]["artists"]] | |
| ) | |
| # write to csv | |
| writer.writerow([name, artists]) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment