64 lines
1.6 KiB
Python
64 lines
1.6 KiB
Python
import spotipy
|
|
import random
|
|
import simplejson as json
|
|
|
|
'''
|
|
generates a list of songs where the first word in each subsequent song
|
|
matches the last word of the previous song.
|
|
'''
|
|
|
|
sp = spotipy.Spotify()
|
|
|
|
|
|
skiplist = set(['dm', 'remix'])
|
|
max_offset = 500
|
|
seen = set()
|
|
|
|
def find_songs_that_start_with_word(word):
|
|
max_titles = 20
|
|
max_offset = 200
|
|
offset = 0
|
|
|
|
out = []
|
|
while offset < max_offset and len(out) < max_titles:
|
|
results = sp.search(q=word, type = 'track', limit=50, offset = offset)
|
|
if len(results['tracks']['items']) == 0:
|
|
break
|
|
|
|
for item in results['tracks']['items']:
|
|
name = item['name'].lower()
|
|
if name in seen:
|
|
continue
|
|
seen.add(name)
|
|
if '(' in name:
|
|
continue
|
|
if '-' in name:
|
|
continue
|
|
if '/' in name:
|
|
continue
|
|
words = name.split()
|
|
if len(words) > 1 and words[0] == word and words[-1] not in skiplist:
|
|
#print " ", name, len(out)
|
|
out.append(item)
|
|
offset += 50
|
|
#print "found", len(out), "matches"
|
|
return out
|
|
|
|
def make_chain(word):
|
|
which = 1
|
|
while True:
|
|
songs = find_songs_that_start_with_word(word)
|
|
if len(songs) > 0:
|
|
song = random.choice(songs)
|
|
print which, song['name'] + " by " + song['artists'][0]['name']
|
|
which += 1
|
|
word = song['name'].lower().split()[-1]
|
|
else:
|
|
break
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
title = ' '.join(sys.argv[1:])
|
|
make_chain(sys.argv[1].lower())
|
|
|