This commit is contained in:
Rich
2021-07-03 09:49:22 +01:00
parent 3a24e0a9db
commit d50fbde5de
3 changed files with 8 additions and 2 deletions

View File

@@ -1,5 +1,6 @@
from werkzeug.security import generate_password_hash, check_password_hash from werkzeug.security import generate_password_hash, check_password_hash
from flask_login import UserMixin from flask_login import UserMixin
from app import login
class User(UserMixin): class User(UserMixin):
@@ -8,3 +9,8 @@ class User(UserMixin):
def check_password(self, password): def check_password(self, password):
return check_password_hash(self.password_hash, password) return check_password_hash(self.password_hash, password)
@login.user_loader
def load_user(id):
return 0

View File

@@ -1,7 +1,7 @@
from flask import render_template, flash, redirect, url_for from flask import render_template, flash, redirect, url_for
from app import app from app import app
from app.forms import LoginForm 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 from app.models import User
@@ -22,6 +22,7 @@ def login():
@app.route("/") @app.route("/")
@app.route("/index") @app.route("/index")
@login_required
def index(): def index():
user = {"username": "Miguel"} user = {"username": "Miguel"}
posts = [ posts = [

View File

@@ -10,7 +10,6 @@
{{ form.password.label }}<br> {{ form.password.label }}<br>
{{ form.password(size=32) }} {{ form.password(size=32) }}
</p> </p>
<p>{{ form.remember_me() }} {{ form.remember_me.label }}</p>
<p>{{ form.submit() }}</p> <p>{{ form.submit() }}</p>
</form> </form>
{% endblock %} {% endblock %}