从超级模式继承“排除”元参数(marshmallow)

时间:2017-12-06 08:03:32

标签: marshmallow

我有一个对象层次结构和一个与它们对应的模式层次结构。此层次结构的中间级别的模式排除特定的继承字段。我希望从它继承的模式将“继承”这个排除,但如果他们在Meta类中添加自己的排除字段,情况似乎并非如此:

from marshmallow import fields
from marshmallow.schema import Schema


class AncestorSchema(Schema):
    a = fields.Str()
    b = fields.Str()


class IntermediateSchema(AncestorSchema):
    c = fields.Str()

    class Meta:
        exclude = ('b',)


class FinalSchema(IntermediateSchema):
    d = fields.Str()

    class Meta:
        exclude = ('c',)


value = dict(
    a="Field A",
    b="Field B",
    c="Field C",
    d="Field D"
)

print(IntermediateSchema().dump(value).data)

>>> {'c': 'Field C', 'a': 'Field A'}

print(FinalSchema().dump(value).data)

>>> {'d': 'Field D', 'a': 'Field A', 'b': 'Field B'}

在上面的示例中,FinalSchema继承自IntermediateSchema(不包括字段b),并在其自己的c类中排除字段Meta。预期的行为是生成的模式将排除bc,但实际上它仅排除c

当然可以手动将superschema的排除字段包含在继承模式的排除字段中,但这不是继承点,此外,它很麻烦。

我想知道是否可以以优雅的方式实现所需的行为,或者模式继承的当前行为是否实际上是一个错误。

检查marshmallow的源代码表明,至少部分支持来自superschemas的元类的数据继承(即ordered元选项值继承自superschemas。)

2 个答案:

答案 0 :(得分:2)

您还需要指定Meta类的基类。您还需要使用某种反射来从基类中获取值并附加到它。

from marshmallow import fields
from marshmallow.schema import Schema


class AncestorSchema(Schema):
    a = fields.Str()
    b = fields.Str()


class IntermediateSchema(AncestorSchema):
    c = fields.Str()

    class Meta:
        exclude = ('b',)


class FinalSchema(IntermediateSchema):
    d = fields.Str()

    class Meta(IntermediateSchema.Meta):
        exclude = self.__bases__[0].exclude + ('c',)

答案 1 :(得分:0)

当前接受的答案无效。未定义“自我”。我找到了一个行之有效的解决方案。

from marshmallow import fields
from marshmallow.schema import Schema


class AncestorSchema(Schema):
    a = fields.Str()
    b = fields.Str()


class IntermediateSchema(AncestorSchema):
    c = fields.Str()

    class Meta:
        exclude = ('b',)


class FinalSchema(IntermediateSchema):
    d = fields.Str()

    def __init__(self, *args, **kwargs):
        self.opts.exclude += ('c',)
        super().__init__(*args, **kwargs)