app.py

import os
from flask import Flask, session, request, redirect, render_template, jsonify, url_for
from flask_session import Session
import spotipy
import uuid
from utils.playlist import Playlist

app = Flask(__name__)
app.config['SECRET_KEY'] = os.urandom(64)
app.config['SESSION_TYPE'] = 'filesystem'
app.config['SESSION_FILE_DIR'] = './.flask_session/'
Session(app)

caches_folder = './.spotify_caches/'
if not os.path.exists(caches_folder):
    os.makedirs(caches_folder)

def session_cache_path():
    return caches_folder + session.get('uuid')

post = {
        "username": '',
        "sign_out": '/sign_out',
        "generate_playlist": '/generate_playlist',
        "playlist_src": ''
    }


@app.route('/')
def index():
    if not session.get('uuid'):
        # Step 1. Visitor is unknown, give random ID
        session['uuid'] = str(uuid.uuid4())

    auth_manager = spotipy.oauth2.SpotifyOAuth(scope='user-read-currently-playing playlist-modify-private user-library-read playlist-modify-public playlist-read-private user-top-read',
                                                cache_path=session_cache_path(),
                                                show_dialog=True)

    if request.args.get("code"):
        # Step 3. Being redirected from Spotify auth page
        auth_manager.get_access_token(request.args.get("code"))
        return redirect('/')

    if not auth_manager.get_cached_token():
        # Step 2. Display sign in link when no token
        auth_url = auth_manager.get_authorize_url()
        return render_template('login.html', signin_url=auth_url)

    # Step 4. Signed in, display data
    spotify = spotipy.Spotify(auth_manager=auth_manager)
    post["username"] = spotify.me()["display_name"]
    post["playlist_src"] = ''
    playlists = spotify.current_user_playlists()
    list_playlists = []
    for playlist in playlists["items"]:
        print(playlist["owner"]["display_name"])
        if playlist["owner"]["display_name"] == post["username"]:
            list_playlists.append(playlist["name"])
    list_playlists = sorted(list_playlists)
    return render_template('home.html', posts=post,list_playlists = list_playlists)

@app.route('/sign_out')
def sign_out():
    os.remove(session_cache_path())
    try:
        # Remove the CACHE file (.cache-test) so that a new user can authorize.
        os.remove(session_cache_path())
    except OSError as e:
        print ("Error: %s - %s." % (e.filename, e.strerror))
    session.clear()
    return redirect('/')


@app.route('/generate_playlist', methods=['GET'])
def generate_playlist():
    # Gets data from the home page
    selected_playlist = request.args.get("selected_playlist")
    origin =request.args.get("origin")
    destination = request.args.get("destination")
    print("Origin: %s\nDestination: %s\nSelected playlist: %s" % (origin, destination, selected_playlist))
    auth_manager = spotipy.oauth2.SpotifyOAuth(cache_path=session_cache_path())
    if not auth_manager.get_cached_token():
        return redirect('/')

    spotify = spotipy.Spotify(auth_manager=auth_manager)

    username = spotify.current_user()['id']
    playlist_data = Playlist(origin, destination, selected_playlist, spotify, username)
    playlist_src = "https://open.spotify.com/embed/playlist/" + playlist_data.playlist_id
    playlist_data.make_roadtrip_playlist()
    return jsonify({'src':playlist_src, 'playlist':playlist_data.selected_playlist})


'''
@app.route('/remove_playlist', methods=['GET'])
def remove_playlist():
    # Gets data from the home page
    playlist_name=request.args.get("playlist")
    print(playlist_name)
    auth_manager = spotipy.oauth2.SpotifyOAuth(cache_path=session_cache_path())
    if not auth_manager.get_cached_token():
        return redirect('/')

    spotify = spotipy.Spotify(auth_manager=auth_manager)

    username = spotify.current_user()['id']
    playlist_id = playlist_name.split("/")[-1]
    spotify.user_playlist_unfollow(username, playlist_id)
    return jsonify({'src':''})
'''

'''
Following lines allow application to be run more conveniently with
`python app.py` (Make sure you're using python3)
(Also includes directive to leverage pythons threading capacity.)
'''
if __name__ == '__main__':
	app.run(threaded=True, port=int(os.environ.get("PORT", 8080)))