132 lines
4.1 KiB
Python
132 lines
4.1 KiB
Python
from flask import Flask
|
|
from flask import render_template
|
|
from flask import session
|
|
from flask import request
|
|
from flask import redirect
|
|
from flask import url_for
|
|
from flask import g
|
|
from flask_migrate import Migrate
|
|
|
|
from flask_babel import Babel, gettext
|
|
|
|
from werkzeug.security import generate_password_hash, check_password_hash
|
|
|
|
from config import secret_key, database_uri
|
|
from model import db, User
|
|
import game as Game
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = secret_key
|
|
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = database_uri
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
|
|
app.config['BABEL_DEFAULT_LOCALE'] = 'en'
|
|
|
|
babel = Babel(app)
|
|
|
|
db.init_app(app)
|
|
|
|
@babel.localeselector
|
|
def get_locale():
|
|
# if a user is logged in, use the locale from the user settings
|
|
user = getattr(g, 'user', None)
|
|
if user is not None:
|
|
return user.locale
|
|
elif session['lang']:
|
|
return session['lang']
|
|
# otherwise try to guess the language from the user accept
|
|
# header the browser transmits. We support fr/en in this
|
|
# example. The best match wins.
|
|
return request.accept_languages.best_match(['fr', 'en'])
|
|
|
|
@babel.timezoneselector
|
|
def get_timezone():
|
|
user = getattr(g, 'user', None)
|
|
if user is not None:
|
|
return user.timezone
|
|
|
|
@app.route('/lang')
|
|
def get_lang():
|
|
return render_template('lang.html')
|
|
|
|
@app.route('/lang', methods=["POST"])
|
|
def set_lang():
|
|
lang = request.form["lang"]
|
|
session['lang'] = lang
|
|
return redirect('/')
|
|
|
|
@app.route("/")
|
|
def home():
|
|
if 'username' in session:
|
|
return render_template("index.html", username=session["username"])
|
|
else:
|
|
return render_template("index.html")
|
|
|
|
@app.route("/signup", methods=["GET", "POST"])
|
|
def signup():
|
|
if request.method == "POST":
|
|
username = request.form['username']
|
|
email = request.form['email']
|
|
password = request.form['password']
|
|
registered_user = User.query.filter_by(user_name=username).first()
|
|
if registered_user is None:
|
|
password_hash = generate_password_hash(password)
|
|
registered_user = User(user_name=username, user_email=email, user_password=password_hash)
|
|
db.session.add(registered_user)
|
|
db.session.commit()
|
|
else:
|
|
return render_template("auth/signup.html", message="Username already used. Try with an other.")
|
|
return redirect(url_for("login"))
|
|
elif request.method == "GET":
|
|
return render_template("auth/signup.html")
|
|
|
|
@app.route("/login", methods=["GET", "POST"])
|
|
def login():
|
|
if request.method == "POST":
|
|
username = request.form['username']
|
|
password = request.form['password']
|
|
user = User.query.filter_by(user_name=username).first()
|
|
if user is None:
|
|
return render_template("auth/login.html", message="No user with this username already registered")
|
|
else:
|
|
password_hash = user.user_password
|
|
if check_password_hash(password_hash, password):
|
|
session["username"] = username
|
|
return redirect(url_for("home"))
|
|
else:
|
|
return render_template("auth/login.html", message="Password incorrect. Try again")
|
|
if request.method == "GET":
|
|
return render_template("auth/login.html")
|
|
|
|
@app.route("/logout")
|
|
def logout():
|
|
# Remove username from the session if it's there
|
|
session.pop("username", None)
|
|
return redirect(url_for("home"))
|
|
|
|
# Game routes
|
|
|
|
@app.route("/game")
|
|
def game():
|
|
return redirect(url_for('new_game'))
|
|
|
|
@app.route("/game/new")
|
|
def new_game():
|
|
question = Game.new_question()
|
|
session["question"] = question
|
|
return render_template("game/question.html", question=question)
|
|
|
|
@app.route("/game/answer", methods=["POST", "GET"])
|
|
def game_answer():
|
|
if request.method == "POST":
|
|
answer = request.form["answer"]
|
|
if answer == session["question"]["species"]:
|
|
message = "You are correct !"
|
|
else:
|
|
message = "You are not correct !"
|
|
return render_template("game/answer.html", message=message)
|
|
elif request.method == "GET":
|
|
return render_template("game/new.html")
|
|
|
|
migrate = Migrate(app, db) |