Fully working spotify song query (added spotipy)

jarvis/skills/entertainement/spotify/__init__.py

@@ -1,5 +1,6 @@
 from jarvis.skills import Skill, SkillRegistering
 from jarvis.skills.decorators import intent_file_handler
+from jarvis.skills.entertainement.spotify import spotify
 
 
 class SpotifySkill(Skill, metaclass=SkillRegistering):
@@ -8,4 +9,11 @@ class SpotifySkill(Skill, metaclass=SkillRegistering):
 
     @intent_file_handler("play_a_song.intent", "PlaySongWithSpotifyIntent")
     def handle_play_a_song(self, data):
-        print("Play song")
+        if 'song' in data:
+            song_lists_matching = spotify.query_song(data['song'])
+            if 'singer' in data:
+                song_lists_matching = spotify.query_song(data['song'], data['singer'])
+
+            if song_lists_matching is not None and len(song_lists_matching) >= 1:
+                print(song_lists_matching[0]['uri'] + " / " + song_lists_matching[0]['name'] + " / " +
+                      song_lists_matching[0]['artists'][0]['name'])

jarvis/skills/entertainement/spotify/spotify.py

@@ -0,0 +1,61 @@
+import re
+
+import spotipy
+from lingua_franca.parse import fuzzy_match
+from spotipy import SpotifyOAuth
+
+from jarvis.utils import config_utils
+
+scope = "user-read-playback-state, user-modify-playback-state, user-read-currently-playing"
+sp = spotipy.Spotify(auth_manager=SpotifyOAuth(scope=scope,
+                                               client_id=config_utils.get_in_config("SPOTIFY_CLIENT_ID"),
+                                               client_secret=config_utils.get_in_config("SPOTIFY_CLIENT_SECRET"),
+                                               redirect_uri='http://localhost:8888/callback/',
+                                               open_browser=False))
+# TODO: Investigate the open_browser and automatic auth renewing without user interaction
+
+
+def get_spotify():
+    return sp
+
+
+def query_song(song, artist=None):
+    if song is not None and artist is not None:
+        song_search = '*{}* artist:{}'.format(song, artist)
+    else:
+        song_search = song
+
+    data = get_spotify().search(q=song_search, type='track')['tracks']['items']
+    if data and len(data) > 0:
+        tracks = [(best_confidence(d['name'], song), d)
+                  for d in data]
+        tracks.sort(key=lambda x: x[0])
+
+        tracks.reverse()  # Place best matches first
+
+        # Find pretty similar tracks to the best match
+        tracks = [t for t in tracks if t[0] > tracks[0][0] - 0.1]
+
+        # Sort remaining tracks by popularity
+        tracks.sort(key=lambda x: x[1]['popularity'])
+        # print([(t[0], t[1]['name'], t[1]['artists'][0]['name']) for t in tracks])  # DEBUG
+        data = [tracks[-1][1]]
+
+        # return tracks[-1][0], {'data': data, 'name': None, 'type': 'track'}
+        return data
+
+
+def best_confidence(title, query):
+    """Find best match for a title against a query.
+    Some titles include ( Remastered 2016 ) and similar info. This method
+    will test the raw title and a version that has been parsed to remove
+    such information.
+    Arguments:
+        title: title name from spotify search
+        query: query from user
+    Returns:
+        (float) best condidence
+    """
+    best = title.lower()
+    best_stripped = re.sub(r'(\(.+\)|-.+)$', '', best).strip()
+    return max(fuzzy_match(best, query), fuzzy_match(best_stripped, query))