bound2fy.py

# bound2fy 0.9
# author: @soxasora
import os
import spotipy
import glob
import yaml
from mutagen.flac import FLAC
from mutagen.mp3 import MP3
from termcolor import colored
from spotipy.oauth2 import SpotifyOAuth

file_config = 'config.yaml'
default_config = {
    'client_id': 'YOUR CLIENT ID',
    'client_secret': 'YOUR CLIENT SECRET',
    'redirect_uri': 'https://localhost:8888/callback'
}

def read_file(file_config):
    with open(file_config, 'r') as f:
        return yaml.safe_load(f)


if not os.path.exists(file_config):
    print(colored("Generating configuration file...", color='yellow'))
    with open(file_config, 'w') as f:
        yaml.dump(default_config, f)
    print(colored("Configuration file has been generated, please check the config file", color='green'))
    quit()
else:
    config = read_file(file_config)
    print(colored("Configuration file has been loaded", color='green'))


# Authentication stage
print(colored("Authenticating to Spotify...", 'yellow'), end='\r', flush=True)
sp = spotipy.Spotify(
    auth_manager=SpotifyOAuth(
        scope="playlist-modify-public playlist-modify-private",  # Can create/modify public playlists
        redirect_uri=config["redirect_uri"],
        client_id=config["client_id"],
        client_secret=config["client_secret"],
        cache_path="token.txt"
    )
)

user_id = sp.current_user()['id']
print("Profile: " + sp.user(user_id)['display_name']
      + colored(" Authenticated successfully to Spotify.", 'green'))

# Directory input
folder = os.path.abspath(input(colored("Drag here your folder and press escape: ", 'yellow')))
folder = folder.replace('"', '')
if folder.endswith(" "):
    folder = folder[:-1]
print(folder)

# Retrieving only flacs and mp3s
print(colored("Buckle up. This may take a while...", color='yellow'))
filelist = glob.glob(folder + "/*.flac") + glob.glob(folder + "/*.mp3")

# Sort list alphabetically
filelist.sort()

# Track IDs list initialization
track_uris = []

# Search the correspondant Spotify track for every file found
for index, song in enumerate(filelist):
    if song.endswith(".mp3"):
        file = MP3(song)
    else:
        file = FLAC(song)

    # Metadata analyzed
    try: title = file["TITLE"][0]
    except: pass
    try: artist = file["ARTIST"][0]
    except: pass
    try: album = file["ALBUM"][0]
    except: pass

    # Searching for the track on Spotify using Artist, Album, Title
    query = "artist:" + artist.replace("'", " ").lower() + " album:" + album.replace("'",
                                                                                     " ").lower() + " track:" + title.replace(
        "'", " ").lower()
    results = sp.search(q=query, type='track')
    if not results['tracks']['items']:
        found = False
        print(colored("Track: " + filelist[index] + " has not been found.", color='red'))
        c = input("Want to search for it manually? Y/N [N]: ")
        match c.lower():
            case 'y':
                while not found:
                    title = input("Enter the title of the track you would like to search for: ").replace("'",
                                                                                                         " ").lower()
                    artist = input("Enter the name of the artist of the track: ").replace("'", " ").lower()
                    album = input("Enter the title of the album of the track: ").replace("'", " ").lower()

                    query = f"artist:{artist} album:{album} track:{title}"
                    results = sp.search(q=query, limit=10, type='track')
                    if not results['tracks']['items']:
                        print(colored("Track: " + title + " by " + artist + "has not been found on Spotify.",
                                      color='red'))
                        c_2 = input("Want to try again? Y/N [N]: ").lower()
                        match c_2.lower():
                            case 'y':
                                found = False
                            case '_':
                                found = True
                    else:
                        found = True
                        print("These tracks have been found: ")
                        for index2, track in enumerate(results['tracks']['items']):
                            print(index2.__str__() + ": " + colored(track['name'], color='yellow'))
                        pick = int(input("Pick a track by its ID [0-99] or skip track [-1]: "))
                        if not pick < 0:
                            uri = results['tracks']['items'][pick]['uri']
                            # Append the track id to the list
                            track_uris.append(uri)
                        else:
                            found = True

            case '_':
                print(colored("File " + filelist[index] + " skipped.", color='yellow'))
    else:
        uri = results['tracks']['items'][0]['uri']
        # Append the first result's track id to the list
        track_uris.append(uri)

# Completed list overview
c = input("Tracks collected, want to see a recap? [Y/N]: ")
match c.lower():
    case 'y':
        for index, i in enumerate(track_uris):
            print(filelist[index] + " >> " + colored(sp.track(i)["name"], color='green'))
    case '_':
        print("List skipped.")


# Avoids "too many IDs" API error by calling it every 50
def add_to_playlist(sp, playlist_id, playlist_tracks):
    offset = 0
    while offset < len(playlist_tracks):
        sp.playlist_add_items(playlist_id=playlist_id, items=playlist_tracks[offset:offset + 50])
        offset += 50


# Playlist creation process
c = input("You're about to create a playlist on your Spotify account. Proceed Y/N [N]: ")
match c.lower():
    case 'y':
        playlist_name = input("Enter the playlist name: ")
        playlist_desc = input("Enter the playlist description: ")
        playlist_privacy = input("Would you like it to be hidden/private? [Y/N]: ")
        match playlist_privacy.lower():
            case 'y':
                playlist_privacy = False
            case '_':
                playlist_privacy = True
        playlist = sp.user_playlist_create(user=user_id, name=playlist_name, public=playlist_privacy, description=playlist_desc)
        # Add every track to the playlist
        add_to_playlist(sp, playlist['id'], track_uris)
    case '_':
        print("No playlist has been created.")

input(colored("Press any key to quit bound2fy...", color='yellow'))