记录一个flask-login的神坑——login_manager.user_loader究竟为什么要写成User.query.get(int(user_id))

login_manager.user_loader究竟为什么要写成User.query.get(int(user_id))

今天在用flask-login做登录的时候出现了一个问题,就是明明我的账号密码都是对的,但是flask-login死活给我跳回登录页面。然后通过检查返回体我发现我的app是正常工作的,是因为登录校验不对才给我跳回登录页面的。我寻思这不对啊我账号密码都对怎么可能呢。
这时候我就想难道是我在login_user的时候登录的对象是None吗,然后开debug模式看了好几遍,数据库查出来的数据是对的。这就很奇怪了。
后来我把目光放在了回调函数上,就是user_loader上,我先放一下我原来的user_loader函数。

@login_manager.user_loader
def load_user(username):
	return User.query.filter_by(username=username).first()

我之所以写成这样是因为觉得id这个字段没有什么用,因为主要都是用username来查询的。
按理说login_user的对象和user_loader查出来的对象一对比就可以了,但是好像login_required死活给我校验得不对。
然后我就去看login_user的源码了,在这里我发现login_user用到了一个session变量,然后login_user往session里面存了user_id

	user_id = getattr(user, current_app.login_manager.id_attribute)()
    session['user_id'] = user_id
    session['_fresh'] = fresh
    session['_id'] = current_app.login_manager._session_identifier_generator()

唔。。。这和我想的不太一样,讲真你不应该把我整个对象存起来吗。然后我就意识到了,login_required的逻辑其实是将user_id传给回调函数,然后如果回调函数查得出东西,login_required就给通过,不行就扔回去登录页面,然后我就看了login_required的源码

def login_required(func):
    @wraps(func)
    def decorated_view(*args, **kwargs):
        if request.method in EXEMPT_METHODS:
            return func(*args, **kwargs)
        elif current_app.login_manager._login_disabled:
            return func(*args, **kwargs)
        elif not current_user.is_authenticated:
            return current_app.login_manager.unauthorized()
        return func(*args, **kwargs)
    return decorated_view

唔。。。居然是个wraps装饰器,讲真其实可读性挺差的,然后我接着找下去发现逻辑太深了就没继续找了。毕竟要考虑时间成本的,所以为了直接验证想法,我把回调函数改成(其实就是网上出现频率最高的回调函数写法,毕竟你们都是你抄我我抄你的,弄得我查找问题的效率特别特别低)

@login_manager.user_loader
def load_user(user_id):
	return User.query.get(int(user_id))

然后跑一下,就登录成功了。唔。。。。果然是这个问题。

写在最后

在解决这个问题的时候我上百度搜的关键字是"flask-login 登录失败"给我出来一堆一模一样的答案,还有个人说自己把user_loader卸载model文件里导致它没生效,然后flask-login报错,我想说我也是把user_loader写在model里,完全没问题,只是他没把LoginManager的实例引入到model里面来而已。
果然这些问题与其上网找答案,还不如自己源码来的简单

以下是一个简单的 Flask 应用程序,使用 SQLAlchemy 和 Flask-Login 进行用户身份验证: ```python from flask import Flask, render_template, request, redirect, url_for from flask_login import LoginManager, login_user, logout_user, login_required, UserMixin from sqlalchemy import create_engine, Column, Integer, String from sqlalchemy.orm import sessionmaker from werkzeug.security import generate_password_hash, check_password_hash app = Flask(__name__) app.secret_key = 'secret_key' # 创建数据库连接 engine = create_engine('sqlite:///users.db', echo=True) # 创建用户表模型 class User(UserMixin, Base): __tablename__ = 'users' id = Column(Integer, primary_key=True) username = Column(String(50), unique=True, nullable=False) password_hash = Column(String(128), nullable=False) def __repr__(self): return '<User %r>' % self.username # 初始化登录管理器 login_manager = LoginManager() login_manager.init_app(app) # 加载用户的回调函数 @login_manager.user_loader def load_user(user_id): Session = sessionmaker(bind=engine) session = Session() return session.query(User).get(int(user_id)) # 登录页面 @app.route('/login', methods=['GET', 'POST']) def login(): if request.method == 'POST': username = request.form['username'] password = request.form['password'] Session = sessionmaker(bind=engine) session = Session() # 根据用户名查询用户 user = session.query(User).filter_by(username=username).first() if user and check_password_hash(user.password_hash, password): # 登录成功,将用户添加到会话中 login_user(user) return redirect(url_for('index')) else: # 登录失败,显示错误信息 return render_template('login.html', error='用户名或密码错误') return render_template('login.html') # 注销页面 @app.route('/logout') @login_required def logout(): # 从会话中删除用户 logout_user() return redirect(url_for('login')) # 主页 @app.route('/') @login_required def index(): return render_template('index.html') if __name__ == '__main__': app.run(debug=True) ``` 这段代码创建了一个名为 `users.db` 的 SQLite 数据库,其中包含一个名为 `users` 的表,该表包含用户的用户名和密码散列值。用户可以在登录页面中进行身份验证,如果验证通过,则将用户添加到会话中,以便在应用程序的其他部分进行访问。如果用户注销,则会从会话中删除用户。主页需要进行身份验证,因此只有已登录的用户才能访问它。
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值