78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
import functools
|
|
|
|
from flask import Blueprint
|
|
from flask import flash
|
|
from flask import current_app
|
|
from flask import g
|
|
from flask import redirect
|
|
from flask import url_for
|
|
from flask import request
|
|
from flask import session
|
|
from flask import render_template
|
|
from flask_mail import Message
|
|
from werkzeug.exceptions import abort
|
|
|
|
from capsulflask.db import get_model
|
|
|
|
bp = Blueprint("auth", __name__, url_prefix="/auth")
|
|
|
|
def account_required(view):
|
|
"""View decorator that redirects anonymous users to the login page."""
|
|
|
|
@functools.wraps(view)
|
|
def wrapped_view(**kwargs):
|
|
if session.get("account") is None:
|
|
return redirect(url_for("auth.login"))
|
|
|
|
return view(**kwargs)
|
|
|
|
return wrapped_view
|
|
|
|
@bp.route("/login", methods=("GET", "POST"))
|
|
def login():
|
|
if request.method == "POST":
|
|
email = request.form["email"]
|
|
error = None
|
|
|
|
if not email:
|
|
error = "Email is required."
|
|
|
|
if error is None:
|
|
token = get_model().login(email)
|
|
|
|
link = f"{current_app.config['BASE_URL']}/auth/magic/{token}"
|
|
|
|
current_app.config["FLASK_MAIL_INSTANCE"].send(
|
|
Message(
|
|
"Click This Link to Login to Capsul",
|
|
body=f"""
|
|
Navigate to {link} to log into capsul.
|
|
""",
|
|
html=f"""
|
|
Navigate to <a href="{link}">{link}</a> to log into capsul.
|
|
""",
|
|
sender=current_app.config['MAIL_DEFAULT_SENDER'],
|
|
recipients=[email]
|
|
)
|
|
)
|
|
|
|
return render_template("check-your-email.html")
|
|
|
|
flash(error)
|
|
|
|
return render_template("login.html")
|
|
|
|
@bp.route("/magic/<string:token>", methods=("GET", ))
|
|
def magiclink(token):
|
|
email = get_model().consumeToken(token)
|
|
if email is not None:
|
|
session.clear()
|
|
session["account"] = email
|
|
return redirect(url_for("index"))
|
|
else:
|
|
abort(404, f"Token {token} doesn't exist or has already been used.")
|
|
|
|
@bp.route("/logout")
|
|
def logout():
|
|
session.clear()
|
|
return redirect(url_for("index")) |