Python Cerberus如何检查动态根密钥

时间:2018-04-10 20:49:02

标签: python python-3.x validation cerberus

我有一个带ID的dict作为我想要验证的根键。换句话说,我要验证的字典的根键是动态的。有没有办法对根密钥运行keyschema?

e.g。 https://repl.it/@crunk1/cerberusrootkeys

import cerberus

v = cerberus.validator.Validator()
schema = {'keyschema': {'type': 'string'}}
d = {'foo': 'bar', 'baz': 'gaz'}

print('I want this to be true.')
print(v.validate(d, schema))

### Output:
# I want this to be true.
# False

我知道我可以做到以下几点:

wrapper = {'nested': d}
schema = {'nested': {'keyschema': {'type': 'string'}}}
v.validate(wrapper, schema)

但我项目的当前结构并不容易实现。

任何解决方案/提示/建议?

1 个答案:

答案 0 :(得分:1)

我设法将一些东西放在一起(https://repl.it/@crunk1/Cerberus-root-types)继承Validator并覆盖validate():

class V(cerberus.Validator):
  def validate(self, document, schema=None, update=False, normalize=True):
    doc = None
    wrapped = False
    if schema is not None:
      root_schema = schema.get('__root__', None)
      wrapped = root_schema is not None
      if wrapped:
        doc = {'__root__': document}
        schema = {'__root__': root_schema}
    elif self.schema is not None:
        root_schema = self.schema.get('__root__', None)
        wrapped = root_schema is not None
        if wrapped:
            doc = {'__root__': document}
            schema = {'__root__': root_schema}

    doc = doc or document
    result = super(V, self).validate(doc, schema, update, normalize)
    if wrapped:
      # Unwrap.
      self.document = self.document['__root__']
      for e in self._errors:
        e.schema_path = tuple(e.schema_path[1:])
        if len(e.document_path) > 1:
          e.document_path = tuple(e.document_path[1:])
    return result

这允许您将根文档视为'type': 'dict''type': 'list'

v = V()
d = {'1': '1', '2': '2'}
schema = {'__root__': {
  'type': 'dict',
  'keyschema': {'coerce': int},
  'valueschema': {'coerce': int},
}}
print(v.validate(d, schema), v.document, v.errors)

l = ['1', '2']
schema = {'__root__': {
  'type': 'list',
  'schema': {'coerce': int},
}}
print(v.validate(l, schema), v.document, v.errors)

l = ['1', 'b']
print(v.validate(l, schema), v.document, v.errors)

输出:

True {1: 1, 2: 2} {}
True [1, 2] {}
False [1, 'b'] {1: ["field '1' cannot be coerced: invalid literal for int() with base 10: 'b'"]}