From 4ef8290c734e83a58dab3a4c3ab7bbe183ee2966 Mon Sep 17 00:00:00 2001 From: Son NK Date: Sat, 6 Jul 2019 17:54:42 +0200 Subject: [PATCH] login and register page: redirect user to dashboard if user is already authenticated --- app/auth/views/login.py | 6 +++++- app/auth/views/register.py | 7 ++++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/app/auth/views/login.py b/app/auth/views/login.py index 2fa59c51..aae1516d 100644 --- a/app/auth/views/login.py +++ b/app/auth/views/login.py @@ -1,5 +1,5 @@ from flask import request, render_template, redirect, url_for -from flask_login import login_user +from flask_login import login_user, current_user from flask_wtf import FlaskForm from wtforms import StringField, validators @@ -15,6 +15,10 @@ class LoginForm(FlaskForm): @auth_bp.route("/login", methods=["GET", "POST"]) def login(): + if current_user.is_authenticated: + LOG.d("user is already authenticated, redirect to dashboard") + return redirect(url_for("dashboard.index")) + form = LoginForm(request.form) if form.validate_on_submit(): diff --git a/app/auth/views/register.py b/app/auth/views/register.py index 1a46ca59..54021d25 100644 --- a/app/auth/views/register.py +++ b/app/auth/views/register.py @@ -1,5 +1,6 @@ import arrow -from flask import request, flash, render_template +from flask import request, flash, render_template, redirect, url_for +from flask_login import current_user from flask_wtf import FlaskForm from wtforms import StringField, validators @@ -23,6 +24,10 @@ class RegisterForm(FlaskForm): @auth_bp.route("/register", methods=["GET", "POST"]) def register(): + if current_user.is_authenticated: + LOG.d("user is already authenticated, redirect to dashboard") + return redirect(url_for("dashboard.index")) + form = RegisterForm(request.form) if form.validate_on_submit():