server.py

from flask import Flask, jsonify, render_template, request, flash, redirect, session, g, url_for
from flask_debugtoolbar import DebugToolbarExtension
from model import connect_to_db, db, User, Following, Rating, Game, System, Genre, UserSystem, GameGenre, GameSystem
from jinja2 import StrictUndefined
from datetime import datetime
import os, math
from igdb_api_python.igdb import igdb
from flask_sqlalchemy import SQLAlchemy
from helper_functions import get_all_similarities, filt_all, pearson, predict, standard_deviation
from random import choice
from sqlalchemy import func

app = Flask(__name__)
app.secret_key = os.environ.get("FLASK_SECRET_KEY")
app.jinja_env.undefined = StrictUndefined

#*****************************************************************************#
@app.route("/")
def welcome():
    """Show welcome page"""

    return render_template("welcome.html")


@app.before_request
def before_everything():
    """Stuff to do before every route"""

    if request.path == '/homepage':
        g.homepage = True
    else:
        g.homepage = False

@app.route("/homepage")
def homepage():
    """Show homepage"""
    
    user_id = session["user_id"]
    system_info = UserSystem.query.filter_by(user_id=user_id).all()
    genre_info = Genre.query.all()
    search_term = request.args.get('search_term')
    print search_term
    return render_template("homepage.html", system_info=system_info, 
                                            genre_info=genre_info,
                                            search_term=search_term)

@app.route("/homepage_search")
def homepage_with_search():
    """Show homepage with search results"""

    search_term = request.args.get("search")
    
    return redirect(url_for('homepage', search_term=search_term))


@app.route("/search.json", methods=["POST"])
def search_for_games():
    """Handle search, return results so JS can render homepage using AJAX"""

    game = request.form.get("game")
    # json_game_info = {"games": []}

    game_info = search(game)

    # game_info = Game.query.filter(Game.name.ilike("%"+game+"%")).all()

    # if game_info:
    #     for game in game_info:
    #         json_game_info["games"].append({"name": game.name, "game_id": game.game_id})
    # else:
    #     json_game_info["games"].append({"name": None, "game_id": 0})

    # return jsonify(json_game_info)
    return jsonify(game_info)


def search(game):
    """Handle search, return results so JS can render homepage using AJAX"""

    json_game_info = {"games": []}
    game_info = Game.query.filter(Game.name.ilike("%"+game+"%")).all()

    if game_info:
        for game in game_info:
            json_game_info["games"].append({"name": game.name, "game_id": game.game_id})
    else:
        json_game_info["games"].append({"name": None, "game_id": 0})

    return json_game_info


@app.route("/user/<user_id>")
def user_profile(user_id):
    """Show user profile"""

    user_info = User.query.filter_by(user_id=user_id).first()
    account_created = user_info.account_created
    account_created = str(account_created)[:11]
    system_info = UserSystem.query.filter_by(user_id=user_id).all()
    rating_info = (db.session.query(Game.name, Rating.score, Game.game_id)
                             .join(Rating).filter(Rating.user_id==user_id)
                             .all())
    num_games =  (db.session.query(func.count(Rating.user_id))
                            .filter(Rating.user_id == user_id)
                            .first())
    num_games = int(num_games[0])

    return render_template("user_profile.html", user_info=user_info,
                                                system_info=system_info,
                                                rating_info=rating_info,
                                                account_created=account_created,
                                                user_id=user_id,
                                                num_games=num_games)

@app.route("/userprofile.json")
def user_genre_data():
    """Return data about user interests"""

    user_id = int(request.args.get("user_id"))

    user_ratings = Rating.query.filter_by(user_id=user_id).all()
    user_rated_games = [(rating.game, rating.score) for rating in user_ratings]
 
    gamegenres_matrix = [(game.gamegenres, score) for game, score in user_rated_games]

    label_data = []
    for gamegenres, score in gamegenres_matrix: 
        for gamegenre in gamegenres:
            label_data.append((gamegenre.genre.name, score))
    label_data = label_data

    genre_score = {}
    for genre, score in label_data:
        if not genre in genre_score:
            genre_score[genre] = [score]
        else:
            genre_score[genre].append(score)

    results = {"labels": [], "weights": []}
    for genre, scores in genre_score.items():
        results["labels"].append(genre)
        results["weights"].append(sum(scores)/float(len(scores)))

    return jsonify(results)


