棉花糖允许保留一个值,但不清除该值

时间:2018-01-19 09:02:52

标签: python marshmallow

我想在用户群中轻柔地强制执行用户设置以前不需要的值,但现在是。

这是设置

从数据库中提取的用户将根据数据库绑定的marshmallow架构进行验证,这允许使用None值。

country = fields.String(validate=OneOf(COUNTRY_CODES), allow_none=True)

新用户将根据禁止无的marshmallow架构进行验证。

country = fields.String(validate=OneOf(COUNTRY_CODES), allow_none=False)

编辑过的用户会根据另一个marshmallow架构进行验证,这是一个棘手的部分。

我希望没有设置字段,如果之前没有设置字段,但是一旦设置,你就不应该删除它。

如何在棉花糖中指定?

1 个答案:

答案 0 :(得分:1)

我可能误解了你的问题,但在我看来,用户是一个新用户或一个现有用户,是一个在架构外部的状态(因为架构是无状态的)所以,它需要通过到验证器。在类似的情况下,我做过类似的事情:

from marshmallow import fields, validates_schema, Schema
from marshmallow.validate import OneOf
from marshmallow.exceptions import ValidationError

class Foo(Schema):
    country = fields.String(validate=OneOf(('A', 'B')), missing=None)

    # Some sort of hint that indicates if it's a new 
    # user or not. Could be record creation date, etc.
    new_user = fields.Boolean(missing=False)

    @validates_schema
    def check_country(self, instance):
        if instance['new_user'] and instance['country'] is None:
            raise ValidationError("count can not be none")

观察:

schema = Foo()

schema.load({})    
>>> UnmarshalResult(data={'new_user': False, 'country': None}, errors={})

schema.load({'new_user':True})    
>>> UnmarshalResult(data={'new_user': True, 'country': None}, errors={'_schema': ['count can not be none']})

schema.load({'new_user':True, 'country': 'A'})
>>> UnmarshalResult(data={'new_user': True, 'country': 'A'}, errors={})

schema.load({'new_user':False})
>>> UnmarshalResult(data={'new_user': False, 'country': None}, errors={})