如何使用 Enum 限制 pydantic 的选择答案

【问题标题】：How to limit choices for pydantic using Enum如何使用 Enum 限制 pydantic 的选择
【发布时间】：2021-07-04 12:46:20
【问题描述】：

我得到了下一个枚举选项：

class ModeEnum(str, Enum):
""" mode """

    map = "map"
    cluster = "cluster"
    region = "region"

这个枚举用于两个 Pydantic 数据结构。在一个数据结构中，我需要所有 Enum 选项。在其他数据结构中，我需要排除region。如果我为此使用自定义验证并尝试输入一些 other 值，标准验证错误消息会通知，允许的值都是三个。

那么在这种情况下最好的决定是什么？

附：我在ModeEnum 中使用map 变量。不好吗？我无法想象它可以覆盖内置 map 对象的情况，但是，可以吗？

【问题讨论】：

【解决方案1】：

这有点小技巧，但如果你用pre=True 标记你的验证器，你应该能够强制它首先运行，然后你可以使用允许的值抛出一个自定义错误。

【讨论】：