如何在Marshmallow中针对创建与更新操作验证必填属性？

针对创建时校验必填字段、更新时跳过部分字段必填校验的场景，这里有几种实用的处理方式：

方法一：使用load方法的partial参数 #

这是最直接的方式，无需修改原Schema，在更新操作时通过partial参数指定允许缺失的字段，或直接传partial=True灵活跳过部分校验。

代码示例： #

class RuleSchema(Schema):
    name = fields.String(required=True)
    code = fields.String(required=True)
    priority_from = fields.Int(required=True)
    priority_to = fields.Int(required=True)

schema = RuleSchema()

# 创建操作：正常校验所有必填字段
schema.load({"name": "rule-01", "code":"a000", "priority_from":1, "priority_to":10})

# 更新操作1：指定允许缺失的字段
schema.load({"name": "rule-01", "code":"a000", "priority_to":11}, partial=["priority_from"])

# 更新操作2：允许所有非核心字段缺失（name和code仍为必填）
schema.load({"name": "rule-01", "code":"a000", "priority_to":11}, partial=True)

注意：partial=True不会忽略required=True的字段，仅允许未传入的字段跳过校验；若需让某些字段在更新时完全可选，需明确指定partial的字段列表。

方法二：创建继承的Schema类 #

如果业务逻辑清晰，直接拆分出两个Schema分别对应创建和更新场景，代码可读性和维护性更强。

代码示例： #

from marshmallow import Schema, fields

class BaseRuleSchema(Schema):
    # 公共必填字段，创建/更新都需校验
    name = fields.String(required=True)
    code = fields.String(required=True)

class CreateRuleSchema(BaseRuleSchema):
    # 创建时额外的必填字段
    priority_from = fields.Int(required=True)
    priority_to = fields.Int(required=True)

class UpdateRuleSchema(BaseRuleSchema):
    # 更新时可选的字段
    priority_from = fields.Int(required=False)
    priority_to = fields.Int(required=False)

# 创建操作
create_schema = CreateRuleSchema()
create_schema.load({"name": "rule-01", "code":"a000", "priority_from":1, "priority_to":10})

# 更新操作
update_schema = UpdateRuleSchema()
update_schema.load({"name": "rule-01", "code":"a000", "priority_to":11})

这种方式把不同场景的校验规则明确分离，适合字段较多、规则差异大的业务场景。

方法三：利用Schema上下文自定义校验逻辑 #

如果需要更灵活的动态校验控制，可借助Schema的context参数，在自定义验证函数中根据操作类型调整规则。

代码示例： #

from marshmallow import Schema, fields, ValidationError

def validate_priority(value, context):
    operation = context.get("operation")
    if operation == "create" and value is None:
        raise ValidationError("该字段在创建时为必填项")

class RuleSchema(Schema):
    name = fields.String(required=True)
    code = fields.String(required=True)
    priority_from = fields.Int(validate=lambda x: validate_priority(x, self.context))
    priority_to = fields.Int(validate=lambda x: validate_priority(x, self.context))

# 创建操作：传入context标记操作类型
create_schema = RuleSchema(context={"operation": "create"})
create_schema.load({"name": "rule-01", "code":"a000", "priority_from":1, "priority_to":10})

# 更新操作：标记为update，跳过必填校验
update_schema = RuleSchema(context={"operation": "update"})
update_schema.load({"name": "rule-01", "code":"a000", "priority_to":11})

这种方式适合复杂场景，比如同一个字段在不同操作下有多种校验规则，不限于必填与否的区分。

内容的提问来源于stack exchange，提问作者Mahamutha M