python cerberus Validator参数校验

最简单用法

# 最简单用法
from cerberus import Validator
schema = {'name1': {'type': 'string'}, 'name2': {'type': 'string'}}
v = Validator(schema)

document = {'name1': 'john doe', 'name2': 0}

if v.validate(document):
    print('data is valid')
else:
    print(v.errors)

输出：

{'name2': ['must be of string type']}

自定义校验方法代码示例： 

# coding=utf-8
from cerberus import Validator


class ParamValidator(Validator):

    def __init__(self, *args, **kwargs):
        self.res = {}
        super(ParamValidator, self).__init__(*args, **kwargs)

    def _validator_username(self, field, value):
        print(" --- in _validator_username ---")
        print("field = {}, value = {}".format(field, value))
        name = value.get('username', None)
        print("name = {}".format(name))
        if not name:
            return self._error('error_info', 'not username')

    def _check_with_gender(self, field, value):
        print("--- in _check_with_gender ---")
        print("field = {} | value = {}".format(field, value))
        self.res.update({"field": field, "value": value})


if __name__ == "__main__":
    # No.1 自定义前缀为:_validator_<validator_name>的校验方法
    document = {'data': {'username': ''}}
    # schema_rule中validator对应的值为username，会自动去找_validator前缀的_validator_username方法
    schema_rule = {"data": {'validator': 'username'}}
    pv = ParamValidator(schema_rule)
    if not pv.validate(document):
        print("pv.document = {}".format(pv.document))
        print("pv.errors = {}".format(pv.errors))
    else:
        print("validator success end ---")

    print('='*20 + '分割线' + '='*20)

    # No.2 - 自定义前缀为_check_with<field>的校验方法
    document = {'name': "1", 'age': 3, 'gender': 'male-'}
    # schema_rule中gender对应的值中check_with字段对应的值为gengder，会自动去找_check_with前缀的_check_with_gender方法执行
    schema_rule = {
        'name': {'type': 'string'},
        'age': {'default': 18, 'type': 'integer'},
        'gender': {'type': "string", "allowed": ["male", "female"], "check_with": "gender"}
    }

    pv = ParamValidator(schema_rule)
    res = pv.validate(document)
    print("res = {}".format(res))
    print("pv.document = {}".format(pv.document))
    print("pv.errors = {}".format(pv.errors))
    print("pv.res = {}".format(pv.res))

输出：

 --- in _validator_username ---
field = data, value = {'username': ''}
name = 
pv.document = {'data': {'username': ''}}
pv.errors = {'error_info': ['not username']}
====================分割线====================
--- in _check_with_gender ---
field = gender | value = male-
res = False
pv.document = {'gender': 'male-', 'age': 3, 'name': '1'}
pv.errors = {'gender': ['unallowed value male-']}
pv.res = {'field': 'gender', 'value': 'male-'}