Flask-login当前用户消失在页面刷新/重定向上

时间:2019-08-22 19:01:11

标签: flask-login flask-pymongo

我有一个登录路线:

@bp.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        guest = Guest(**mongo.db.guests.find_one({'username': form.username.data}))
        if guest is None or not guest.check_password(form.password.data):
            flash('Invalid username or password')
            return redirect(url_for('auth.login'))
        login_user(guest, remember=form.remember_me.data)
        print(current_user)

        flash(f'Logged in {guest.name} successfully')

        next_url = request.args.get('next')
        if not next_url or url_parse(next_url).netloc != '':
            next_url = url_for('main.index')
        return redirect(next_url)

    return render_template('login.html', form=form)

当用户成功登录后,它会闪烁并正确地重定向用户。但是,在目标页面(main.index或profile.view)上,用户尚未登录。

索引路由/ html

@bp.route('/')
def index():
    return render_template('index.html')

{% block content %}
    <h2>Home page</h2>
    <h1>Hi, {{ current_user.username }}!</h1>
    <p><a href={{ url_for('auth.register') }}>Register</a></p>
    <p><a href={{ url_for('auth.login') }}>Sign in</a></p>
    <p><a href={{ url_for('main.index') }}>Home page</a> (accessible to anyone)</p>
    <p><a href={{ url_for('profile.view') }}>Profile View</a> (login required)</p>
    <p><a href={{ url_for('profile.edit') }}>Profile Edit</a> (login required)</p>
{% endblock %}

重定向到profile.view时

@bp.route('/view')
@login_required
def view():
    return 'This is the profile view'

它将点击@login_required装饰器,然后重定向回登录名。

我正在使用Flask-login(duh)和Flask-Pymongo作为数据库。

为什么烧瓶登录当前用户在页面刷新/重定向时消失了?谢谢!

1 个答案:

答案 0 :(得分:0)

我的@login_manager.user_loader方法不正确。

已更改:

@login_manager.user_loader
def load_user(user_id):
    from flaskr.models import Guest
    guest = mongo.db.guests.find_one({'_id': user_id})
    if guest is None:
        return None
    return Guest(**guest)

收件人:

@login_manager.user_loader
def load_user(user_id):
    from flaskr.models import Guest
    from flask_pymongo import ObjectId
    guest = mongo.db.guests.find_one({'_id': ObjectId(user_id)})
    if guest is None:
        return None
    return Guest(**guest)

因为user_id不是Flask_Pymongo.ObjectId,所以数据库搜索始终失败。