From d50fbde5de0a9ff65c9d3a60f9d908d8775fd697 Mon Sep 17 00:00:00 2001
From: Rich <SiloDS@gmail.com>
Date: Sat, 3 Jul 2021 09:49:22 +0100
Subject: [PATCH] .

---
 app/models.py            | 6 ++++++
 app/routes.py            | 3 ++-
 app/templates/login.html | 1 -
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/app/models.py b/app/models.py
index e008cd5..d514f43 100644
--- a/app/models.py
+++ b/app/models.py
@@ -1,5 +1,6 @@
 from werkzeug.security import generate_password_hash, check_password_hash
 from flask_login import UserMixin
+from app import login
 
 
 class User(UserMixin):
@@ -8,3 +9,8 @@ class User(UserMixin):
 
     def check_password(self, password):
         return check_password_hash(self.password_hash, password)
+
+
+@login.user_loader
+def load_user(id):
+    return 0
diff --git a/app/routes.py b/app/routes.py
index bca8c12..6f4bdca 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -1,7 +1,7 @@
 from flask import render_template, flash, redirect, url_for
 from app import app
 from app.forms import LoginForm
-from flask_login import current_user, login_user
+from flask_login import current_user, login_user, login_required
 from app.models import User
 
 
@@ -22,6 +22,7 @@ def login():
 
 @app.route("/")
 @app.route("/index")
+@login_required
 def index():
     user = {"username": "Miguel"}
     posts = [
diff --git a/app/templates/login.html b/app/templates/login.html
index 8e66c33..098be60 100644
--- a/app/templates/login.html
+++ b/app/templates/login.html
@@ -10,7 +10,6 @@
         {{ form.password.label }}<br>
         {{ form.password(size=32) }}
     </p>
-    <p>{{ form.remember_me() }} {{ form.remember_me.label }}</p>
     <p>{{ form.submit() }}</p>
 </form>
 {% endblock %}
\ No newline at end of file