diff --git a/db/moona_data.db b/db/moona_data.db index 37a7600..3b531a2 100644 Binary files a/db/moona_data.db and b/db/moona_data.db differ diff --git a/main.py b/main.py index 9364cd9..d7627ab 100644 --- a/main.py +++ b/main.py @@ -1,7 +1,7 @@ from random import randint from flask import Flask, render_template -from flask_login import LoginManager, login_user +from flask_login import LoginManager, login_user, logout_user, login_required from werkzeug.utils import redirect from data import db_session @@ -41,6 +41,13 @@ def main_page(): return render_template('base.html', title='moona') +@app.route('/logout') +@login_required +def logout(): + logout_user() + return redirect("/") + + @app.route('/login', methods=['GET', 'POST']) def login(): form = LoginForm() diff --git a/templates/base.html b/templates/base.html index 0329686..babf251 100644 --- a/templates/base.html +++ b/templates/base.html @@ -35,11 +35,20 @@

О нас

-
- - Зарегистрироваться - Войти -
+ {% if current_user.is_authenticated %} +
+ {{ current_user.name }} +
+ {% else %} +
+ + Зарегистрироваться + Войти +
+ {% endif %}