Python 将二进制验证器压缩为类批注

发布于01月23日

我想用Python语言创建一个数据 struct ，将Pydatics的验证器压缩到一行参数中:

也就是说.从这个开始

from pydantic import BaseModel, validator


class MyClass(BaseModel):
    a: str
    b: str
    c: str
    
    @validator('b')
    def calculate_b(cls, v, values):
        return values['a'] + v
      
    @validator('c')
    def calculate_c(cls, v, values):
        return values['a'] + values['b'] + v

对(类似)这件事

class MyClass:
    a: str
    b: str = calculate_b(...)
    c: str = calculate_c(...)

推荐答案

_{我的僵尸版本是:2.5.0}

Note:您使用的@validator在V2中已弃用，并将在V3中删除.如果你愿意的话，你应该用field_validator来代替.

您可以使用Annotated+AfterValidator(实际上field_validator的默认模式是"after"):

from typing import Annotated

from pydantic import AfterValidator, BaseModel


class MyClass(BaseModel):
    a: str
    b: Annotated[str, AfterValidator(lambda v, values: values.data["a"] + v)]
    c: Annotated[str, AfterValidator(lambda v, values: values.data["a"] + values.data["b"] + v)]


obj = MyClass(a="a", b="b", c="c")
print(obj)
# a='a' b='ab' c='aabc'

如果注释变得很长，您可以在类外将其用作别名.但我觉得这没问题.

这values个是ValidationInfo类型，为了访问数据，可以使用.data属性.