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

Question

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

电子邮件(唯一)

用户名(唯一，主键)

密码

确认密码

问题:

如果用户填写了所有信息，并且用户名或电子邮件已在数据库中，我会收到完整性错误，这没关系。我可以 try catch 它并显示其中之一已被使用的信息。

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

希望您能解决问题。

这就是我使用的代码(其中也已经有验证功能，但它工作正常并且并不重要):

@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的验证方法

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 %}