Python 如何根据其他字段设置 Pydantic 字段值

发布于05月22日

from pydantic import BaseModel


class Grafana(BaseModel):
    user: str
    password: str
    host: str
    port: str
    api_key: str | None = None
    GRAFANA_URL = f"http://{user}:{password}@{host}:{port}"
    API_DATASOURCES = "/api/datasources"
    API_KEYS = "/api/auth/keys"

使用PYDANIC，我在GRAFANA_URL中得到了两个未绑定变量的错误消息，分别对应于user、password等.

有没有办法解决这个问题？在常规类中，我只会在__init__方法中创建GRAFANA_URL.对于皮丹蒂克，我不确定该如何继续.

Option A: Use a `@validator`

有关详情，请参阅validators documentation.

from typing import Any

from pydantic import BaseModel, validator


class Model(BaseModel):
    foo: str
    bar: str
    foobar: str = ""

    @validator("foobar", always=True)
    def set_if_empty(cls, v: str, values: dict[str, Any]) -> str:
        if v == "":
            return values["foo"] + values["bar"]
        return v


obj = Model(foo="a", bar="b")
print(obj)  # foo='a' bar='b' foobar='ab'

如此一来，foobar仍然是一个常规的模范领域.

请注意，要使其起作用，foobar必须定义为after、foo和bar.否则你就得用root validator美元了.

Option B: Make it a `@property`

from pydantic import BaseModel


class Model(BaseModel):
    foo: str
    bar: str

    @property
    def foobar(self) -> str:
        return self.foo + self.bar


obj = Model(foo="a", bar="b")
print(obj)         # foo='a' bar='b'
print(obj.foobar)  # ab

则foobar将不再是模型字段，因此不是模式的一部分.这可能与你相关，也可能与你无关.

Option C: Make it a `@computed_field` (Pydantic v2 only!)

Defining computed fields将适用于《平凡的2》.

from pydantic import BaseModel, computed_field


class Model(BaseModel):
    foo: str
    bar: str

    @computed_field
    @property
    def foobar(self) -> str:
        return self.foo + self.bar


obj = Model(foo="a", bar="b")
print(obj)         # foo='a' bar='b' foobar='ab'