diff --git a/app/models.py b/app/models.py new file mode 100644 index 0000000..4a3ce90 --- /dev/null +++ b/app/models.py @@ -0,0 +1,14 @@ +from app import db +from flask_login import UserMixin +from werkzeug.security import generate_password_hash, check_password_hash + +class User(UserMixin, db.Model): + id = db.Column(db.Integer, primary_key=True) + username = db.Column(db.String(64), index=True, unique=True) + password_hash = db.Column(db.String(128)) + + def set_password(self, password): + self.password_hash = generate_password_hash(password) + + def check_password(self, password): + return check_password_hash(self.password_hash, password)