Using multiple BaseSettings with different prefixes about pydantic-settings HOT 6 CLOSED

import os
from typing import Any


from pydantic_settings import BaseSettings, EnvSettingsSource, PydanticBaseSettingsSource, SettingsConfigDict
from pydantic.fields import FieldInfo


class CustomEnvSettingsSource(EnvSettingsSource):
    def get_field_value(self, field: FieldInfo, field_name: str) -> tuple[Any, str, bool]:
        if field_name in KubectlSecrets.model_fields:
            env_name = (KubectlSecrets.model_config['env_prefix'] + field_name).lower()
            return self.env_vars.get(env_name), field_name, False
        else:
            return super().get_field_value(field, field_name)


class KubectlSecrets(BaseSettings):
    credentials: str
    google_auth_key: str

    model_config = SettingsConfigDict(env_prefix = "K8S_SECRET_")
        
    @classmethod
    def settings_customise_sources(
        cls,
        settings_cls: type[BaseSettings],
        init_settings: PydanticBaseSettingsSource,
        env_settings: PydanticBaseSettingsSource,
        dotenv_settings: PydanticBaseSettingsSource,
        file_secret_settings: PydanticBaseSettingsSource,
    ) -> tuple[PydanticBaseSettingsSource, ...]:
        return init_settings, CustomEnvSettingsSource(settings_cls), dotenv_settings, file_secret_settings


class Settings(KubectlSecrets):
    environment: str = "development"
    redis_db: str = "0"

    model_config = SettingsConfigDict(env_prefix = "")




os.environ["K8S_SECRET_CREDENTIALS"] = "dummy"
os.environ["K8S_SECRET_GOOGLE_AUTH_KEY"] = "dummy_key"
os.environ["ENVIRONMENT"] = "prod"
os.environ["REDIS_DB"] = "1"

print(Settings())

import os

from pydantic_settings import BaseSettings, SettingsConfigDict

os.environ["K8S_SECRET_CREDENTIALS"] = "dummy"
os.environ["K8S_SECRET_GOOGLE_AUTH_KEY"] = "dummy_key"
os.environ["ENVIRONMENT"] = "prod"
os.environ["REDIS_DB"] = "1"

class KubectlSecrets(BaseSettings):
    credentials: str
    google_auth_key: str

    model_config = SettingsConfigDict(env_prefix = "K8S_SECRET_")


class Settings(BaseSettings):
    kubectl_secrets: KubectlSecrets = KubectlSecrets()
    environment: str = "development"
    redis_db: str = "0"

    model_config = SettingsConfigDict(env_prefix = "")

print(Settings())