重写WTForms框架的validate()校验方法,并捕捉Forms校验异常,返回json格式

重新定义WTforms中的validate方法(增加了def validate_for_api),增加异常处理!


from wtforms import Form, validators, PasswordField
from app.libs.errors.forms_exception import FromsParameterException


class BaseClientForm(Form):
    secretkey = PasswordField(validators=[validators.DataRequired("秘钥不能为空..."),
                                          validators.Length(min=6, max=16, message="密码长度必须为6-16位")])

    def __init__(self,formdata=None, obj=None, prefix='', data=None, meta=None, **kwargs):
        super().__init__(formdata=formdata, obj=obj, prefix=prefix, data=data, meta=meta, **kwargs)

    # 自定义校验器,并处理异常!
    def validate_for_api(self):
        if not super().validate():
            raise FromsParameterException(self.errors)
重写HTTPException异常中的get_body和get_headers方法,返回异常json格式
 
 
from werkzeug.exceptions import HTTPException
from flask import json
from flask import request
 
#重写HTTPException异常中的get_body和get_headers方法,返回异常json格式
class APIException(HTTPException):
 
    # 重写父类的方法
    def get_body(self, environ=None):
        """Get the HTML body."""
        return json.dumps(dict(
            code= self.code,
            name= self.name,
            requert = request.method+">>"+request.url,
            description= self.get_description(environ)
        ))
 
    #重写父类的方法
    def get_headers(self, environ=None):
        """Get a list of headers."""
        return [('Content-Type', 'application/json')]
 
class FromsParameterException(APIException):
    # 重写父类的属性
    code = None
    #如果description没有传入,则这个为默认参数(看HTTPException的构造函数)
    description = "WTForms校验参数错误!"
 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值