Проверка для пользовательского типа в Цербере - PullRequest
Новая
       24

Проверка для пользовательского типа в Цербере

0 голосов
/ 09 мая 2018

Мне очень нравится Цербер, но я не могу понять простой случай в документах. Я хочу, чтобы тип fudge представлял собой строку с ~ в начале. Я просто не могу понять, как это сделать .. 

fudge_type = cerberus.TypeDefinition('fudge_type', (str,), ())

class MyValidator(cerberus.Validator):
    types_mapping = cerberus.Validator.types_mapping.copy()
    types_mapping['fudge'] = fudge_type

    def __init__(self, *args, **kwargs):
        if 'additional_context' in kwargs:
            self.additional_context = kwargs['additional_context']
        super(MyValidator, self).__init__(*args, **kwargs)

    @property
    def additional_context(self):
        self._error(field, "INVALID!")
        return self._config.get('additional_context', 'bar')

    def _validate_type_fudge(self, field, value):
        self._error(field, "INVALID!")
        make_use_of(self.additional_context)

validator = MyValidator()
validator.validate({'name': 'yada'}, {'name': {'type': 'fudge'}})  # => True

Это похоже на простой случай ... поэтому я чувствую, что делаю что-то совершенно не так.

1 Ответ

0 голосов
/ 10 мая 2018

В Цербере 1.2 вы можете добиться этого, как показано ниже: 

import cerberus

SCHEMA = {
    'fudge': {
        'type': 'mytype'
    }
}

class MyValidator(cerberus.Validator):
    def _validate_type_mytype(self, value):
        if value.startswith('~'):
            return True

validator = MyValidator()
validator.validate({'fudge': 'yada'}, SCHEMA)

Так что не нужно взламывать с TypeDefinition.

...