Refactor

src/oidc-test/settings.py

@@ -0,0 +1,71 @@
+import string
+import random
+from typing import Type, Tuple
+
+from pydantic import BaseModel, computed_field
+from pydantic_settings import (
+    BaseSettings,
+    PydanticBaseSettingsSource,
+    YamlConfigSettingsSource,
+)
+
+
+class OIDCProvider(BaseModel):
+    name: str = ""
+    url: str = ""
+    client_id: str = ""
+    client_secret: str = ""
+    is_swagger: bool = False
+
+    @computed_field
+    @property
+    def provider_url(self) -> str:
+        return self.url + "/.well-known/openid-configuration"
+
+    @computed_field
+    @property
+    def token_url(self) -> str:
+        return "auth/" + self.name
+
+
+class OIDCSettings(BaseModel):
+    show_session_details: bool = False
+    providers: list[OIDCProvider] = []
+    swagger_provider: str = ""
+
+    def get_swagger_provider(self) -> OIDCProvider:
+        for provider in self.providers:
+            if provider.is_swagger:
+                return provider
+        else:
+            raise UserWarning("Please define a provider for Swagger with id_swagger")
+
+
+class Settings(BaseSettings):
+    """Settings wil be read from an .env file"""
+
+    oidc: OIDCSettings = OIDCSettings()
+    secret_key: str = "".join(random.choice(string.ascii_letters) for _ in range(16))
+
+    class Config:
+        env_nested_delimiter = "__"
+
+    @classmethod
+    def settings_customise_sources(
+        cls,
+        settings_cls: Type[BaseSettings],
+        init_settings: PydanticBaseSettingsSource,
+        env_settings: PydanticBaseSettingsSource,
+        dotenv_settings: PydanticBaseSettingsSource,
+        file_secret_settings: PydanticBaseSettingsSource,
+    ) -> Tuple[PydanticBaseSettingsSource, ...]:
+        return (
+            init_settings,
+            env_settings,
+            file_secret_settings,
+            YamlConfigSettingsSource(settings_cls, "settings.yaml"),
+            dotenv_settings,
+        )
+
+
+settings = Settings()