server.py

"""server for flix2comix
"""

from jinja2 import StrictUndefined
from flask import (Flask, jsonify, render_template,
                   redirect, request, flash, session, Markup)
from flask_debugtoolbar import DebugToolbarExtension
from model import (User, Movie, Comic, MovieRating, ComicRec,
                   connect_to_db, db)
from sqlalchemy.orm.exc import NoResultFound
import os

# from passlib.hash import argon2

import analysis


app = Flask(__name__)

app.secret_key = "monkey"

app.jinja_env.endefined = StrictUndefined


@app.route('/')
def index():
    """Homepage."""
    return render_template("homepage.html")


@app.route('/register', methods=["POST"])
def process_register_form():
    email = request.form.get("email")
    password = request.form.get("password")
    nickname = request.form.get("nickname")

    #Check if user already exists, if not create user and add to table
    try:
        user_check = User.query.filter_by(email=email).one()
        flash("Email already registered!")
        return render_template("/")
    except NoResultFound:
        # password = argon2.hash(password)
        user = User(email=email,
                    password=password,
                    nickname=nickname)
        db.session.add(user)
        db.session.commit()

        session['user_id'] = user.user_id
        session['movies_dismissed'] = []
        flash("You are registered & logged in.")
        return redirect("/movies")




@app.route('/login-validation', methods=["POST"])
def process_login_form():
    """confirm user's login, add cookie to session.
    """
    email = request.form.get("email")
    password = request.form.get("password") 

    try:
        user_check = User.query.filter_by(email=email).one()
    except NoResultFound:
        flash("Email not found")
        return redirect("/login")

    # if user found, check if password matches
    # if argon2.verify(password, user_check.password):    #hash not workign on heroku
    if password == user_check.password:
        session['user_id'] = user_check.user_id
        session['movies_dismissed'] = []
        flash("Logged in successfully")
        return redirect("/")
    else:
        flash("Wrong password!")
        return redirect("/")


@app.route('/logout')
def logout():
    del session['user_id']
    del session['movies_dismissed']
    flash("Logged out successfully")
    return redirect("/")


@app.route('/movies')
def show_movies():
    """display the page in which user will rate movies.
    """
    return render_template("movies.html")


@app.route('/getMovie')
def getMovie():
    """gets one movie object from db, checking that the user
        has not already rated or dismissed it. Builds html
        for the mini-form to display on the movie page, and returns HMTL.
    """
    seen_list = session['movies_dismissed']
    u_id = session['user_id']
    random_movie = analysis.get_random_movie(seen_list, u_id)
    if random_movie:
        session['movies_dismissed'].append(random_movie.movie_id)
        movie_html = Markup(render_template('movie.html', movie=random_movie))
        return movie_html
    else:
        no_more_html = Markup(render_template('no_more.html'))
        return no_more_html


@app.route('/rate', methods=['POST'])
def submit_movie_rating():
    """handle ajax call to process one movie.
    """

    movie_id = int(request.form.get("movie_id"))
    rating = int(request.form.get("rating")) - 3
    u_id = session["user_id"]

    analysis.process_user_rating(u_id, movie_id, rating)

    return redirect('/getMovie')


@app.route('/skipMovie', methods=['POST'])
def skip_movie():
    movie_id = int(request.form.get("movie_id"))
    session['movies_dismissed'].append(movie_id)
    return redirect('/getMovie')


@app.route('/rateCount')
def getMovieCount():
    """form user_id, get the number of movies this user has rated so far in their history.
    """
    u_id = session["user_id"] 
    # build separate function to reduce server activity? sending entire object.
    movies = analysis.get_user_movies(u_id)
    count = len(movies)

    return str(count)


@app.route('/update_rate', methods=["POST"])
def update_movie_ratings():
    """collect user's movie ratings, add info to database, reload user page.
    """
    ##############DRY??? refactor into rate??#######
    # must iterate through entire form, since we don't know how many repsonses.
    ratings = request.form
    u_id = session["user_id"]

    for mv, r in ratings.items():
        #convert from string response
        mv = int(mv)
        r = int(r)
        # only for movies rated (not 0, which is default)
        if r != 0:
            # recalibrating 1:5 stars, into -2:2 -- algorithm more accurate.
            r = r - 3
            # call function to handle each rating.
            analysis.process_user_rating(u_id, mv, r)

    return redirect("/user")


