Python Flask 捕获多个完整性错误 = 如果用户或电子邮件抛出完整性错误，则会出现不同的错误消息

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

在模板中，您可以将错误呈现如下：

{% for field in form.errors %}
{% for error in form.errors[field] %}
    <div class="alert alert-error">
        <strong>Error!</strong> {{error}}
    </div>
{% endfor %}
{% endfor %}