echo_flask_ask_sonos.py

'''
The script uses the flask extension flask-ask, which was written specifically for python-
based Alexa apps.
Uses flask_ask_mp.py to do the actual interactions with Sonos
Also need ngrok http 5000
and url will look like 1234.ngrok.io/sonos
I do have a lambda program but it is just to really proxy to the right raspi
'''

from flask import Flask #, request
from flask_ask import Ask, statement
import itertools
import random
from operator import itemgetter 
import pysolr
from multiprocessing.connection import Client 
from config import solr_uri #, user_id #,last_fm_api_key, user_id

app = Flask(__name__)
app.config['ASK_VERIFY_REQUESTS'] = False
ask = Ask(app, '/sonos')

address = ('localhost', 6000)
conn = Client(address, authkey='secret password')

solr = pysolr.Solr(solr_uri+'/solr/sonos_companion/', timeout=10) #port 8983 is incorporated in the ngrok url

def get_track(artist, title):    
    # title must be present; artist is optional
    print "artist =",artist
    print "title =",title

    s = 'title:' + ' AND title:'.join(title.split())
    if artist:
        s = s + ' artist:' + ' AND artist:'.join(artist.split())

    result = solr.search(s, rows=1) #**{'rows':1})
    if not len(result):
        return 

    track = result.docs[0]
    return track['uri']

@ask.intent('PlayAlbum', mapping={'album':'myalbum', 'artist':'myartist'})
def play_album(album, artist):
    # album must be present; artist is optional

    print "album =",album
    print "artist=",artist

    if album:
        s = 'album:' + ' AND album:'.join(album.split())
        if artist:
            s = s + ' artist:' + ' AND artist:'.join(artist.split())

        result = solr.search(s, fl='score,track,uri,album', sort='score desc', rows=25) #**{'rows':25}) #only brings back actual matches but 25 seems like max for most albums
        tracks = result.docs
        if  tracks:
            selected_album = tracks[0]['album']
            try:
                tracks = sorted([t for t in tracks],key=itemgetter('track'))
            except KeyError:
                pass
            # The if t['album']==selected_album only comes into play if we retrieved more than one album
            uris = [t['uri'] for t in tracks if t['album']==selected_album]

            conn.send({'action':'play', 'uris':uris})

            output_speech = "Using Flask Ask, I will play {} songs from {}".format(len(uris), selected_album)
        else:
            output_speech = "I couldn't find any songs from album {}.".format(album)

    else:
        output_speech = "I couldn't even find the album."

    return statement(output_speech)

@ask.intent('Shuffle', mapping={'artist':'myartist'})
def shuffle(artist):
    if not artist:
        return statement("I couldn't find the artist you were looking for.  Sorry.")

    s = 'artist:' + ' AND artist:'.join(artist.split())
    result = solr.search(s, fl='artist,title,uri', rows=500) 
    count = len(result)
    if not count:
        return statement("I couldn't find any tracks for {}".format(artist))

    print "Total track count for {} was {}".format(artist, count)
    tracks = result.docs
    k = 10 if count >= 10 else count
    selected_tracks = random.sample(tracks, k)
    uris = [t.get('uri') for t in selected_tracks]
    conn.send({'action':'play', 'uris':uris})
    return statement("I will shuffle songs by {}.".format(artist))

@ask.intent('Mix', mapping={'artist1':'myartista', 'artist2':'myartistb'})
def mix(artist1, artist2):
    print "artist1, artist2 = ",artist1,artist2
    uris = []
    for artist in (artist1, artist2):
        if artist:
            s = 'artist:' + ' AND artist:'.join(artist.split())
            result = solr.search(s, fl='artist,title,uri', rows=500) 
            count = len(result)
            if count:
                print "Total track count for {} was {}".format(artist, count)
                tracks = result.docs
                k = 5 if count >= 5 else count
                selected_tracks = random.sample(tracks, k)
                uris.append([t.get('uri') for t in selected_tracks])
            else:
                output_speech = "I couldn't find any tracks for {}".format(artist)
                return statement(output_speech)
        else:
            output_speech = "I couldn't find one or both of the artists you were looking for."
            return statement(output_speech)

    iters = [iter(y) for y in uris]
    uris = list(it.next() for it in itertools.cycle(iters))
    conn.send({'action':'play', 'uris':uris})
    output_speech = "I will shuffle mix songs by {} and {}.".format(artist1, artist2)
    return statement(output_speech)

#note the decorator will set add to None when accessed via URL but not when called directly by add_track
@ask.intent('PlayTrack', mapping={'title':'mytitle', 'artist':'myartist'})
def play_track(title, artist, add): #note the decorator will set add to None
    # title must be present; artist is optional
    print "artist =",artist
    print "title =",title
    print "add =", add

    if not title:
        return statement("I couldn't find the track.")

    s = 'title:' + ' AND title:'.join(title.split())
    if artist:
        s = s + ' artist:' + ' AND artist:'.join(artist.split())

    result = solr.search(s, rows=1) #**{'rows':1})
    if not len(result):
        return statement("I couldn't find the track {} by {}.".format(title,artist))

    track = result.docs[0]
    uri = track['uri']
    action = 'add' if add else 'play'
    conn.send({'action':action, 'uris':[uri]})
    return statement("I will {} {} by {} from album {}".format(action, track['title'], track['artist'], track['album']))

@ask.intent('AddTrack', mapping={'title':'mytitle', 'artist':'myartist'})
def add_track(title, artist):
    r = play_track(title, artist, True)
    return r

@ask.intent('AMAZON.ResumeIntent')
def resume():
    conn.send({'action':'resume'})
    return statement("I will resume what was playing.")

@ask.intent('AMAZON.PauseIntent')
def pause():
    conn.send({'action':'pause'})
    return statement("I will pause what was playing.")

@ask.intent('AMAZON.NextIntent')
def next():
    conn.send({'action':'next'})
    return statement("I will skip to the next track.")

@ask.intent('TurnTheVolume')
def turn_the_volume(volume):
    if volume in ('increase','louder','higher','up'):
        conn.send({'action':'louder'})
        return statement("I will turn the volume up.")
    elif volume in ('decrease', 'down','quieter','lower'):
        conn.send({'action':'quieter'})
        return statement("I will turn the volume down.")
    else:
        return statement("I don't know what you asked me to do to the volume.")

try:
    app.run(debug=True,
            port=5000,
            threaded=False,
            use_reloader=False,
            use_debugger=True,
            host='0.0.0.0'
            )
finally:
    print "Disconnecting clients"

print "Done"