73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from flask import g, render_template, request, redirect, Blueprint, flash, url_for
|
|
from realms.modules.auth.models import User
|
|
from realms.modules.auth.forms import LoginForm, RegistrationForm
|
|
from realms import config
|
|
|
|
blueprint = Blueprint('auth', __name__, url_prefix=config.RELATIVE_PATH)
|
|
|
|
|
|
@blueprint.route("/logout")
|
|
def logout_page():
|
|
User.logout()
|
|
flash("You are now logged out")
|
|
return redirect(url_for(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(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 config.REGISTRATION_ENABLED:
|
|
flash("Registration is disabled")
|
|
return redirect(url_for(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(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("/") |