refactor

1 files changed, 3 insertions, 3 deletions

diff --git a/app/models.py b/app/models.py
index cb90fda..ed9138f 100644
--- a/app/models.py
+++ b/app/models.py
@@ -4,11 +4,11 @@ import datetime
 import jwt
 import time
 
+from flask import current_app
 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
 
@@ -34,7 +34,7 @@ class User(UserMixin, db.Model):
     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"],
+            current_app.config["SECRET_KEY"],
             algorithm="HS256",
         ).decode("utf-8")
 
@@ -42,7 +42,7 @@ class User(UserMixin, db.Model):
     def verify_reset_password_token(token):
         try:
             _id = jwt.decode(
-                token, app.config["SECRET_KEY"], algorithms=["HS256"]
+                token, current_app.config["SECRET_KEY"], algorithms=["HS256"]
             )["reset_password"]
         except:
             return None