Pydantic で無効なフィールドを禁止する

2563 ワード

結論から

from pydantic import BaseModel, ValidationError, Extra


class Model(BaseModel, extra=Extra.forbid):
    a: str


try:
    Model(a='spam', b='oh no')
except ValidationError as e:
    print(e)
    """
    1 validation error for Model
    b
      extra fields not permitted (type=value_error.extra)
    """

これにより、追加のフィールドを無効化することができ、タイポを抑制できる。APIに無効な入力が与えられた時にエラーが吐けるので便利。

JSON Schema で言うところの additionalProperty: false

プログラムの健全性を保つ上で、通常、設定すべき項目だと思うが、フィールドが多く、何度も利用するクラスの場合、バリデーションが重くなることがあるので注意。

参考