@app.route("/recommendation")
def get_recommendation():
    """Show a game the user would probably like"""

    games_info = []
    # count how many games the user has rated
    user_id = session["user_id"]
    num_games =  (db.session.query(func.count(Rating.user_id))
                            .filter(Rating.user_id == user_id)
                            .first())
    num_games = int(num_games[0])

    ratings = db.session.query(Rating.score).filter_by(user_id=user_id).all()
    ratings = [rating[0] for rating in ratings]
    # calculate standard deviation
    deviation = standard_deviation(ratings)
    # if this user rates everything the same, send them this error page
    if deviation == 0:
        games_info.append("no")
        return render_template("recommendation.html", games_info=games_info)
    # if you've rated more than 10 games and you don't rate everything the same:
    if num_games >= 10 and deviation != 0:
        recommendations = []

        systems = request.args.getlist("systems")
        genres = request.args.getlist("genres")

        # if systems aren't checked, use all of current users' systems
        if systems == []:
            # get users' system ids
            user_system = UserSystem.query.filter_by(user_id=user_id).all()
            for i in user_system:
                systems.append(i.system_id)
        # if genres aren't checked, use all genres
        if genres == []:
            all_genres = Genre.query.all()
            for i in all_genres:
                genres.append(i.genre_id)

        filt_games = (db.session.query(Game.game_id)
                                .join(GameSystem, GameGenre)
                                .filter(GameSystem.system_id.in_(systems),
                                        GameGenre.genre_id.in_(genres))
                                .all())

        filt_games = [game[0] for game in filt_games]

        for game in filt_games:
            try:
                sims = get_all_similarities(game, users, games, user_id)
            except:
                games_info.append(num_games) 
                return render_template("recommendation.html", games_info=games_info)
            raw_pred = predict(sims, users, game)
            # print "Printing raw_pred"
            # print raw_pred

            if raw_pred > 3.0:
                recommendation = (game, raw_pred)
                recommendations.append(recommendation)
        for rec in recommendations:
            one_game_info = Game.query.filter(Game.game_id==rec[0]).first()
            name = one_game_info.name
            game_id = one_game_info.game_id
            if not one_game_info.covers:
                cover = None
            else:
                cover = one_game_info.covers[0].cover_url

            if isinstance(rec[1], (str, unicode)) == False:
                percentage = round(rec[1] * 20, 2)
            else:
                percentage = rec[1]
            games_info.append({"name": name, 
                               "percentage": percentage,
                               "game_id": game_id,
                               "cover": cover})
    else:
        games_info.append(num_games)

    return render_template("recommendation.html", games_info=games_info)


@app.route("/games/<game_id>")
def show_game_details(game_id):

    user_id = session["user_id"]
    rating = Rating.query.filter(Rating.user_id==user_id, Rating.game_id==game_id).first()
    game_info = Game.query.filter_by(game_id=game_id).first()

    return render_template("game_details.html", game_info=game_info, rating=rating)


@app.route("/gamerating.json", methods=["POST"])
def game_rating():
    """Update or add game rating to database"""

    rating = request.form.get("rating")
    game_id = request.form.get("game_id")
    current_user = session["user_id"]

    existing_rating = Rating.query.filter_by(user_id=current_user, game_id=game_id).first()

    if existing_rating is None:
        new_rating = Rating(user_id=current_user, game_id=game_id,score=rating)

        db.session.add(new_rating)
        db.session.commit()

    else:
        existing_rating.score = rating
        db.session.commit()

    return jsonify({"rating": rating})

@app.route("/register", methods=["GET"])
def register():
    """Show registration form"""

    platform_list = []

    platforms = System.query.all()

    for platform in platforms:
        platform_list.append(platform.name)

    return render_template("registration.html", platform_list=platform_list)


