Question

我有一个名为Account的表，其中有两种方法（投资和提款），我在提款方法上遇到麻烦。如果贷记金额大于帐户余额，我希望收到ValidationError消息。

这是我的审判书

def Withdraw(self):
        account = Accounts.query.filter_by(name=self.name).order_by(Account.id.desc()).first()
        if self.credit > account.balance:
            raise ValidationError('There is no enough balance in: '+ str(account.name)+ ' please try to Withdraw another account!')
        else:
            account_balance = float(float(account.balance) - float(self.credit))
            trans = Account(name=account.name, descrip=self.descrip, credit=self.credit,balance=account_balance)
            db.session.add(trans)
            db.session.commit()

我在代码下面得到了这个概念

def validate_email(self, email):
        user = User.query.filter_by(email=email.data).first()
        if user:
            raise ValidationError('This emaill is alrealy exist in our system please choose another email')

我希望足以解释我的问题，请让我获得帮助）：

Answer 1

Flask-WTForms验证是在验证表单时运行的特定（子类）功能。

最好的办法是通过覆盖表单验证功能：

class MyForm(FlaskForm):
    field1 = ..
    field2 = ..
    credit = ..

    def validate(self):
        """Overwrite the Base validation function"""
        rv = FlaskForm.validate(self)
        if not rv:
            return False

        # now custom validation code:
        account = Accounts.query.filter_by(name=self.name).order_by(Account.id.desc()).first()
        if self.credit > account.balance:
            self.credit.errors.append('There is not enough balance...')
            return False

        return True

如果返回True，则可以进行贷记/借记操作，自由地继续应用，否则会产生验证错误。

如果我是您，我将确保我的数据库表具有约束以确保您不能提交错误的数据。数据库的完整性具有更高的优先级，尽管Webform验证是好的，但不一定足以保持数据库的完整性。

如果贷记金额大于帐户余额，wtforms会引发ValidationError

1 个答案: