init

play_outside/config.py

@@ -0,0 +1,98 @@
+import pydantic
+from dotenv import load_dotenv
+from pydantic_settings import BaseSettings
+from pydantic_settings import BaseSettings, SettingsConfigDict
+from pydantic import BaseModel
+from pydantic import Field
+from typing import Any
+from fastapi.templating import Jinja2Templates
+from urllib.parse import quote_plus
+from typing import Optional
+from datetime import datetime
+from datetime import timezone
+from functools import lru_cache
+import os
+import jinja2
+
+from rich.console import Console
+
+console = Console()
+
+
+class ApiServer(BaseModel):
+    app: str = "play_outside.api:app"
+    port: int = 8200
+    reload: bool = True
+    log_level: str = "info"
+    host: str = "0.0.0.0"
+    workers: int = 1
+    forwarded_allow_ips: str = "*"
+    proxy_headers: bool = True
+
+
+if hasattr(jinja2, "pass_context"):
+    pass_context = jinja2.pass_context
+else:
+    pass_context = jinja2.contextfunction
+
+
+@pass_context
+def https_url_for(context: dict, name: str, **path_params: Any) -> str:
+    request = context["request"]
+    http_url = request.url_for(name, **path_params)
+    return str(http_url).replace("http", "https", 1)
+
+
+def get_templates(config: BaseSettings) -> Jinja2Templates:
+    templates = Jinja2Templates(directory="templates")
+    templates.env.filters["quote_plus"] = lambda u: quote_plus(str(u))
+    templates.env.filters["timestamp"] = lambda u: datetime.fromtimestamp(
+        u, tz=timezone.utc
+    ).strftime("%B %d, %Y")
+    templates.env.globals["https_url_for"] = https_url_for
+    templates.env.globals["config"] = config
+    templates.env.globals["datetime"] = datetime
+    templates.env.globals["len"] = len
+    templates.env.globals["int"] = int
+    console.print(f'Using environment: {os.environ.get("ENV")}')
+
+    if os.environ.get("ENV") in ["dev", "qa", "prod"]:
+        templates.env.globals["url_for"] = https_url_for
+        console.print("Using HTTPS")
+    else:
+        console.print("Using HTTP")
+
+    return templates
+
+
+class Config(BaseSettings):
+    env: str = "prod"
+    open_weather_api_key: Optional[str] = None
+    api_server: ApiServer = ApiServer()
+    the_templates: Optional[Jinja2Templates] = Field(None, exclude=True)
+    model_config = SettingsConfigDict(
+        env_prefix="PLAY_OUTSIDE_",
+        env_nested_delimiter="__",
+        case_sensitive=False,
+    )
+
+    @property
+    def templates(self) -> Jinja2Templates:
+        if self.the_templates is None:
+            self.the_templates = get_templates(self)
+        return self.the_templates
+
+    @pydantic.validator("open_weather_api_key", pre=True, always=True)
+    def validate_open_weather_api_key(cls, v):
+        if v is None:
+            v = os.getenv("OPEN_WEATHER_API_KEY")
+        return v
+
+
+@lru_cache
+def get_config(env: Optional[str] = None):
+    if env is None:
+        env = os.environ.get("ENV", "prod")
+    load_dotenv(dotenv_path=f".env.{env}")
+    config = Config(env=env)
+    return config