mirror of
https://github.com/KevinMidboe/Spotify-Bot.git
synced 2025-10-29 18:00:15 +00:00
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import pprint
|
|
import sys
|
|
import spotipy
|
|
import os
|
|
import config as cnf
|
|
import spotipy.util as util
|
|
|
|
|
|
def getUsernameAndLinks():
|
|
username = cnf.username
|
|
if len(sys.argv) > 1:
|
|
trackLinks = sys.argv[1:]
|
|
else:
|
|
print("Usage: %s track_id ..." % (sys.argv[0],))
|
|
sys.exit()
|
|
|
|
return username, trackLinks
|
|
|
|
def checkAndConvertLink(trackLinks):
|
|
linkString = trackLinks[0]
|
|
#Just getting the base-62 number out of the spotify link so that links from browser, desktop and app can be used
|
|
trackStringIndex = linkString.find('track/')
|
|
questionMarkIndex = linkString.find('?')
|
|
print(trackStringIndex)
|
|
print(questionMarkIndex)
|
|
if questionMarkIndex != -1:
|
|
print("Getting the base-62 numbers out of hyperlink")
|
|
trackLinks = [linkString[trackStringIndex + len('track/'): questionMarkIndex]] #Only supports adding one song at a time so far.
|
|
|
|
else:
|
|
print("Link format is already in base-62 form")
|
|
pass
|
|
|
|
return trackLinks
|
|
|
|
def addSongToPlaylist():
|
|
|
|
username, trackLinks = getUsernameAndLinks()
|
|
trackLinks = checkAndConvertLink(trackLinks)
|
|
|
|
#This gets the authentication needed to add song to the user's playlist
|
|
scope = 'playlist-modify-public'
|
|
token = util.prompt_for_user_token(username, scope, client_id=cnf.clientID, client_secret=cnf.clientSecret, redirect_uri = cnf.redirectUri)
|
|
if token:
|
|
sp = spotipy.Spotify(token)
|
|
sp.trace = False
|
|
sp.trace_out = False
|
|
|
|
#Append to spotify playlist
|
|
results = sp.user_playlist_add_tracks(cnf.username, cnf.playlistID, trackLinks)
|
|
print(results)
|
|
else:
|
|
print("Can't get token for", username)
|
|
|
|
if __name__ == '__main__':
|
|
addSongToPlaylist()
|