spotify_album_downloader.py

# Imports from standard library
import json
import time
import os
import urllib.request
import sys

PYTHON_EXECUTABLE = os.path.join(os.path.dirname(sys.executable), 'python.exe') # Not pythonw.exe

# Third party packages check
print('[Top Level] Importing third party libraries')
try:
    import spotipy
    from spotipy.oauth2 import SpotifyClientCredentials
except ImportError:
    print('spotipy is not installed. You can install this by executing the following in a terminal:')
    print('\t"{0}" -m pip install spotipy'.format(PYTHON_EXECUTABLE))
    input()
    sys.exit(1)

try:
    from mutagen.mp3 import MP3
    from mutagen.id3 import ID3, APIC, error, TYER
    from mutagen.easyid3 import EasyID3
except ImportError:
    print('mutagen is not installed. You can install this by executing the following in a terminal:')
    print('\t"{0}" -m pip install mutagen'.format(PYTHON_EXECUTABLE))
    input()
    sys.exit(1)

try:
    import youtube_dl
except ImportError:
    print('youtube-dl is not installed. You can install this by executing the following in a terminal:')
    print('\t"{0}" -m pip install youtube-dl'.format(PYTHON_EXECUTABLE))
    input()
    sys.exit(1)

try:
    from bs4 import BeautifulSoup
except ImportError:
    print('beautifulsoup4 is not installed. You can install this by executing the following in a terminal:')
    print('\t"{0}" -m pip install beautifulsoup4'.format(PYTHON_EXECUTABLE))
    input()
    sys.exit(1)


# Get Keys
with open('settings.json') as data_file:
    settings = json.load(data_file)

# Create Spotify object
print('[Top Level] Setting up Spotify object...')
client_credentials_manager = SpotifyClientCredentials(client_id=settings['spotify']['client_id'], client_secret=settings['spotify']['client_secret'])
sp = spotipy.Spotify(client_credentials_manager=client_credentials_manager)
sp.trace=False


def chunks(l, n):
    # Thanks to http://stackoverflow.com/questions/312443/how-do-you-split-a-list-into-evenly-sized-chunks
    """Yield successive n-sized chunks from l."""
    for i in range(0, len(l), n):
        yield l[i:i + n]

def stripString(text):
    return "".join([i for i in text if i not in [i for i in '/\\:*?"><|']])

def my_hook(d):
    if '_percent_str' in d:
        if d['status'] == 'downloading':
            print ("\r" + d['_percent_str'], end='')
    if d['status'] == 'finished':
            print ('\rDone downloading, now converting ...')

class MyLogger(object):
    def debug(self, msg):
        pass

    def warning(self, msg):
        pass

    def error(self, msg):
        print (msg)

def downloadYoutubeToMP3(link):
    try:
        ydl_opts = {
            'format': 'bestaudio/best',
            'postprocessors': [{
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3',
            }],
            'logger': MyLogger(),
            'progress_hooks': [my_hook]
        }
        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            a = ydl.download([link])
        return True
    except Exception as e:
        print (repr(e))
        return False


song_data = {} # song_data[uri] = {'artist':x, 'album':x, 'title':x, 'ablum_art':x, 'track':x}
individual_songs = []

uri = input("[Top Level] Enter uri: ")

start_time = time.time()

username = settings['spotify_username']
playlist_id = uri.split(':')[2]
offset = 0
results = sp.user_playlist_tracks(username, playlist_id, offset=offset)
individual_songs += results['items']
while (results['next'] != None):
    offset += 100
    results = sp.user_playlist_tracks(username, playlist_id, offset=offset)
    individual_songs += results['items']

for song in individual_songs:
    song = song['track']
    song_data[song['uri']] = {'artist' : song['artists'][0]['name'],
                              'album' : song['album']['name'],
                              'title' : song['name'],
                              'ablum_art' : song['album']['images'][0]['url'],
                              'track' : str(song['track_number'])}

