如何在Pydantic中合并两个字段生成完整数据库URL？

核心思路 #

Pydantic配置类在类定义阶段无法直接引用其他字段的值，必须借助其验证器（validator，v1）或计算字段（computed_field，v2）机制，在实例化过程中完成字段值拼接。同时需注意：配置类中的PROJECT字段要对应.env里的PROJECT_NAME环境变量，需通过Field的env参数做映射。

实现方式 #

方式1：覆盖原DB_URL字段（推荐） #

直接用拼接后的完整数据库地址替换原DB_URL字段，使用时无需额外区分字段。

Pydantic v2 版本

from pydantic_settings import BaseSettings, Field, computed_field

class DummyConfig(BaseSettings):
    # 映射.env中的PROJECT_NAME变量
    PROJECT: str = Field(..., env="PROJECT_NAME")
    # 读取.env中的原始DB_URL，用别名隐藏内部字段
    _raw_db_url: str = Field(..., alias="DB_URL")

    @computed_field
    @property
    def DB_URL(self) -> str:
        # 处理原始URL末尾的斜杠，避免拼接后出现//
        return f"{self._raw_db_url.rstrip('/')}/{self.PROJECT}"

    class Config:
        env_file = ".env"
        populate_by_name = True  # 允许使用别名映射字段

Pydantic v1 版本

from pydantic import BaseSettings, Field, validator

class DummyConfig(BaseSettings):
    PROJECT: str = Field(..., env="PROJECT_NAME")
    DB_URL: str = Field(...)

    @validator("DB_URL")
    def assemble_full_db_url(cls, raw_url, values):
        # 确保PROJECT字段已被解析
        if "PROJECT" in values:
            return f"{raw_url.rstrip('/')}/{values['PROJECT']}"
        return raw_url

    class Config:
        env_file = ".env"

方式2：新增独立字段FULL_DB_URL #

保留原始DB_URL，新增字段存储拼接后的完整地址。

Pydantic v2 版本

from pydantic_settings import BaseSettings, Field, computed_field

class DummyConfig(BaseSettings):
    PROJECT: str = Field(..., env="PROJECT_NAME")
    DB_URL: str = Field(...)

    @computed_field
    @property
    def FULL_DB_URL(self) -> str:
        return f"{self.DB_URL.rstrip('/')}/{self.PROJECT}"

    class Config:
        env_file = ".env"

Pydantic v1 版本

from pydantic import BaseSettings, Field, validator

class DummyConfig(BaseSettings):
    PROJECT: str = Field(..., env="PROJECT_NAME")
    DB_URL: str = Field(...)
    FULL_DB_URL: str = None

    @validator("FULL_DB_URL", pre=True, always=True)
    def assemble_full_db_url(cls, _, values):
        # 确保两个字段都已解析完成
        if "PROJECT" in values and "DB_URL" in values:
            return f"{values['DB_URL'].rstrip('/')}/{values['PROJECT']}"
        return _

    class Config:
        env_file = ".env"

使用示例 #

# 实例化配置类，自动加载.env文件并完成拼接
config = DummyConfig()

# 方式1中直接访问DB_URL
print(config.DB_URL)
# 方式2中访问新增的FULL_DB_URL
# print(config.FULL_DB_URL)

# 输出结果：postgresql+psycopg2://postgres:postgres@postgres:5432/dummy

内容的提问来源于stack exchange，提问作者mehekek