Flask中mongodb實現flask_login保持登入

標籤：

最近在學習Flask，使用flask-login時，一直無法完成保持登入的狀態，網上的例子都是使用SQLAlchemy，但是我用的是mongodb。

網上的例子使用SQLAlchemy時，定義User類時是這麼寫的：

class User(UserMixin,db.Model):    __tablename__ = ‘users‘    id = db.Column(db.Integer, primary_key=True)    username = db.Column(db.String(64), unique=True, index=True)    role_id = db.Column(db.Integer, db.ForeignKey(‘roles.id‘))    def __repr__(self):        return ‘<User %r>‘ % self.username

然後在登陸時用如下語句進行匹配查詢：

       user = User.query.filter_by(username=form.name.data).first()

我設定了下斷點，發現這個語句返回的是這樣的：

從圖上能看出來這個語句將整個User類返回了

而我的定義是這樣的：

class User():    def __init__(self, username, email, password):        self.username = username        self.email = email        self.password_hash = self.set_password(password)        self.db = MongoClient().blog.User    def new_user(self):        collection = {            ‘name‘: self.username,            ‘email‘: self.email,            ‘password‘: self.password_hash        }        self.db.insert(collection)    @property    def password(self, password):        raise AttributeError(‘password is not a readable attribute‘)    def set_password(self, password):        return generate_password_hash(password)

尋找匹配語句：

        user = MongoClient().blog.User.find_one({‘email‘: form.email.data})

返回的是一個json格式文檔：

那麼flask-login是怎麼實現在不同頁面保持登入的呢？

 

 Hello,{% if current_user.is_authenticated %} {{ current_user.name }}

flask-login是通過load_user將一個帳號匯入，然後再通過is_active和is_anonymous等驗證顯示。

我們必須在mongodb返回的資訊中添加這幾個bool值，才能完成驗證操作。

我的解決方案是另建一個新類，從mongodb尋找並匯出資訊，然後重新建立一個新的類，然後將新建立的類傳給load_user。

class temp(UserMixin):    is_active = True    is_anonymous = False    is_authenticated = True    def __init__(self, id, username, email, password):        self.id = str(id)        self.name = username        self.email = email        self.password_hash = password    def get_id(self):        return self.id    def __repr__(self):        return self.username

@login_manager.user_loaderdef load_user(user_id):    user = MongoClient().blog.User.find_one({‘_id‘: ObjectId(user_id)})    return temp(id=user.get(‘_id‘), username=user.get(‘name‘), email=user.get(‘email‘),                password=user.get(‘password‘))

 

Flask中mongodb實現flask_login保持登入