mirror of
https://github.com/spotipy-dev/spotipy.git
synced 2026-06-19 09:13:53 +00:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import argparse
|
|
import logging
|
|
|
|
import spotipy
|
|
from spotipy.oauth2 import SpotifyClientCredentials
|
|
|
|
|
|
logger = logging.getLogger('examples.artist_recommendations')
|
|
logging.basicConfig(level='INFO')
|
|
|
|
client_credentials_manager = SpotifyClientCredentials()
|
|
sp = spotipy.Spotify(client_credentials_manager=client_credentials_manager)
|
|
|
|
|
|
def get_args():
|
|
parser = argparse.ArgumentParser(description='Recommendations for the '
|
|
'given artist')
|
|
parser.add_argument('-a', '--artist', required=True, help='Name of Artist')
|
|
return parser.parse_args()
|
|
|
|
|
|
def get_artist(name):
|
|
results = sp.search(q='artist:' + name, type='artist')
|
|
items = results['artists']['items']
|
|
if len(items) > 0:
|
|
return items[0]
|
|
else:
|
|
return None
|
|
|
|
|
|
def show_recommendations_for_artist(artist):
|
|
results = sp.recommendations(seed_artists=[artist['id']])
|
|
for track in results['tracks']:
|
|
logger.info('Recommendation: %s - %s', track['name'],
|
|
track['artists'][0]['name'])
|
|
|
|
|
|
def main():
|
|
args = get_args()
|
|
artist = get_artist(args.artist)
|
|
if artist:
|
|
show_recommendations_for_artist(artist)
|
|
else:
|
|
logger.error("Can't find that artist", args.artist)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|