Question

目前，我正在使用棉花糖模式来验证请求，我有一个列表，我需要验证它的内容。

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

checks字段是一个列表，它仅应包含以下特定值["booking", "reservation", "flight"]

Answer 1

如果您要检查列表中只有该三个元素的顺序，请使用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

Answer 2

除了Jerome的答案，我还发现，如果您需要做一些需要更多逻辑的事情，那么您可以这样做：

def validate_check(check: str):
  return check in ["booking", "reservation", "flight"]

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

或使用lambda：

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

验证棉花糖中的列表

2 个答案: