31 lines
926 B
Python
31 lines
926 B
Python
|
from flask import session, redirect, url_for
|
||
|
|
from flask_login import login_user
|
||
|
|
|
||
|
|
from app.config import MFA_USER_ID
|
||
|
|
from app.log import LOG
|
||
|
|
|
||
|
|
|
||
|
|
def after_login(user, next_url):
|
||
|
|
"""
|
||
|
|
Redirect to the correct page after login.
|
||
|
|
If user enables MFA: redirect user to MFA page
|
||
|
|
Otherwise redirect to dashboard page if no next_url
|
||
|
|
"""
|
||
|
|
if user.enable_otp:
|
||
|
|
session[MFA_USER_ID] = user.id
|
||
|
|
if next_url:
|
||
|
|
return redirect(url_for("auth.mfa", next_url=next_url))
|
||
|
|
else:
|
||
|
|
return redirect(url_for("auth.mfa"))
|
||
|
|
else:
|
||
|
|
LOG.debug("log user %s in", user)
|
||
|
|
login_user(user)
|
||
|
|
|
||
|
|
# 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"))
|