mirror of
https://github.com/JonasunderscoreJones/turbo-octo-potato.git
synced 2025-10-25 19:19:19 +02:00
first dump I could find from my laptop
This commit is contained in:
parent
8d949a52b0
commit
034419b9a5
5 changed files with 310 additions and 0 deletions
136
discographymaker.py
Normal file
136
discographymaker.py
Normal file
|
|
@ -0,0 +1,136 @@
|
|||
SPOTIFY_KEY = ""
|
||||
SPOTIFY_SECRET = ""
|
||||
SPOTIFY_USER_ID = ""
|
||||
|
||||
VERBOSE = True
|
||||
|
||||
import spotipy
|
||||
from spotipy.oauth2 import SpotifyOAuth
|
||||
|
||||
artists = []
|
||||
albums = []
|
||||
|
||||
def verboseprint(message: str) -> None:
|
||||
if VERBOSE:
|
||||
print(message)
|
||||
|
||||
def remove_duplicates(input_list: list) -> list:
|
||||
return list(set(input_list))
|
||||
|
||||
def spotifyauth() -> spotipy.Spotify:
|
||||
return spotipy.Spotify(auth_manager=SpotifyOAuth(client_id=SPOTIFY_KEY,
|
||||
client_secret=SPOTIFY_SECRET,
|
||||
redirect_uri="http://localhost:6969",
|
||||
scope="playlist-modify-public playlist-modify-private"))
|
||||
|
||||
def getDiscographyArtist(sp: spotipy.Spotify, first = False):
|
||||
while True:
|
||||
if first:
|
||||
user_artist_input = input("Input your first Artist (name or spotify ID). Enter other's when prompted next:\n")
|
||||
else:
|
||||
print("======================")
|
||||
user_artist_input = input("Input your next Artist (name or Spotify ID). Leave empty if all have been inputed\n")
|
||||
if user_artist_input == "": return
|
||||
|
||||
print("Looking up Artist...")
|
||||
try:
|
||||
if len(user_artist_input) == 22:
|
||||
if input("Did you Enter an ID? [ENTER for Yes, No for treating your Input as a name]: ") == "":
|
||||
return user_artist_input
|
||||
search_results = sp.search(q=user_artist_input, type='artist', limit=1)
|
||||
if search_results['artists']['items'][0]['name'] != user_artist_input:
|
||||
correct_input = input(" The Artist doesn't exist on Spotify. Did you mean \"" + search_results['artists']['items'][0]['name'] + "\" [ENTER for Yes, No for retry]: ")
|
||||
if correct_input == "":
|
||||
return search_results['artists']['items'][0]['uri'].replace("spotify:artist:", "")
|
||||
else:
|
||||
print("All good, try again!")
|
||||
continue
|
||||
return search_results['artists']['items'][0]['uri'].replace("spotify:artist:", "")
|
||||
except TimeoutError:
|
||||
print("\nNetwork unreachable. Please Try again...\n")
|
||||
|
||||
def ArtistAlbumGarbageHandler(apiresponse: list):
|
||||
this_artist_album_part = []
|
||||
for i in apiresponse['items']:
|
||||
this_artist_album_part.append((i['name'], i['id'], i['release_date'], i['artists'][0]['name']))
|
||||
verboseprint("Found " + str(len(this_artist_album_part)) + " Albums for " + apiresponse['items'][0]['artists'][0]['name'])
|
||||
return this_artist_album_part
|
||||
|
||||
|
||||
def getArtistAlbum(sp: spotipy.Spotify, artists: list[str]) -> list[(str, str, str, str)]:
|
||||
albums = []
|
||||
for artist in artists:
|
||||
try:
|
||||
this_artist_albums = []
|
||||
this_artist_album_part_garbage = sp.artist_albums(artist_id=artist, album_type="album,single,compilation", limit=50)
|
||||
this_artist_albums = ArtistAlbumGarbageHandler(this_artist_album_part_garbage)
|
||||
if this_artist_album_part_garbage['total'] > 50:
|
||||
print("There currently is a bug in the Spotify API that prevents fetching anything after the first 50 Albums. Given that your artist " + this_artist_album_part_garbage['items'][0]['artists'][0]['name'] + " has more than 50 Albums, you'll need to manually add the missing ones.")
|
||||
if input("Alternatively, you can end the script here since albums will be out of order [(E/e) to end] ") in ["E", "e", "end", "End"]:
|
||||
exit()
|
||||
|
||||
albums.append(this_artist_albums)
|
||||
|
||||
except TimeoutError:
|
||||
print("\nNetwork unreachable. THIS IS NOT RECOVERABLE. Please restart the process")
|
||||
exit()
|
||||
return albums
|
||||
|
||||
def insertion_sort(data_list):
|
||||
for i in range(1, len(data_list)):
|
||||
current_album = data_list[i]
|
||||
current_date = current_album[2] # Using index 2 for the release date
|
||||
|
||||
j = i - 1
|
||||
while j >= 0 and data_list[j][2] > current_date:
|
||||
data_list[j + 1] = data_list[j]
|
||||
j -= 1
|
||||
data_list[j + 1] = current_album
|
||||
return data_list
|
||||
|
||||
def createPlaylist(sp: spotipy.Spotify, songs: list[(str, str, str, str)], artist: str = None):
|
||||
if artist == None:
|
||||
artist = input("Primary Artist: ")
|
||||
playlist = sp.user_playlist_create(SPOTIFY_USER_ID, artist + " Chronological Discography", description="Full Discography of " + artist + "and Solo Releases - no inst., no OSTs")
|
||||
print('New Playlist created')
|
||||
print("Name: " + playlist['name'])
|
||||
print("ID: " + playlist['id'])
|
||||
print("Description:", playlist['description'])
|
||||
print("FAILED TO SET DESCRIPTION (Spotify API Bug)\nThe playlist was created anyways") if playlist['description'] == None else None
|
||||
while len(songs) > 100:
|
||||
sp.playlist_add_items(playlist['id'], songs[:100])
|
||||
songs = songs[100:]
|
||||
sp.playlist_add_items(playlist['id'], songs)
|
||||
|
||||
return playlist
|
||||
|
||||
def get_song_uris(album_id):
|
||||
album_tracks = sp.album_tracks(album_id)
|
||||
song_uris = [track['uri'] for track in album_tracks['items']]
|
||||
return song_uris
|
||||
|
||||
verboseprint("Authenticating...")
|
||||
sp = spotifyauth()
|
||||
verboseprint("Authenticated!")
|
||||
|
||||
artist = getDiscographyArtist(sp, True)
|
||||
while artist != None:
|
||||
artists.append(artist)
|
||||
artist = getDiscographyArtist(sp)
|
||||
|
||||
def sort_key(item):
|
||||
return item[2]
|
||||
|
||||
verboseprint("Fetching Albums...")
|
||||
albums_unsorted = [item for sublist in getArtistAlbum(sp, remove_duplicates(artists)) for item in sublist]
|
||||
albums = insertion_sort(albums_unsorted)
|
||||
verboseprint("Found " + str(len(albums))+ " Albums!")
|
||||
|
||||
all_song_uris = []
|
||||
for album_id in albums:
|
||||
album_song_uris = get_song_uris(album_id[1])
|
||||
all_song_uris.extend(album_song_uris)
|
||||
|
||||
playlist = createPlaylist(sp, all_song_uris, albums[0][3] if len(artists) == 1 else None)
|
||||
|
||||
print("Playlist created! Check your Spotify!")
|
||||
Loading…
Add table
Add a link
Reference in a new issue