@app.route('/update_comic', methods=["POST"])
def update_comic_ratings():
    """collect user's comic ratings, add info to database, reload user page.
    """
    ##############DRY??? refactor into rate??#######
    # must iterate through entire form, since we don't know how many repsonses.
    ratings = request.form
    u_id = session["user_id"]

    for cm, r in ratings.items():
        #convert from string response
        cm = int(cm)
        r = int(r)
        # only for movies rated (not 0, which is default)
        if r != 0:
            # recalibrating 1:5 stars, into -2:2 -- algorithm more accurate.
            r = r - 3
            # call function to handle each rating.
            analysis.process_comic_rating(u_id, cm, r)

    return redirect("/user")

@app.route('/update_user', methods=["POST"])
def update_user():
    """collect user's changes, update database, refresh page
    """
    ##############DRY??? refactor into rate??#######
    # must iterate through entire form, since we don't know how many repsonses.
    user_info = request.form
    u_id = session["user_id"]

    # for cm, r in ratings.items():
    #     #convert from string response
    #     cm = int(cm)
    #     r = int(r)
    #     # only for movies rated (not 0, which is default)
    #     if r != 0:
    #         # recalibrating 1:5 stars, into -2:2 -- algorithm more accurate.
    #         r = r - 3
    #         # call function to handle each rating.
    #         analysis.process_comic_rating(u_id, cm, r)

    return redirect("/user")

############# method post???
@app.route('/comic', methods=["POST", "GET"])
def show_comic():
    """display to the user all the info on their recommended comic.
        will load automatically after they rate movies,
        but they also may want to call again to get a 2nd, 3rd book.
        function makes sure not to recommend same book twice to same user. ;)
    """
    # handle if no user_id??
    u_id = session["user_id"]

    try:
        recommended_comic = analysis.get_comic_rec(u_id)
        return render_template("comics.html", comic=recommended_comic)
    except:
        # if no more comics to recommend :)
        flash("Young grasshopper, you have learned all I can teach you.")
        return redirect("/user")



    return render_template("comics.html", comic=recommended_comic)


@app.route('/user')
def show_user_info():
    """show user nickname, email. reset password button?
        list of their comic recs so far, and edit.
        list of movie ratings so far, and edit.
    """
    # handle if no user_id??
    u_id = session["user_id"]
    user = User.query.get(u_id)
    movies = analysis.get_user_movies(u_id)
    comics = analysis.get_user_comics(u_id)

    return render_template("user.html", user=user, movies=movies, comics=comics)


##################################################

if __name__ == '__main__':

    #helps with debugging
    app.debug = True
    #not caching on reload
    app.jinja_env.auto_reload = app.debug

    connect_to_db(app, os.environ.get("DATABASE_URL"))

    #use debug toolbar
    # DebugToolbarExtension(app)

    PORT = int(os.environ.get("PORT", 5000))

    DEBUG = "NO_DEBUG" not in os.environ

    app.run(host="0.0.0.0", port=PORT, debug=DEBUG)




###################333
#  hide/show in page...
# @app.route('/register', methods=["GET"])
# def display_register_form():

#     return render_template("register_form.html")


# @app.route('/login', methods=["GET"])
# def display_login_form():
#     """show the login form for exisitng users
#     """

#     return render_template("login_form.html")

# before Ajax:
# @app.route('/rate', methods=["POST"])
# def submit_movie_ratings():
#     """collect user's movie ratings, add info to database, send to comics page.
#     """
#     # must iterate through entire form, since we don't know how many repsonses.
#     ratings = request.form
#     u_id = session["user_id"]

#     for mv, r in ratings.items():
#         #convert from string response
#         mv = int(mv)
#         r = int(r)
#         # only for movies rated (not 0, which is default)
#         if r != 0:
#             # recalibrating 1:5 stars, into -2:2 -- algorithm more accurate.
#             r = r - 3
#             # call function to handle each rating.
#             analysis.process_user_rating(u_id, mv, r)

#     return redirect("/comic")