Python Flask catch multiple Integrity errors=用户或电子邮件抛出完整性时的不同错误消息

Question

我有一个简单的注册功能，新用户必须提供以下信息：

电子邮件（唯一）

用户名（唯一，主键）

密码

确认密码

问题：

如果用户填写了所有信息，并且用户名或电子邮件已经在数据库中，我会得到一个完整性错误，这是可以的。我可以尝试捕捉它并显示其中一个已经被使用的信息。在

但如果我能确切地知道已经使用了什么，那就更有用了。例如，电子邮件可以是好的，但如果使用用户名，它应该告诉“选择其他用户名”。在

希望你有问题。在

这就是我使用的代码（其中也有一个验证功能，但它运行良好，并不重要）：

@app.route('/register', methods=['GET', 'POST'])
def register():
form = NewUserForm()
if form.validate_on_submit():
    new_user = User(email=form.email.data, username=form.username.data, password=make_hash(form.password.data))
    try:
        if new_user:
            db_session.add(new_user)
            db_session.commit()
            token = generate_confirmation_token(new_user.email)

            confirm_url = url_for('confirm_email', token=token, _external=True)

            html = render_template('activate.html', confirm_url=confirm_url)

            subject = "Bitte bestätigen Sie Ihre Registrierung"

            send_email(new_user.email, subject, html)

            login_user(new_user)
            flash('Registrierung erfolgreich! Wir haben Ihnen einen Bestätigungsemail auf die, von Ihnen angegebene Adresse zugeschickt. Bestätigen Sie bitte Ihre Registrierung.')
            return redirect(url_for('register'))
        else:
            flash('Neuer Nutzer konnte nicht angelegt werden!')
    except IntegrityError:
        flash('Nutzer oder Email ist bereits vergeben, anderer Nutzername erforderlich')
return render_template('register.html', form=form)

我需要的是一种方法来指定用户或电子邮件是否抛出了完整性错误，并显示适当的消息。在

谢谢

Answer 1

如果您使用的是wtf表单，那么可以重写NewUserForm的validate方法

class NewUserForm(Form):
    username = TextField('Username', [validators.Required()])

    def __init__(self, *args, **kwargs):
        Form.__init__(self, *args, **kwargs)
        self.user = None

    def validate(self):
        rv = Form.validate(self)
        if not rv:
            return False

        user = User.query.filter_by(
            username=self.username.data).first()
        if user is not None:
            self.username.errors.append(' username taken ')
            return False

        self.user = user
        return True

在模板中，可以按如下方式呈现错误：

^{pr2}$