for song in song_data:
    try:
        print ("")
        search_term = song_data[song]['artist'] + " " + song_data[song]['title'] + " lyrics"
        Search_URL = ''.join([i for i in filter(lambda x: x in set('0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!"#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~ \t\n\r\x0b\x0c'), "https://www.youtube.com/results?search_query=" + '+'.join(search_term.replace("&", "%26").replace("=", "%3D").replace("*", "%3D").split(" ")))])

        print ("Getting " + Search_URL)
        page = urllib.request.urlopen(Search_URL)
        soup = BeautifulSoup(page.read(), "html.parser")
        search_page = soup.find_all('ol', 'item-section')[0]
        results = soup.find_all('div', 'yt-lockup-dismissable')
        for tile in results:
            try:
                link = tile.find_all('a')[0]['href']
                if "&list=" in link:
                    continue
                else:
                    print ("Video link = https://www.youtube.com" + link)
                    video_URL = "https://www.youtube.com" + link
                    break
            except:
                print ("ERROR GETING YOUTUBE URL")
        else:
            video_URL = "Error"
        if video_URL == "Error":
            print ("Failed on: " + song_data[song]['artist'] + "  - " + song_data[song]['title'])
            continue

        files_in_cd = os.listdir(os.getcwd())
        for i in files_in_cd:
            if i.endswith(".mp3"):
                os.remove(os.getcwd() + "\\" + i)

        for i in range(3):
            a = downloadYoutubeToMP3(video_URL)
            if not a:
                print ('Video download attempt {0} failed'.format(i + 1))
            else:
                break
        if not a:
            print ("Failed on: " + song_data[song]['artist'] + "  - " + song_data[song]['title'])
            print('[Top Level] If youtube-dl if out of date, execute the following in terminal: "{0}" -m pip install youtube-dl --upgrade'.format(PYTHON_EXECUTABLE))

            continue
        print ("Download Complete")
        files_in_cd = os.listdir(os.getcwd())
        for i in files_in_cd:
            if i.endswith(".mp3"):
                file = os.getcwd() + "\\" + i
        try:
            print ("Tagging \\" + file.split("\\")[-1])
        except:
            print ("Tagging (Special charaters in name)")

        audio = MP3(file, ID3=ID3)
        try:
            audio.add_tags()
        except error:
            pass
        urllib.request.urlretrieve(song_data[song]['ablum_art'], (os.getcwd() + "/TempAArtImage.jpg"))
        audio.tags.add(APIC(encoding=3, mime='image/jpeg', type=3, desc=u'cover', data=open(os.getcwd() + "/TempAArtImage.jpg", 'rb').read()))
        audio.save()
        os.remove(os.getcwd() + "/TempAArtImage.jpg")
        audio = EasyID3(file)
        audio["tracknumber"] = song_data[song]['track']
        audio["title"] = song_data[song]['title']
        audio["album"] = song_data[song]['album']
        audio["artist"] = song_data[song]['artist']
        audio.save()

        if not os.path.exists(os.getcwd() + "/output/"):
            os.makedirs(os.getcwd() + "/output/")
        title = stripString(song_data[song]['title'])
        artist = stripString(song_data[song]['artist'])
        album = stripString(song_data[song]['album'])

        try:
            os.rename(file, (os.getcwd() + "/output/" + stripString(artist + " - " + title + ".mp3")))
            print ("Saved at: " + os.getcwd() + "/output/" + stripString(artist + " - " + title + ".mp3"))
        except Exception as e:
            print ("Could not rename")
            print (repr(e))
            for i in range(10):
                try:
                    print ("Attempting: " + os.getcwd() + "/output/" + stripString(artist + " - " + title + "(" + str(i+1) + ").mp3"))
                    os.rename(file, os.getcwd() + "/output/" + stripString(artist + " - " + title + "(" + str(i+1) + ").mp3"))
                    print ("Saved at: " + os.getcwd() + "/output/" + stripString(artist + " - " + title + "(" + str(i+1) + ").mp3"))
                    break
                except Exception as ex:
                    print ("Rename Error on " + i + ": " + str(repr(ex)))
            else:
                print ("Could not rename (2nd layer)")
                os._exit(5)
    except Exception as e:
        print("Some error happened somewhere????")
        print (e)
        a = input("Ack")


a = input("Complete")