add errors, email, and reset password

1 files changed, 20 insertions, 0 deletions

diff --git a/app/models.py b/app/models.py
index 0483ff7..cb90fda 100644
--- a/app/models.py
+++ b/app/models.py
@@ -1,11 +1,14 @@
 # -*- coding: utf-8 -*-
 
 import datetime
+import jwt
+import time
 
 from flask_login import UserMixin
 
 from werkzeug.security import generate_password_hash, check_password_hash
 
+from app import app
 from app import db
 from app import login
 
@@ -28,6 +31,23 @@ class User(UserMixin, db.Model):
     def check_password(self, password):
         return check_password_hash(self.password_hash, password)
 
+    def get_reset_password_token(self, expires_in=600):
+        return jwt.encode(
+            {"reset_password": self.id, "exp": time.time() + expires_in},
+            app.config["SECRET_KEY"],
+            algorithm="HS256",
+        ).decode("utf-8")
+
+    @staticmethod
+    def verify_reset_password_token(token):
+        try:
+            _id = jwt.decode(
+                token, app.config["SECRET_KEY"], algorithms=["HS256"]
+            )["reset_password"]
+        except:
+            return None
+        return User.query.get(_id)
+
 
 class Dataset(db.Model):
     id = db.Column(db.Integer, primary_key=True)