mirror of
https://github.com/mgeeky/decode-spam-headers.git
synced 2026-02-22 13:33:30 +01:00
MAESTRO: add backend settings config
This commit is contained in:
0
backend/app/core/__init__.py
Normal file
0
backend/app/core/__init__.py
Normal file
64
backend/app/core/config.py
Normal file
64
backend/app/core/config.py
Normal file
@@ -0,0 +1,64 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from functools import lru_cache
|
||||
from typing import Any
|
||||
|
||||
from pydantic import Field, field_validator
|
||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
|
||||
|
||||
class Settings(BaseSettings):
|
||||
"""Application configuration loaded from environment variables.
|
||||
|
||||
Environment variables are prefixed with `WHA_` by default.
|
||||
"""
|
||||
|
||||
model_config = SettingsConfigDict(env_prefix="WHA_", case_sensitive=False)
|
||||
|
||||
cors_origins: list[str] = Field(
|
||||
default_factory=lambda: ["http://localhost:3000"],
|
||||
description="Allowed CORS origins.",
|
||||
)
|
||||
rate_limit_requests: int = Field(
|
||||
default=60,
|
||||
ge=1,
|
||||
description="Max requests per window for rate limiting.",
|
||||
)
|
||||
rate_limit_window_seconds: int = Field(
|
||||
default=60,
|
||||
ge=1,
|
||||
description="Rate limit window in seconds.",
|
||||
)
|
||||
analysis_timeout_seconds: int = Field(
|
||||
default=30,
|
||||
ge=1,
|
||||
description="Hard timeout for analysis in seconds.",
|
||||
)
|
||||
debug: bool = Field(default=False, description="Enable debug mode.")
|
||||
|
||||
@field_validator("cors_origins", mode="before")
|
||||
@classmethod
|
||||
def parse_cors_origins(cls, value: Any) -> list[str]:
|
||||
if value is None:
|
||||
return ["http://localhost:3000"]
|
||||
if isinstance(value, list):
|
||||
return [str(item).strip() for item in value if str(item).strip()]
|
||||
if isinstance(value, str):
|
||||
text = value.strip()
|
||||
if not text:
|
||||
return []
|
||||
if text.startswith("["):
|
||||
try:
|
||||
parsed = json.loads(text)
|
||||
if isinstance(parsed, list):
|
||||
return [str(item).strip() for item in parsed if str(item).strip()]
|
||||
except json.JSONDecodeError:
|
||||
pass
|
||||
return [item.strip() for item in text.split(",") if item.strip()]
|
||||
return [str(value).strip()]
|
||||
|
||||
|
||||
@lru_cache
|
||||
def get_settings() -> Settings:
|
||||
return Settings()
|
||||
@@ -10,6 +10,7 @@ readme = "../README.md"
|
||||
requires-python = ">=3.13"
|
||||
dependencies = [
|
||||
"fastapi>=0.110",
|
||||
"pydantic-settings>=2.2",
|
||||
"uvicorn[standard]>=0.27",
|
||||
]
|
||||
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
fastapi>=0.110
|
||||
pydantic-settings>=2.2
|
||||
uvicorn[standard]>=0.27
|
||||
pytest>=8.0
|
||||
pytest-asyncio>=0.23
|
||||
|
||||
29
backend/tests/test_config.py
Normal file
29
backend/tests/test_config.py
Normal file
@@ -0,0 +1,29 @@
|
||||
from app.core.config import Settings
|
||||
|
||||
|
||||
def test_settings_defaults() -> None:
|
||||
settings = Settings()
|
||||
assert settings.analysis_timeout_seconds == 30
|
||||
assert settings.debug is False
|
||||
assert settings.rate_limit_requests == 60
|
||||
assert settings.rate_limit_window_seconds == 60
|
||||
assert settings.cors_origins == ["http://localhost:3000"]
|
||||
|
||||
|
||||
def test_settings_env_override(monkeypatch) -> None:
|
||||
monkeypatch.setenv(
|
||||
"WHA_CORS_ORIGINS",
|
||||
'["https://example.com", "http://localhost:3000"]',
|
||||
)
|
||||
monkeypatch.setenv("WHA_RATE_LIMIT_REQUESTS", "10")
|
||||
monkeypatch.setenv("WHA_RATE_LIMIT_WINDOW_SECONDS", "120")
|
||||
monkeypatch.setenv("WHA_ANALYSIS_TIMEOUT_SECONDS", "45")
|
||||
monkeypatch.setenv("WHA_DEBUG", "true")
|
||||
|
||||
settings = Settings()
|
||||
|
||||
assert settings.cors_origins == ["https://example.com", "http://localhost:3000"]
|
||||
assert settings.rate_limit_requests == 10
|
||||
assert settings.rate_limit_window_seconds == 120
|
||||
assert settings.analysis_timeout_seconds == 45
|
||||
assert settings.debug is True
|
||||
Reference in New Issue
Block a user