Initial commit

Main.py

@@ -0,0 +1,70 @@
+import os
+import spotipy
+from datetime import date
+from dotenv import load_dotenv
+from termcolor import colored
+
+
+def chunks(l, n):
+    for i in range(0, len(l), n):
+        yield l[i:i+n]
+
+
+# Check if the environment file exists
+if os.path.exists('.env'):
+    load_dotenv()
+    pass
+else:
+    with open('.env', 'x') as f:
+        f.write("# .env\n"
+                "SPOTIFY_CLIENT_ID=\n"
+                "SPOTIFY_CLIENT_SECRET=")
+    print(colored("Fill in the required fields in '.env'.", "yellow"))
+    exit(1)
+
+# Ensure we have our ID and secret from Spotify's API
+if os.getenv("SPOTIFY_CLIENT_ID") is None or os.getenv("SPOTIFY_CLIENT_SECRET") is None:
+    print(colored("Ensure you have included your Spotify credentials. If you don't have them, please make an app at "
+          "developer.spotify.com", "red"))
+    exit(1)
+
+print("Connecting to Spotify. " + colored("Check your browser in case there is a login prompt.", "yellow"))
+
+spotify = spotipy.Spotify(auth_manager=spotipy.oauth2.SpotifyOAuth(
+    client_id=os.getenv("SPOTIFY_CLIENT_ID"),
+    client_secret=os.getenv("SPOTIFY_CLIENT_SECRET"),
+    redirect_uri="http://localhost:8080/callback/",
+    scope="user-library-read playlist-modify-public playlist-modify-private")
+)
+
+print("Connected to Spotify successfully. Starting the converter.")
+
+flag = True  # Spotify doesn't want to return all liked songs. Set to false when done.
+offset = 0  # Offset to spotify songs
+tracks = []
+
+while flag:
+    chunk = spotify.current_user_saved_tracks(50, offset)
+
+    for track in chunk['items']:
+        tracks.append(track['track']['uri'])
+
+    if chunk['next'] is None:
+        flag = False
+
+    offset += 50
+
+# Check if the user wants the playlist to be public
+playlist_public = input(colored("Do you want the playlist to be public? (y/n)", "green") +
+                        colored(" [n] ", "yellow")) == "y"
+
+new_playlist = spotify.user_playlist_create(spotify.current_user()['id'],
+                                            "Liked Songs - " + date.today().strftime("%m-%d-%Y"),
+                                            playlist_public,
+                                            description="Auto generated from this user's liked songs.")
+
+# Use chunks of 100 due to Spotify limitations
+for chunk in chunks(tracks, 100):
+    spotify.playlist_add_items(new_playlist['id'], chunk)
+
+print("Complete! Check out your new playlist at " + new_playlist['external_urls']['spotify'])