@app.route("/register", methods=["POST"])
def register_process():
    """Get information from registration form."""

    username = request.form.get("username")
    email = request.form.get("email")
    password = request.form.get("password")
    systems = request.form.getlist("systems")
    account_created = datetime.now()

    existing_username = User.query.filter_by(username=username).first()
    existing_email = User.query.filter_by(email=email).first()

    # check if the username is in use
    if existing_username is None and existing_email is None:
        #check if the email is in use
        new_user = User(username=username, email=email, password=password, 
                        account_created=account_created)

        db.session.add(new_user)
        db.session.commit()
        get_user_rating(games)

        for system in systems:
            # add each system to the database for the specific user
            system_id = db.session.query(System.system_id).filter(System.name==system).first()
            new_user_system = UserSystem(user_id=new_user.user_id, system_id=system_id)

            db.session.add(new_user_system)
            db.session.commit()

        flash("Successfully registered " + username + "!")
        return redirect("/")

    else:
        flash("Username or email already in use")
        # TODO probably handle this in AJAX on the form and be more specific
        # as to whether it was the username or email that failed

    return redirect("/")


@app.route("/add_systems", methods=["GET"])
def add_systems():
    """Show page that allows a user to add more systems to their profile"""

    user_id = session["user_id"]
    owned_systems_list = []
    owned_systems = [us.system.name 
                     for us in UserSystem.query.filter_by(user_id=user_id).all()]
    all_systems = [sys.name for sys in  System.query.all()]
    not_owned_systems_list = sorted(set(all_systems) - set(owned_systems))


    return render_template("add_systems.html", not_owned_systems_list=not_owned_systems_list)


@app.route("/add_systems", methods=["POST"])
def process_add_systems():
    """Add new systems to db"""

    user_id = session["user_id"]
    systems = request.form.getlist("systems")

    for system in systems:
        system_id = db.session.query(System.system_id).filter(System.name==system).first()
        new_system = UserSystem(user_id=user_id, system_id=system_id)

        db.session.add(new_system)
        db.session.commit()

    flash("Successfully added systems")

    return redirect("/user/" + str(user_id))

@app.route("/login", methods=["GET"])
def attempt_login():
    """Show login page"""

    return render_template("welcome.html")


@app.route("/login", methods=["POST"])
def login():
    """Attempt to log the user in"""

    email = request.form.get("email")
    password = request.form.get("password")

    existing_email = User.query.filter_by(email=email).first()

    if existing_email is not None and existing_email.password == password:
        # add user to session
        session["user_id"] = existing_email.user_id

        flash("Successfully logged in!")
        return redirect('/homepage')

    elif existing_email is None:
        flash("Incorrect email.")
        return redirect('/')
    else:
        flash("Incorrect password.")
        return redirect('/')

@app.route("/logout")
def do_logout():
    """Log user out."""

    flash("Goodbye!")
    session["user_id"] = ""

    return redirect("/")


def get_user_rating(games):
    """Get all ratings for all users"""

    # userrating = {}

    # for game in games:
    #     gamerating = Rating.query.filter_by(game_id=game, user_id=user.user_id).first()
    #     if gamerating:
    #         userrating[game] = gamerating.score
    #     else:
    #         userrating[game] = 0

    #     users[user.user_id] = userrating


    # makes a dictionary with every game_id having a value of zero
    users = {}

    for rating in Rating.query.all():
        if rating.user_id not in users:
            users[rating.user_id] = { game_id : 0 for game_id in games}
        users[rating.user_id][rating.game_id] = rating.score

    return users

#*****************************************************************************#

if __name__ == "__main__":

    from flask import Flask

    app.debug = True
    app.jinja_env.auto_reload = app.debug  # make sure templates, etc. are not cached in debug mode

    connect_to_db(app)
    print "Connected to DB."

    games = set(game.game_id for game in db.session.query(Game.game_id).all())

    print "Loaded games"

    # users = {}
    # print "Printing all users"
    # print User.query.all()
    # for user in User.query.all():
    #     users = get_user_rating(users, user, games)
    users = get_user_rating(games)
    print "Loaded users"
    # from pprint import pprint
    # # pprint(users)
    # print "Printing games"
    # print games


    app.run(port=5000, host='0.0.0.0')