Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

wtforms raise a validation error after the form is validated

I have a registration form that collects credit card info. The workflow is as follows:

  • User enters registration data and card data through stripe.
  • The form is validated for registration data.
  • If the form is valid, payment is processed.
  • If payment goes through, it's all good, the user is registered and moves on.
  • If the payment fails, i want to be able to raise a validation error on a hidden field of the form. Is that possible?

Here's a the form submission code:

def register():
form = RegistrationForm()
if form.validate_on_submit():

    user = User(
        [...]
    )

    db.session.add(user)

    #Charge
    amount = 10000

    customer = stripe.Customer.create(
        email=job.company_email,
        card=request.form['stripeToken']
    )
    try:

        charge = stripe.Charge.create(
            customer=customer.id,
            amount=amount,
            currency='usd',
            description='Registration payment'
        )
    except StripeError as e:
        ***I want to raise a form validation error here if possible.***

    db.session.commit()
    return redirect(url_for('home'))

return render_template('register.html', form=form)
like image 483
applechief Avatar asked Nov 07 '13 08:11

applechief


2 Answers

I solved it by manually appending errors to the field i wanted.

It looks like that

try:

    [...]
except StripeError as e:
    form.payment.errors.append('the error message')
else:
    db.session.commit()
    return redirect(url_for('home'))
like image 174
applechief Avatar answered Oct 04 '22 17:10

applechief


on your wtform itself you can add a method prefixed with validate_ in order to raise an exception.

class RegistrationForm(Form):
  amount = IntegerField('amount', validators=[Required()])

  validate_unique_name(self, field):
    if field.data > 10000:
      raise ValidationError('too much money')

in my case I used it like so to validate that a user was not in the database already:

class Registration(Form):
  email = StringField('Email', validators=[Required(), Email()]) # field for email
  # ...
  def validate_email(self, field): # here is where the magic is
    if User.query.filter_by(email=field.data).first(): # check if in database
      raise ValidationError("you're already registered")
like image 37
corvid Avatar answered Oct 04 '22 17:10

corvid