server.py

"""One minute getaway."""

from jinja2 import StrictUndefined
from flask import Flask, render_template, request, redirect, session, url_for, jsonify, flash
from api import obtain_song_URL, get_song_info, get_track_ids, get_song_id_title
from goo_info import obtain_google_api_key
from loginapi import grab_env_variables, requires_auth, callback_handling
from model import connect_to_db
from image_finder import get_option_images
from queries import check_new_user, checkin_user, save_current_getaway, prior_getaways, get_getaway
import json

# imported Flash class above, then create an instance of it below
app = Flask(__name__)
app.secret_key = "JMD"
app.jinja_env.undefined = StrictUndefined
#this will throw an error if a variable is undefined in jinja


@app.route('/')
def index():
    """Homepage"""

    return render_template("home.html")


@app.route('/songprocess', methods=["POST"])
def song_process():
    """Looks for song and returns song information"""

    user_song = request.form.get("sname")
    root = get_song_info(user_song)
    track_ids = get_track_ids(root)

    if len(track_ids) == 0:
        flash("No song found with that name.")
        return redirect('/')

    elif len(track_ids) > 1:
        songid_title = get_song_id_title(root, track_ids)
        return render_template("song_prob.html", songs=songid_title)

    elif len(track_ids) == 1:
        track_id = track_ids[0].get('id')
        return redirect(url_for("select_location", track_id=track_id))


@app.route('/select_loc', methods=["GET"])
def select_location():
    """Select location and validate location."""

    track_id = request.args.get("track_id")
    session['track_id'] = track_id
    song_name = request.args.get("song_name")
    session['song_name'] = song_name
    google_api = obtain_google_api_key()
    return render_template("select_loc.html", track_id=track_id,
                           google_api=google_api, song_name=song_name)


@app.route('/play', methods=["GET"])
def getaway():
    """Plays song and shows picture."""

    # track_id = request.args.get("track_id")
    track_id = session['track_id']
    track_id = int(track_id)
    surl = obtain_song_URL(track_id)

    lurl = request.args.get("URLphoto")
    locname = request.args.get("locname")
    song_name = session['song_name']

    session['lurl'] = lurl
    session['locname'] = locname
    session['song_name'] = song_name

    return render_template("play.html", surl=surl, lurl=lurl, locname=locname,
                           track_id=track_id, song_name=song_name)


@app.route('/login', methods=["GET"])
def social_user_login():
    """Provides login using social media."""

    env_variables = grab_env_variables()
    AUTH0_CLIENT_SECRET = env_variables[0]
    AUTH0_CLIENT_ID = env_variables[1]
    AUTH0_DOMAIN = env_variables[2]
    AUTH0_CALLBACK_URL = env_variables[3]

    return render_template("login.html", AUTH0_CLIENT_SECRET=AUTH0_CLIENT_SECRET,
                           AUTH0_CLIENT_ID=AUTH0_CLIENT_ID,
                           AUTH0_DOMAIN=AUTH0_DOMAIN,
                           AUTH0_CALLBACK_URL=AUTH0_CALLBACK_URL)


@app.route('/callback')
def handle_callback():
    """Callback function for social login."""

    callback_handling()
    return redirect('/dashboard')


@app.route('/dashboard', methods=["GET"])
@requires_auth
def dashboard():
    """User information page once logged in."""

    track_id = session.get("track_id")
    lurl = session.get("lurl")
    locname = session.get("locname")
    song_name = session.get("song_name")
    user = session["profile"]
    user_find = user['user_id'].split("|")
    user_id = user_find[1]
    user_name = user['name']
    check_new_user(user_id, user_name)

    return render_template("dashboard.html", user=user, lurl=lurl, locname=locname,
                           track_id=track_id, song_name=song_name)


@app.route('/save_getaway', methods=["GET"])
@requires_auth
def save_getaway():

    user = session.get("profile")
    user_find = user['user_id'].split("|")
    user_id = user_find[1]
    track_id = session.get("track_id")
    location_url = session.get("lurl")
    location_name = session.get("locname")
    song_name = session.get("song_name")
    getaway = save_current_getaway(user_id, track_id, location_url, location_name,
                                   song_name)
    return render_template("getaway_saved.html")


@app.route('/getaways', methods=["GET"])
@requires_auth
def getaways():

    user = session.get("profile")
    user_find = user['user_id'].split("|")
    user_id = user_find[1]
    user_getaways = prior_getaways(user_id)
    return render_template("user_getaways.html", user_getaways=user_getaways)


@app.route('/replay', methods=["GET"])
def replay_getaway():

    getaway_id = request.args.get("getaway_id")
    getaway = get_getaway(getaway_id)
    seven_digital_track_id = int(getaway.song.seven_digital_track_id)

    surl = obtain_song_URL(seven_digital_track_id)

    return render_template("replay_getaway.html", surl=surl, getaway=getaway)


@app.route('/logout')
@requires_auth
def logout():
    """Provides logout and redirect to Homepage."""

    session.clear()
    return redirect('https://michdcode.auth0.com/v2/logout?returnTo=http://127.0.0.1:5000/')


@app.route('/options', methods=["GET"])
def provide_options():
    """Provides options once getaway has finished."""

    images = get_option_images()
    lurl = session.get("lurl")
    locname = session.get("locname")
    song_name = session.get("song_name")

    return render_template("options.html", images=images, lurl=lurl, locname=locname,
                           song_name=song_name)

if __name__ == "__main__":
    connect_to_db(app)
    app.debug = True
    app.run()

#server only runs if executed from terminal, cannot be imported module