File size: 1,437 Bytes
6fd136c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
"""
This script will keep only the songs that are in the Spotify "Disney Hits" playlist
"""
from dotenv import load_dotenv
load_dotenv()
import json
from collections import defaultdict
import spotipy
from spotipy.oauth2 import SpotifyClientCredentials
name = "Disney hits"
spotify = spotipy.Spotify(auth_manager=SpotifyClientCredentials())
results = spotify.search(q="playlist:" + name, type="playlist", limit=5)
items = results["playlists"]["items"]
uri = "spotify:playlist:37i9dQZF1DX8C9xQcOrE6T"
playlist = spotify.playlist(uri)
with open("data/lyrics.json", "r") as f:
data = json.load(f)
spotify_tracks = {}
for item in playlist["tracks"]["items"]:
track = item["track"]
track_name = track["name"].lower().split("-")[0].strip()
print(track_name)
spotify_tracks[track_name] = {
"id": track["id"],
"embed_url": f"https://open.spotify.com/embed/track/{track['id']}?utm_source=generator",
}
# here we add only songs that are in the Disney spotify playlist
data_filtered = defaultdict(list)
tot = 0
for movie, lyrics in data.items():
for lyric in lyrics:
name = lyric["name"].lower()
if name in spotify_tracks:
data_filtered[movie].append(
{**lyric, **{"embed_url": spotify_tracks[name]["embed_url"]}}
)
tot += 1
print(tot)
with open("data/lyrics_with_spotify_url.json", "w") as f:
json.dump(data_filtered, f)
|