cerberus:验证可选字段至少出现一次

时间:2018-07-27 17:31:54

标签: python validation cerberus

我正在使用cerberus验证数据。我的一个字段是可选的-不需要为每个项目都出现。但是,必须在整个数据数组中至少填充一次密钥。

例如,假设我要验证密钥'c'出现在我的数据列表中的至少一个字典中:

from cerberus import Validator

has_c = {'data': [{'a': 1, 'b': 2}, {'b': 2}, {'c': 3}]}
no_c = {'data': [{'a': 1, 'b': 2}, {'a': 1}]}

schema = {'data':
          {'type': 'list',
           'schema': {
               'type': 'dict',
               'schema': {
                   'a': {'required': True},
                   'b': {'required': True},
                   'c': {'required': False, 'at_least_one': True}
               }
           }
           }
          }

class MyValidator(Validator) # Some fancy code...
....

v = MyValidator()

v.validate(has_c, schema) # Passes
v.validate(no_c, schema) # Fails

这似乎在cerberus之外可行,但我想尽可能地将该方法保留在我的验证器中。

1 个答案:

答案 0 :(得分:1)

如果您希望该方法位于Validator子类中,那么您将想要创建一个自定义规则,就像您想的那样。

from cerberus import Validator

test_with_c = {'data': [{'a': 1, 'b': 2}, {'b': 2}, {'c': 3}]}
test_with_no_c = {'data': [{'a': 1, 'b': 2}, {'a': 1}]}

class MyValidator(Validator):
    def _validate_has_c(self, has_c, field, value):
        seen_c = False
        for v in value:
            if "c" in v:
                seen_c = True
        if has_c and not seen_c:
            self._error(field, "Must contain a 'c' key")

schema = {
    "data": {
        "type": "list",
        "has_c": True
    }
}

v = MyValidator(schema)

print(v(test_with_c), v.errors)
print(v(test_with_no_c), v.errors)

运行此操作将产生您想要的在一个元素中寻找c键的结果。运行该代码会产生

True {}
False {'data': ["Must contain a 'c' key"]}