You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.2 KiB
77 lines
2.2 KiB
# -*- coding: utf-8 -*-
|
|
"""Public section, including homepage and signup."""
|
|
from flask import (
|
|
Blueprint,
|
|
current_app,
|
|
flash,
|
|
redirect,
|
|
render_template,
|
|
request,
|
|
url_for,
|
|
)
|
|
from flask_login import login_required, login_user, logout_user
|
|
|
|
from harambee_led_bord.extensions import login_manager
|
|
from harambee_led_bord.public.forms import LoginForm
|
|
from harambee_led_bord.user.forms import RegisterForm
|
|
from harambee_led_bord.user.models import User
|
|
from harambee_led_bord.utils import flash_errors
|
|
|
|
blueprint = Blueprint("public", __name__, static_folder="../static")
|
|
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
"""Load user by ID."""
|
|
return User.get_by_id(int(user_id))
|
|
|
|
|
|
@blueprint.route("/", methods=["GET", "POST"])
|
|
def home():
|
|
"""Home page."""
|
|
form = LoginForm(request.form)
|
|
current_app.logger.info("Hello from the home page!")
|
|
# Handle logging in
|
|
if request.method == "POST":
|
|
if form.validate_on_submit():
|
|
login_user(form.user)
|
|
flash("You are logged in.", "success")
|
|
redirect_url = request.args.get("next") or url_for("user.members")
|
|
return redirect(redirect_url)
|
|
else:
|
|
flash_errors(form)
|
|
return render_template("public/home.html", form=form)
|
|
|
|
|
|
@blueprint.route("/logout/")
|
|
@login_required
|
|
def logout():
|
|
"""Logout."""
|
|
logout_user()
|
|
flash("You are logged out.", "info")
|
|
return redirect(url_for("public.home"))
|
|
|
|
|
|
@blueprint.route("/register/", methods=["GET", "POST"])
|
|
def register():
|
|
"""Register new user."""
|
|
form = RegisterForm(request.form)
|
|
if form.validate_on_submit():
|
|
User.create(
|
|
username=form.username.data,
|
|
email=form.email.data,
|
|
password=form.password.data,
|
|
active=True,
|
|
)
|
|
flash("Thank you for registering. You can now log in.", "success")
|
|
return redirect(url_for("public.home"))
|
|
else:
|
|
flash_errors(form)
|
|
return render_template("public/register.html", form=form)
|
|
|
|
|
|
@blueprint.route("/about/")
|
|
def about():
|
|
"""About page."""
|
|
form = LoginForm(request.form)
|
|
return render_template("public/about.html", form=form)
|
|
|