Python Cerberus, как проверить динамические корневые ключи

Question

У меня есть dict с идентификаторами в качестве корневых ключей, которые я хочу проверить. Другими словами, корневые ключи dict, которые я хочу проверить, являются динамическими. Есть ли способ запустить keychema против корневых ключей?

например, 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)

но нынешняя структура моего проекта не позволяет этого легко сделать.

Какие-либо решения / советы / предложения?

Answer 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'"]}