5

我正在尝试按照Flask Mega Tutorial结合 Flask 来学习 Peewee ORM 。在本教程的第 5 部分中，我使用 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)

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

欢迎所有提示！

4

1 回答 1

12

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

这将允许您稍微调整代码：

user = User.select().where(User.email == resp.email).first()
if not user:  # or if user is None:
    nickname = resp.nickname
    ...
于 2014-01-15T16:17:31.840 回答