remove redundant code

This commit is contained in:
Son NK 2019-12-27 17:36:35 +01:00 committed by Son Nguyen Kim
parent 5b01071bec
commit de0368c20f
1 changed files with 14 additions and 15 deletions

View File

@ -29,24 +29,23 @@ def mfa():
if otp_token_form.validate_on_submit():
totp = pyotp.TOTP(user.otp_secret)
if otp_token_form.validate_on_submit():
token = otp_token_form.token.data
token = otp_token_form.token.data
if totp.verify(token):
del session[MFA_USER_ID]
if totp.verify(token):
del session[MFA_USER_ID]
login_user(user)
flash(f"Welcome back {user.name}!")
# User comes to login page from another page
if next_url:
LOG.debug("redirect user to %s", next_url)
return redirect(next_url)
else:
LOG.debug("redirect user to dashboard")
return redirect(url_for("dashboard.index"))
login_user(user)
flash(f"Welcome back {user.name}!")
# User comes to login page from another page
if next_url:
LOG.debug("redirect user to %s", next_url)
return redirect(next_url)
else:
flash("Incorrect token", "warning")
LOG.debug("redirect user to dashboard")
return redirect(url_for("dashboard.index"))
else:
flash("Incorrect token", "warning")
return render_template("auth/mfa.html", otp_token_form=otp_token_form)