diff --git a/mo/web/auth.py b/mo/web/auth.py index e0d3b46b4be7a503506db94be5a9ce85c1ed8984..379fa2df4e9b6f607e0685468941fcd4cb72dfea 100644 --- a/mo/web/auth.py +++ b/mo/web/auth.py @@ -33,7 +33,7 @@ class LoginForm(FlaskForm): reset = wtforms.SubmitField('Zapomenuté heslo') -def login_and_redirect(user: db.User, url: Optional[str] = None): +def login_and_redirect(user: db.User, flash_msg: Optional[str] = None, url: Optional[str] = None): session.clear() session['uid'] = user.user_id if not url: @@ -43,6 +43,8 @@ def login_and_redirect(user: db.User, url: Optional[str] = None): url = url_for('index') else: url = request.script_root + url + if flash_msg: + flash(flash_msg, 'success') return redirect(url) @@ -110,7 +112,7 @@ def incarnate(id): raise werkzeug.exceptions.NotFound() app.logger.info('Login: Uživatel #%s se převtělil na #%s', g.user.user_id, new_user.user_id) - return login_and_redirect(new_user) + return login_and_redirect(new_user, flash_msg='Převtělení proběhlo') @app.route('/user/settings') @@ -179,7 +181,7 @@ def reset(): app.logger.info('Login: Přihlásil se uživatel <%s> po resetování hesla', user.email) db.get_session().commit() flash('Nastavení nového hesla a přihlášení do systému proběhlo úspěšně', 'success') - return login_and_redirect(user) + return login_and_redirect(user, flash_msg='Heslo nastaveno') class RegStatus(Enum):