棉花糖中的验证列表

问题描述

目前，我正在使用MarshmAllow模式来验证请求， 我有一个列表，我需要验证其中的内容。

class PostValidationSchema(Schema):
    checks = fields.List(
        fields.String(required=True)
    )

Checks字段是一个列表，它应该只包含这些特定值["booking", "reservation", "flight"]

---

解决方案

如果要检查列表中只按该顺序包含这三个元素，请使用Equal验证器。

from marshmallow import Schema, fields, validate


class PostValidationSchema(Schema):
    checks = fields.List(
        fields.String(required=True),
        validate=validate.Equal(["booking", "reservation", "flight"])
    )

schema = PostValidationSchema()

schema.load({"checks": ["booking", "reservation", "flight"]})  # OK
schema.load({"checks": ["booking", "reservation"]})  # ValidationError

如果列表可以包含任意数量的元素，并且这些元素只能是这三个特定值之一，则使用OneOf验证器。

from marshmallow import Schema, fields, validate


class PostValidationSchema(Schema):
    checks = fields.List(
        fields.String(
            required=True,
            validate=validate.OneOf(["booking", "reservation", "flight"])
        ),
    )

schema = PostValidationSchema()

schema.load({"checks": ["booking", "reservation", "flight"]})  # OK
schema.load({"checks": ["booking", "reservation"]})  # OK
schema.load({"checks": ["booking", "dummy"]})  # ValidationError