Создана кнопка выхода

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()

templates/base.html

@@ -35,11 +35,20 @@
                 <h2 style="color:#56a3ff">О нас</h2>
             </a>
         </ul>
+        {% if current_user.is_authenticated %}
+        <a href="/logout"><div style="style=position:absolute; width:148px; height:44px; left:255px; -webkit-border-radius:
+             22px;-moz-border-radius: 22px;border-radius: 22px; border:2px solid #FFFFFF; background-color:#1daff0; top:
+             98px;"><img src="../{{ current_user.photo }}" width="40" height="40" style="border-radius: 50%">
+            <strong style="color: #ffffff">{{ current_user.name }}</strong>
+    </div></a>
+    {% else %}
     <div>
         <a class="btn btn-primary " style="background-color:#41b6ff" type="submit" href="/register"><strong>
             Зарегистрироваться</strong></a>
-            <a class="btn btn-primary" style="background-color:#38aaff" type="submit" href="/login"><strong>Войти</strong></a>
+        <a class="btn btn-primary" style="background-color:#38aaff" type="submit"
+           href="/login"><strong>Войти</strong></a>
     </div>
+    {% endif %}
     </div>
 </nav>
 <!-- Begin page content -->