BaseSettings是否支持无需重写构造函数添加任意输入参数？

你当前的代码问题在于，MyConfigs(err_if_not_found)是在类定义阶段执行的，此时用的是err_if_not_found字段的默认值False，无法获取到实例化时传入的True参数。要实现通过MyConfig(err_if_not_found=True)传递参数给MyConfigs，同时不重写BaseSettings的构造函数，可以利用Pydantic v2的customise_sources特性，自定义配置源的加载逻辑。

方案一：基于DotEnv文件加载（如果MyConfigs返回env文件路径） #

如果你的MyConfigs(err_if_not_found).configs返回的是环境配置文件的路径（比如.env文件路径），可以通过修改默认的DotEnv配置源来实现：

from pydantic import BaseSettings
from pydantic_settings import SettingsConfigDict, PydanticBaseSettingsSource
from pydantic_settings.sources import DotEnvSettingsSource

# 假设你的自定义配置加载类是这样的
class MyConfigs:
    def __init__(self, err_if_not_found: bool):
        self.err_if_not_found = err_if_not_found
    
    @property
    def configs(self):
        # 根据err_if_not_found返回对应的env文件路径
        return "path/to/your/config.env" if self.err_if_not_found else "path/to/default.env"

class MyConfig(BaseSettings):
    err_if_not_found: bool = False
    # 你的其他配置字段
    app_name: str = "default_app"
    
    model_config = SettingsConfigDict(
        env_prefix="MY_APP",
    )

    @classmethod
    def customise_sources(
        cls,
        settings_cls: type[BaseSettings],
        init_settings: PydanticBaseSettingsSource,
        env_settings: PydanticBaseSettingsSource,
        dotenv_settings: PydanticBaseSettingsSource,
        file_secret_settings: PydanticBaseSettingsSource,
    ) -> tuple[PydanticBaseSettingsSource, ...]:
        # 从实例化传入的参数中获取err_if_not_found值（没有则用默认值）
        init_data = init_settings.init_kwargs
        err_val = init_data.get("err_if_not_found", cls.model_fields["err_if_not_found"].default)
        
        # 创建自定义的DotEnv配置源，使用MyConfigs生成的文件路径
        custom_dotenv = DotEnvSettingsSource(
            settings_cls,
            env_file=MyConfigs(err_val).configs,
            env_prefix=cls.model_config["env_prefix"],
        )
        
        # 保留原有配置源的优先级：初始化参数 > 自定义DotEnv > 系统环境变量 > 文件密钥
        return (
            init_settings,
            custom_dotenv,
            env_settings,
            file_secret_settings,
        )

# 使用方式
config = MyConfig(err_if_not_found=True)
print(config.app_name)  # 会加载MyConfigs(True)对应的env文件中的MY_APP_APP_NAME值

方案二：自定义配置源（如果MyConfigs返回配置字典） #

如果MyConfigs(err_if_not_found).configs直接返回配置字典，你可以自定义一个配置源来加载这些配置：

from pydantic import BaseSettings, ModelField
from pydantic_settings import SettingsConfigDict, PydanticBaseSettingsSource

class MyConfigs:
    def __init__(self, err_if_not_found: bool):
        self.err_if_not_found = err_if_not_found
    
    @property
    def configs(self):
        # 根据err_if_not_found返回对应的配置字典
        if self.err_if_not_found:
            return {"app_name": "custom_app"}
        else:
            return {"app_name": "default_app"}

# 自定义配置源
class MyCustomConfigSource(PydanticBaseSettingsSource):
    def __init__(self, settings_cls: type[BaseSettings], err_if_not_found: bool):
        super().__init__(settings_cls)
        self.err_configs = MyConfigs(err_if_not_found).configs

    def get_field_value(self, field: ModelField, field_name: str) -> tuple[object, str, bool]:
        # 从自定义配置中获取字段值
        value = self.err_configs.get(field_name)
        # 返回（值，来源描述，是否找到值）
        return value, "custom config source", value is not None

class MyConfig(BaseSettings):
    err_if_not_found: bool = False
    app_name: str = "fallback_app"
    
    model_config = SettingsConfigDict(
        env_prefix="MY_APP",
    )

    @classmethod
    def customise_sources(
        cls,
        settings_cls: type[BaseSettings],
        init_settings: PydanticBaseSettingsSource,
        env_settings: PydanticBaseSettingsSource,
        dotenv_settings: PydanticBaseSettingsSource,
        file_secret_settings: PydanticBaseSettingsSource,
    ) -> tuple[PydanticBaseSettingsSource, ...]:
        # 获取实例化时传入的err_if_not_found值
        init_data = init_settings.init_kwargs
        err_val = init_data.get("err_if_not_found", cls.model_fields["err_if_not_found"].default)
        
        # 创建自定义配置源
        custom_source = MyCustomConfigSource(settings_cls, err_val)
        
        # 配置源优先级：初始化参数 > 自定义配置 > DotEnv > 系统环境变量 > 文件密钥
        return (
            init_settings,
            custom_source,
            dotenv_settings,
            env_settings,
            file_secret_settings,
        )

# 使用方式
config = MyConfig(err_if_not_found=True)
print(config.app_name)  # 输出 "custom_app"

核心原理 #

customise_sources是Pydantic提供的类方法，允许你自定义配置源的加载顺序和逻辑。通过它，你可以在配置加载阶段获取到实例化时传入的参数，动态生成对应的配置源，完全保留BaseSettings的原生特性，不需要重写构造函数。

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