python - AttributeError: 'SelectQuery' 对象没有属性 'is_active'

Question

我正在尝试按照 Flask Mega Tutorial 学习 Peewee ORM 与 Flask 的结合。 .在 part 5 of the tutorial我使用 OpenID 创建登录名。在克服了一堆障碍之后，我现在在下面一行粘贴的函数中得到一个 AttributeError:login_user(user, remember = remember_me)。

@oid.after_login
def after_login(resp):
    if resp.email is None or resp.email == "":
        flash('Invalid login. Please try again.')
        return redirect(url_for('login'))
    user = User.select().where(User.email == resp.email)    
    if user.wrapped_count() == 0:
        nickname = resp.nickname
        if nickname is None or nickname == "":
            nickname = resp.email.split('@')[0]
        user = User(nickname = nickname, email = resp.email, role = models.ROLE_USER)
        user.save()
    remember_me = False
    if 'remember_me' in session:
        remember_me = session['remember_me']
        session.pop('remember_me', None)
    login_user(user, remember = remember_me)
    return redirect(request.args.get('next') or url_for('index'))

is_active 在我的用户模型中找到如下:

class User(db.Model):
    nickname = TextField()
    email = TextField()
    role = IntegerField(default = ROLE_USER)

    def is_authenticated(self):
        return True

    def is_active(self):
        return True

    def is_anonymous(self):
        return False

    def get_id(self):
        return self.id

    def __repr__(self):
        return '<User %r>' % (self.nickname)

虽然我不知道我在这里做错了什么。有人可以帮助我解决我在这里做错的事情吗？

欢迎所有提示!

Answer 1

如错误所示，user = User.select().where(User.email == resp.email) 返回一个 SelectQuery，而不是一个实例用户。您需要包含一个额外的方法调用来实际获取记录，类似于 .first()。 first 将返回 User 或 None 的实例。

这将允许您稍微调整您的代码:

user = User.select().where(User.email == resp.email).first()
if not user:  # or if user is None:
    nickname = resp.nickname
    ...