from cluster_module import ClusterModule
from database_wrapper import DatabaseWrapper
from random import shuffle
cm = ClusterModule()
db = DatabaseWrapper()
stations_output_file = "stations.pickle"
artists_output_file = "artistsandstations.pickle"
stationslist = []
stationstoartist = {} # dict where stations are mapped to predicted and seed artists
artistsforstation = []
artistsforotherstations = []
# define a list of artists, select 100 from db
allartists = db.get100ArtistFromTop500()
# select 100 artists randomly
# shuffle(allartists)
artist_list = allartists[:100]
# reverse artist list
artist_list.reverse()
counter = 0
for seedartist in artist_list:
counter = counter + 1
print "artist number " + str(counter)
