mirror of
https://github.com/simple-login/app.git
synced 2024-11-13 07:31:12 +01:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
from flask import render_template, flash, redirect, url_for
|
||
|
|
from flask_login import login_required, current_user
|
||
|
|
from flask_wtf import FlaskForm
|
||
|
|
from wtforms import PasswordField, validators
|
||
|
|
|
||
|
|
from app.dashboard.base import dashboard_bp
|
||
|
|
from app.extensions import db
|
||
|
|
|
||
|
|
|
||
|
|
class LoginForm(FlaskForm):
|
||
|
|
password = PasswordField("Password", validators=[validators.DataRequired()])
|
||
|
|
|
||
|
|
|
||
|
|
@dashboard_bp.route("/fido_cancel", methods=["GET", "POST"])
|
||
|
|
@login_required
|
||
|
|
def fido_cancel():
|
||
|
|
if current_user.fido_uuid is None:
|
||
|
|
flash("You haven't registed a security key", "warning")
|
||
|
|
return redirect(url_for("dashboard.index"))
|
||
|
|
|
||
|
|
password_check_form = LoginForm()
|
||
|
|
|
||
|
|
if password_check_form.validate_on_submit():
|
||
|
|
password = password_check_form.password.data
|
||
|
|
|
||
|
|
if current_user.check_password(password):
|
||
|
|
current_user.fido_pk = None
|
||
|
|
current_user.fido_uuid = None
|
||
|
|
current_user.fido_credential_id = None
|
||
|
|
db.session.commit()
|
||
|
|
flash("We've unlinked your security key.", "success")
|
||
|
|
return redirect(url_for("dashboard.index"))
|
||
|
|
else:
|
||
|
|
flash("Incorrect password", "warning")
|
||
|
|
|
||
|
|
return render_template("dashboard/fido_cancel.html", password_check_form=password_check_form)
|