From e224c9d5f5c8119e6d268e80055da40ba9592ae9 Mon Sep 17 00:00:00 2001
From: mrmur <andreiduvakin@mail.ru>
Date: Mon, 4 Apr 2022 10:07:22 +0500
Subject: [PATCH] =?UTF-8?q?=D0=A1=D0=BE=D0=B7=D0=B4=D0=B0=D0=BD=D0=B0=20?=
 =?UTF-8?q?=D0=BA=D0=BD=D0=BE=D0=BF=D0=BA=D0=B0=20=D0=B2=D1=8B=D1=85=D0=BE?=
 =?UTF-8?q?=D0=B4=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 db/moona_data.db    | Bin 12288 -> 12288 bytes
 main.py             |   9 ++++++++-
 templates/base.html |  19 ++++++++++++++-----
 3 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/db/moona_data.db b/db/moona_data.db
index 37a76001011e3ea882106ef26931e0c8019f4367..3b531a2339ed8353c810f99966b31dc23603cee3 100644
GIT binary patch
delta 189
zcmWN{y^exF6hPsS*o&2gMqJlcxm+(W*!k0$SeaM~Gs7U_3UP2P^$|2ag{ATJEac=X
z&o#V;x0Yev+cU#_h0nUtc{jSv`s2s?eeA5>MVC>T3OoHv?30d<n_`=nxjjGqM_GE<
z?DHfa7KfzZnN&E|+X!86j2=|=X^mODny${1$wDuc?gJ1>a>*mkIgB81h>`CHQh{(J
jQwlr;hJF-tE=Y2hVUtw{%0NgioDl-z0I2f*ST*wt<R3U%

delta 178
zcmZojXh@hK&B!-V#+i|CW5P;)MwiWk3gY~e73GzA7#JA#FsN!uGn(@<OxBXOsV_*%
zPDwMeD$YnWGBvX@GBW@Im56ljawAWZOutZfA5)ja0u@V(6tk2hBXdh*vqVEPlN4if
zOCu9gqm;Bn(-g~OQ$r)mWb;&G^Hd8{0|Qf2vm_$}i$v2zWAijq%fv);6JsL-BO_e{
c6I}!2(&E%2E}$zI1Q;0jpEGRcR`|*f0PR{V<p2Nx

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 @@
                 <h2 style="color:#56a3ff">О нас</h2>
             </a>
         </ul>
-        <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>
-        </div>
+        {% 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>
+    </div>
+    {% endif %}
     </div>
 </nav>
 <!-- Begin page content -->