我有一个父棉花糖架构“ParentSchema”和 2 个嵌套子架构“ChildSchema1”和“ChildSchema2”。两个嵌套子模式都利用 @validates_schema用于棉花糖现场验证的装饰器。它们看起来像这样:
ParenSchema.py
from marshmallow import Schema, fields
from schemas.childSchema1 import ChildSchema1
from schemas.childSchema2 import ChildSchema2
class ParentSchema(Schema):
child1 = fields.Nested(ChildSchema1)
child2 = fields.Nested(ChildSchema2)
foo = fields.String()
bar = fields.String()
ChildSchema1.py
from marshmallow import Schema, pprint, post_load
from marshmallow import fields, ValidationError, validates, validates_schema
class ChildSchema1(Schema):
field1 = fields.String()
field2 = fields.String()
common_field = fields.String()
@validates("common_field")
def validate_common_field(self, common_field):
try:
# Validation logic
except:
raise ValidationError('common_field is not valid')
ChildSchema2.py
from marshmallow import Schema, pprint, post_load
from marshmallow import fields, ValidationError, validates, validates_schema
class ChildSchema2(Schema):
common_field = fields.String()
field3 = fields.String()
@validates("common_field")
def validate_common_field(self, common_field):
try:
# Exact Same Validation logic as common_field from ChildSchema1
except:
raise ValidationError('common_field is not valid')
鉴于 ChildSchema1 和 ChildSchema2 都有一个同名的字段,具有相同的验证器函数,我很乐意遵循 DRY Principle并将该函数拉出到 ParentSchema.py。
我发现的一个解决方案是创建一个单独的类来托管共享验证函数:
def validate_common_field(self, common_field):
try:
# Exact Same Validation logic as common_field from ChildSchema1
except:
raise ValidationError('common_field is not valid')
并通过删除 @validates 装饰器并在架构字段中使用 validate
参数来使用 ChildSchema1 和 ChildSchema2 中的数据,如下所示。
ChildSchema1.py
from marshmallow import Schema, pprint, post_load
from marshmallow import fields, ValidationError, validates, validates_schema
from validators import *
class ChildSchema1(Schema):
field1 = fields.String()
field2 = fields.String()
common_field = fields.String(validate=validate_common_field)
验证器.py from marshmallow import ValidationError,验证
def validate_common_field(self, common_field):
try:
# Validation logic
except:
raise ValidationError('common_field is not valid')
我是一名优秀的程序员,十分优秀!