server.py

import os, json, time, math, datetime
from flask import Flask, render_template, redirect, url_for, request, jsonify,session
from flask.ext.login import (LoginManager, current_user, login_required,
    login_user, logout_user, UserMixin, AnonymousUser, flash,
    confirm_login, fresh_login_required)
from LoginForm import LoginForm
from mongo import User, Deck, Card
from json import JSONEncoder
import uuid
import srs
from random import choice

app = Flask(__name__)

## Debugging
app.debug = (os.environ.get('ENV', 'production') != True)
app.secret_key= "why do I feel awake at 4am?"
app.config.from_object(__name__)

##login Manager setup
login_manager = LoginManager()
login_manager.login_view = "login"
login_manager.login_message = u"Please log in to access this page."
login_manager.refresh_view = "reauth"
login_manager.setup_app(app)

class DbUser(UserMixin):
    """Wraps User object for Flask-Login"""
    def __init__(self, user):
        self._user = user

    def get_id(self):
        return unicode(self._user.username)

    def is_anonymous(self):
        return False

    def is_authenticated(self):
        return True

@login_manager.user_loader
def load_user(user_id):
    user = User.query.filter(User.username == user_id).first()
    if user:
        return DbUser(user)
    else:
        return None

#################
# Routing stuff #
#################

@app.route('/decks/<deckName>/start', methods=['GET'])
def initSRS(deckName):
    user = User.query.filter(User.username == request.args.get('username')).first()
    for deck in user.decks:
        if (unicode(deck.userId) == unicode(deckName)):
            for card in deck.cards:
                #setup session
                session[deck.userId] = [0,0,0,0]
                return jsonify(id=card.uniqueId,front=card.front,back=card.back)
    return jsonify(success=False)

def scoreCard(username, deckName, cardName, score, user):
    for deck in user.decks:
        if (unicode(deck.userId) == unicode(deckName)):
            for card in deck.cards:
                if card.uniqueId == cardName:
                    if score < 3 :
                        card.eFactor = 0
                        card.interval = 0
                    else:
                        card.eFactor = float(card.eFactor) - 0.8 + (0.28*float(score)) - (0.02*float(score)*float(score))
                        if card.eFactor < 1.3:
                            card.eFactor = 1.3
                interval = 1
                if card.eFactor < 3:
                    card.repetition = 1
                if card.repetition == 1:
                    card.interval = 1
                elif card.repetition == 2:
                    card.interval = 6
                elif card.repetition > 2:
                    card.interval = math.round(card.interval*card.eFactor)
                card.lastDone = datetime.datetime.now()
                print card.eFactor
                print card.interval
                card.save()
                user.save()

def grabNextCard(cards):
    print cards
    return choice(cards)

@app.route('/decks/<deckName>/score', methods=['POST'])
def scoreSRS(deckName):
    username = request.form['username']
    deckName = deckName
    cardName = request.form['uniqueId']
    score = request.form['score']
    user = User.query.filter(User.username == unicode(request.form['username'])).first()


    scoreCard(username, deckName, cardName, score, user)

    for deck in user.decks:
        if (unicode(deck.userId) == unicode(deckName)):
            newCard = grabNextCard(deck.cards);
            return jsonify(id=newCard.uniqueId,front=newCard.front,back=newCard.back)
    return jsonify(success=False)

#                return jsonify(id=card.uniqueId,front=card.front,back=card.back)

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        check = User.query.filter(User.username == unicode(request.form['username'])).first()
        if check:
            flash("user already exists")
            return render_template('register.html')
        if request.form['password'] != request.form['confirm_password']:
            flash("passwords do not match")
            return render_template('register.html')
        user = User(username = request.form['username'], password = request.form['password'], decks = [])
        user.save()
        login_user(DbUser(user))
        return redirect('/')
    return render_template('register.html')

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    next = request.args.get("next")
    user = User.query.filter(User.username == unicode(form.username.data), User.password == unicode(form.password.data)).first()
    if user:
        if login_user(DbUser(user)):
            return redirect("/")
    return render_template('login.html')


@app.route('/profile/settings')
@login_required
def show_settings(deckName):
    pass

@app.route('/profile/')
@login_required
def show_profile():
    pass

#@app.route('/decks/<deckName>/<cardName>/delete', methods=['POST'])
#@login_required
#def delete_card(cardName):
#    print request.json
#    user = User.query.filter(User.username == current_user.get_id()).first()
#    for deck in user.decks:
#        if (unicode(deck.userId) == unicode(deckName)):
#            for card in deck:
#                if (card.uniqueId = cardId):
#
#            deck.cards.append(Card(front=request.json['front'], back=request.json['back'],interval=3,eFactor=3.0))
#    return jsonify(success=True)

@app.route('/decks/<deckName>/cards', methods=['POST'])
@login_required
def create_card(deckName):
    print request.json
    user = User.query.filter(User.username == request.json['username']).first()
    card = None
    for deck in user.decks:
        if (unicode(deck.userId) == unicode(deckName)):
            card = srs.makeCard(request.json['front'],request.json['back'])
            deck.cards.append(card)
            user.save()
    return jsonify(id=card.uniqueId, front=card.front, back=card.back)

@app.route('/decks')
@login_required
def decks_index():
    user = User.query.filter(User.username == current_user.get_id()).first()
    decks = map(lambda deck: {'id': deck.userId , "name": deck.name, 'cards': [{'id': card.uniqueId, 'front': card.front, 'back': card.back} for card in deck.cards]}, user.decks)
    return render_template('index.html', decks=json.dumps(decks))

@app.route('/')
def homepage():
    if current_user.is_authenticated():
        return redirect(url_for('decks_index'))
    else:
        return redirect(url_for('register'))

@app.route('/decks', methods=['POST'])
@login_required
def decks_create():
    user = User.query.filter(User.username == request.json['username']).first()
    deck = Deck(name=request.json['name'], userId=str(uuid.uuid1()), cards=[])
    user.decks.append(deck)
    user.save()
    return jsonify(name=deck.name, id=deck.userId, cards=[])

@app.route("/logout")
@login_required
def logout():
    logout_user()
    return redirect("/")


#run everything! move into an __init__.py?
if __name__ == '__main__':
    # Bind to PORT if defined, otherwise default to 5000.
    port = int(os.environ.get('PORT', 5000))
    app.run(host='0.0.0.0', port=port)