Add logout page

4 files changed, 16 insertions, 5 deletions

diff --git a/app/routes.py b/app/routes.py
index ca556ad..7ae1fb1 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -1,6 +1,6 @@
 from app import app, db
 from flask import render_template, redirect, url_for, flash
-from flask_login import login_user
+from flask_login import login_user, login_required, logout_user
 from forms import LoginForm, RegisterForm
 from tables.user import User
 
@@ -9,6 +9,7 @@ def index():
     return render_template('base/index.html')
 
 @app.route("/home", methods=["GET"])
+@login_required
 def home():
     return render_template('base/home.html')
 
@@ -32,6 +33,11 @@ def login():
 
     return render_template('user/login.html', form=form)
 
+@app.route("/logout", methods=['GET', 'POST'])
+def logout():
+    logout_user()
+    return redirect('/')
+
 @app.route("/register", methods=['GET', 'POST'])
 def register():
     form = RegisterForm()
@@ -54,3 +60,4 @@ def register():
         return redirect(url_for('login'))
     else:
         return render_template('user/register.html', form=form)
+
diff --git a/app/templates/base.html b/app/templates/base.html
index 3139dcd..c539baa 100644
--- a/app/templates/base.html
+++ b/app/templates/base.html
@@ -47,19 +47,21 @@
 				</div>
 			</div>
 
-			<!-- TODO: change based on login state -->
 			<div class="navbar-end">
 				<div class="navbar-item">
 					<div class="buttons">
-						<a class="button is-primary" href="register"><strong>Sign up</strong></a>
-						<a class="button is-light" href="login">Log in</a>
+						{% if current_user.is_authenticated %}
+							<a class="button is-light" href="/logout">Log Out</a>
+						{% else %}
+							<a class="button is-primary" href="/register"><strong>Sign Up</strong></a>
+							<a class="button is-light" href="/login">Log In</a>
+						{% endif %}
 					</div>
 				</div>
 			</div>
 		</div>
 	</nav>
 
-	<!-- Actual content -->
 	{% block content %}{% endblock %}
 
 	<!-- TODO: footer -->
diff --git a/app/templates/base/home.html b/app/templates/base/home.html
index 357a399..14dd441 100644
--- a/app/templates/base/home.html
+++ b/app/templates/base/home.html
@@ -1,4 +1,5 @@
 {% extends 'base.html' %}
 
 {% block content %}
+	User Homepage
 {% endblock %}
diff --git a/app/templates/base/index.html b/app/templates/base/index.html
index 357a399..3e91518 100644
--- a/app/templates/base/index.html
+++ b/app/templates/base/index.html
@@ -1,4 +1,5 @@
 {% extends 'base.html' %}
 
 {% block content %}
+	Index Page
 {% endblock %}