时间戳

首页 > 解决方案 > 验证棉花糖中的列表

python - 验证棉花糖中的列表

问题描述

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

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

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

标签： pythonpython-3.6marshmallow

解决方案

如果您的意思是检查列表中只有这三个元素的顺序，那么使用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

推荐阅读