mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-11-04 01:47:36 +03:00
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
from pydantic_settings import BaseSettings as PydanticSettings
|
|
|
|
from dependency_injector import providers
|
|
|
|
# Test 1: to check the getattr
|
|
config1: providers.Configuration = providers.Configuration()
|
|
provider1: providers.Provider[Dict[str, Any]] = providers.Factory(dict, a=config1.a)
|
|
|
|
# Test 2: to check the from_*() method
|
|
config2 = providers.Configuration()
|
|
|
|
config2.from_dict({})
|
|
config2.from_value({})
|
|
|
|
config2.from_ini("config.ini")
|
|
config2.from_ini(Path("config.ini"))
|
|
|
|
config2.from_yaml("config.yml")
|
|
config2.from_yaml(Path("config.yml"))
|
|
|
|
config2.from_json("config.json")
|
|
config2.from_json(Path("config.json"))
|
|
|
|
config2.from_env("ENV", "default")
|
|
config2.from_env("ENV", as_=int, default=123)
|
|
config2.from_env("ENV", as_=float, required=True)
|
|
config2.from_env("ENV", as_=lambda env: str(env))
|
|
|
|
config2.from_pydantic(PydanticSettings())
|
|
|
|
# Test 3: to check as_*() methods
|
|
config3 = providers.Configuration()
|
|
int3: providers.Callable[int] = config3.option.as_int()
|
|
float3: providers.Callable[float] = config3.option.as_float()
|
|
int3_custom: providers.Callable[int] = config3.option.as_(int)
|
|
|
|
# Test 4: to check required() method
|
|
config4 = providers.Configuration()
|
|
option4: providers.ConfigurationOption = config4.option.required()
|
|
|
|
# Test 5: to check get/set config files' methods and init arguments
|
|
# Test 5: ini
|
|
config5_ini = providers.Configuration(
|
|
ini_files=["config.ini", Path("config.ini")],
|
|
)
|
|
config5_ini.set_ini_files(["config.ini", Path("config.ini")])
|
|
config5_ini_files: list[str | Path] = config5_ini.get_ini_files()
|
|
|
|
# Test 5: yaml
|
|
config5_yaml = providers.Configuration(
|
|
yaml_files=["config.yml", Path("config.yml")],
|
|
)
|
|
config5_yaml.set_yaml_files(["config.yml", Path("config.yml")])
|
|
config5_yaml_files: list[str | Path] = config5_yaml.get_yaml_files()
|
|
|
|
# Test 5: json
|
|
config5_json = providers.Configuration(
|
|
json_files=["config.json", Path("config.json")],
|
|
)
|
|
config5_json.set_json_files(["config.json", Path("config.json")])
|
|
config5_json_files: list[str | Path] = config5_json.get_json_files()
|
|
|
|
# Test 5: pydantic
|
|
config5_pydantic = providers.Configuration(
|
|
pydantic_settings=[PydanticSettings()],
|
|
)
|
|
config5_pydantic.set_pydantic_settings([PydanticSettings()])
|
|
config5_pydantic_settings: list[PydanticSettings] = (
|
|
config5_pydantic.get_pydantic_settings()
|
|
)
|
|
|
|
# Test 6: to check init arguments
|
|
config6 = providers.Configuration(
|
|
name="config",
|
|
strict=True,
|
|
default={},
|
|
)
|