ovdashboard/api/ovdashboard_api/settings.py

"""
Configuration definition.

Converts per-run (environment) variables and config files into the
"python world" using `pydantic`.

Pydantic models might have convenience methods attached.
"""

from typing import Any, Optional

from pydantic import BaseModel, BaseSettings, root_validator


class DavSettings(BaseModel):
    """
    Connection to a DAV server.
    """

    protocol: Optional[str] = None
    host: Optional[str] = None
    username: Optional[str] = None
    password: Optional[str] = None
    path: Optional[str] = None

    @property
    def url(self) -> str:
        """
        Combined DAV URL.
        """

        return f"{self.protocol}://{self.host}{self.path}"

    @classmethod
    @property
    def default(cls) -> "DavSettings":
        return cls(
            protocol="https",
            host="example.com",
            username="ovdashboard",
            password="secret",
        )


class Settings(BaseSettings):
    """
    Per-run settings.
    """

    production_mode: bool = False
    log_level: str = "INFO" if production_mode else "DEBUG"
    cache_time: int = 30
    cache_size: int = 30

    # doesn't even have to be reachable
    ping_host: str = "10.0.0.0"
    ping_port: int = 1

    main_host: str = "0.0.0.0"
    main_port: int = 8000
    api_prefix: str = "api/v1"
    openapi_url: str = "/openapi.json"
    docs_url: Optional[str] = None if production_mode else "/docs"
    redoc_url: Optional[str] = None if production_mode else "/redoc"

    webdav: DavSettings = DavSettings()
    webdav_retries: int = 20
    webdav_prefix: str = "/ovdashboard"
    config_path: str = "config.txt"

    caldav: DavSettings = DavSettings()

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"
        env_nested_delimiter = "__"

    @root_validator(pre=True)
    @classmethod
    def validate_dav_settings(cls, values: dict[str, Any]) -> dict[str, Any]:
        # ensure both settings dicts are created
        for key in ("webdav", "caldav"):
            if key not in values:
                values[key] = {}

        default_dav = DavSettings.default.dict()

        for key in default_dav:
            # if "webdav" value is not specified, use default
            if key not in values["webdav"]:
                values["webdav"][key] = default_dav[key]

            # if "caldav" value is not specified, use "webdav" value
            if key not in values["caldav"]:
                values["caldav"][key] = values["webdav"][key]

        # add default "path"s if None
        if values["webdav"]["path"] is None:
            values["webdav"]["path"] = "/remote.php/webdav"

        if values["caldav"]["path"] is None:
            values["caldav"]["path"] = "/remote.php/dav"

        return values


SETTINGS = Settings()