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

时间:2016-06-03 10:18:16

标签: python python-2.7 error-handling flask sqlalchemy

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

电子邮件(唯一)

用户名(唯一,主键)

密码

确认密码

问题:

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

但如果我能确切地说出已经使用过的东西,那将会更有用。例如,电子邮件可以正常,但如果使用用户名,则应该告诉“选择其他用户名”。

希望你遇到问题。

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

@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)

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

由于

1 个答案:

答案 0 :(得分:1)

如果您使用的是wtf表单,则可以覆盖NewUserForm的验证方法

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

在模板中,您可以将错误呈现如下:

using System.Windows;