settings.py

from typing import Literal

import yaml

# Compatability shim to support users with v1 and v2 installed
try:
    from pydantic.v1 import BaseSettings, AnyUrl
except ImportError:
    from pydantic import BaseSettings, AnyUrl


from .dot_proxygen import settings_file


def _yaml_settings_file_source(_):
    with settings_file().open() as yaml_file:
        settings = yaml.safe_load(yaml_file)
        return settings or {}


class Settings(BaseSettings):
    endpoint_url: AnyUrl = "https://proxygen.prod.api.platform.nhs.uk"
    spec_output_format: Literal["json", "yaml"] = "yaml"
    api: str = None

    class Config:
        env_prefix: Literal["PROXYGEN_CREDENTIALS_"]

        @classmethod
        def customise_sources(
            cls,
            init_settings,
            env_settings,
            file_secret_settings,
        ):
            return (
                init_settings,
                env_settings,
                _yaml_settings_file_source,
            )


SETTINGS = Settings()