Skip to content
Snippets Groups Projects
Commit 242f7cac authored by Martin Mareš's avatar Martin Mareš
Browse files

Při přihlášení uživatele lze zobrazit flash message

Pozor, nelze to udělat před voláním login_and_redirect(),
protože to zresetuje session.
parent 45841254
Branches
No related tags found
1 merge request!86Registrace
...@@ -33,7 +33,7 @@ class LoginForm(FlaskForm): ...@@ -33,7 +33,7 @@ class LoginForm(FlaskForm):
reset = wtforms.SubmitField('Zapomenuté heslo') 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.clear()
session['uid'] = user.user_id session['uid'] = user.user_id
if not url: if not url:
...@@ -43,6 +43,8 @@ def login_and_redirect(user: db.User, url: Optional[str] = None): ...@@ -43,6 +43,8 @@ def login_and_redirect(user: db.User, url: Optional[str] = None):
url = url_for('index') url = url_for('index')
else: else:
url = request.script_root + url url = request.script_root + url
if flash_msg:
flash(flash_msg, 'success')
return redirect(url) return redirect(url)
...@@ -110,7 +112,7 @@ def incarnate(id): ...@@ -110,7 +112,7 @@ def incarnate(id):
raise werkzeug.exceptions.NotFound() 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) 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') @app.route('/user/settings')
...@@ -179,7 +181,7 @@ def reset(): ...@@ -179,7 +181,7 @@ def reset():
app.logger.info('Login: Přihlásil se uživatel <%s> po resetování hesla', user.email) app.logger.info('Login: Přihlásil se uživatel <%s> po resetování hesla', user.email)
db.get_session().commit() db.get_session().commit()
flash('Nastavení nového hesla a přihlášení do systému proběhlo úspěšně', 'success') 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): class RegStatus(Enum):
... ...
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment