Flask-sqlalchemy、Flask-WTF结合查询的一个疑问?

我是新手。

目的:在用户填写注册表单提交到数据库后,我想接着获取刚才注册成功的用户数据的ID字段。

可以运行的代码片段(A):

if form.validate_on_submit():
    user = Member(username=form.username.data,
                  email=form.email.data,
                  signature=form.signature.data,
                  password=form.password.data)
    db.session.add(user)
    userinfo = Member.query.filter_by(username='010').first()
    flash('添加会员成功。' + str(userinfo.id) + str(form.username.data), 'alert alert-success')

出错的代码片段(B):

if form.validate_on_submit():
    user = Member(username=form.username.data,
                  email=form.email.data,
                  signature=form.signature.data,
                  password=form.password.data)
    db.session.add(user)
    userinfo = Member.query.filter_by(username=str(form.username.data)).first()
    flash('添加会员成功。' + str(userinfo.id) + str(form.username.data), 'alert alert-success')

我的问题是: 代码片段A和B的唯一不同在于「Member.query.filter_by」的查询传递的参数,B运行报错信息为:

AttributeError: 'NoneType' object has no attribute 'id' 我没错的话,应该报错的意思是:form.username.data当前的用户在数据库中查询不到,因此是个「NoneType」也就不存在ID字段了。

但我又认为db.session.add(user)在查询之前运行了,应该已经提交成功了。

我学艺不精,始终无法理解为什么会这样,请大神点拨我一下,十分感谢。

2 comments