python-dependency-injector/examples/providers/configuration/configuration_pydantic.py
2024-02-18 17:02:30 -05:00

41 lines
999 B
Python

"""`Configuration` provider values loading example."""
import os
from typing import Annotated
from dependency_injector import containers, providers
from pydantic import Field
from pydantic_settings import BaseSettings
# Emulate environment variables
os.environ["AWS_ACCESS_KEY_ID"] = "KEY"
os.environ["AWS_SECRET_ACCESS_KEY"] = "SECRET"
class AwsSettings(BaseSettings):
access_key_id: str = Field(alias="aws_access_key_id")
secret_access_key: str = Field(alias="aws_secret_access_key")
class Settings(BaseSettings):
aws: AwsSettings = AwsSettings()
optional: str = Field(default="default_value")
class Container(containers.DeclarativeContainer):
config = providers.Configuration()
if __name__ == "__main__":
container = Container()
container.config.from_pydantic(Settings())
assert container.config.aws.access_key_id() == "KEY"
assert container.config.aws.secret_access_key() == "SECRET"
assert container.config.optional() == "default_value"