realms-wiki/realms/modules/auth/views.py

from flask import current_app, render_template, request, redirect, Blueprint, flash, url_for
from realms.modules.auth.models import User
from realms.modules.auth.forms import LoginForm, RegistrationForm

blueprint = Blueprint('auth', __name__)


@blueprint.route("/logout")
def logout_page():
    User.logout()
    flash("You are now logged out")
    return redirect(url_for(current_app.config['ROOT_ENDPOINT']))


@blueprint.route("/login", methods=['GET', 'POST'])
def login():
    form = LoginForm()

    if request.method == "POST":
        if not form.validate():
            flash('Form invalid', 'warning')
            return redirect(url_for('auth.login'))

        if User.auth(request.form['email'], request.form['password']):
            return redirect(request.args.get("next") or url_for(current_app.config['ROOT_ENDPOINT']))
        else:
            flash('Email or Password Incorrect', 'warning')
            return redirect(url_for('auth.login'))

    return render_template("auth/login.html", form=form)


@blueprint.route("/register", methods=['GET', 'POST'])
def register():

    if not current_app.config['REGISTRATION_ENABLED']:
        flash("Registration is disabled")
        return redirect(url_for(current_app.config['ROOT_ENDPOINT']))

    form = RegistrationForm()

    if request.method == "POST":

        if not form.validate():
            flash('Form invalid', 'warning')
            return redirect(url_for('auth.register'))

        if User.get_by_username(request.form['username']):
            flash('Username is taken', 'warning')
            return redirect(url_for('auth.register'))

        if User.get_by_email(request.form['email']):
            flash('Email is taken', 'warning')
            return redirect(url_for('auth.register'))

        User.create(request.form['username'], request.form['email'], request.form['password'])
        User.auth(request.form['email'], request.form['password'])

        return redirect(request.args.get("next") or url_for(current_app.config['ROOT_ENDPOINT']))

    return render_template("auth/register.html", form=form)


@blueprint.route("/settings", methods=['GET', 'POST'])
def settings():
    return render_template("auth/settings.html")


@blueprint.route("/logout")
def logout():
    User.logout()
    return redirect("/")