Model/backend/app/config.py

import os
from functools import lru_cache
from pydantic_settings import BaseSettings, SettingsConfigDict
from typing import Optional


def resolve_env_file() -> Optional[str]:
    env = os.getenv("ENVIRONMENT", "local")

    if env == "local":
        return "backend/.env"

    if env == "test":
        return "backend/.env.test"

    # prod = no env file
    return None


class Settings(BaseSettings):
    API_KEY: str
    API_KEY_NAME: str = "X-API-KEY"
    SECRET_KEY: str
    ENVIRONMENT: str
    DATA_BUCKET: str
    PLAN_TRIGGER_BUCKET: str
    ENGINE_SQS_URL: str

    # Third parties
    EPC_AUTH_TOKEN: str
    GOOGLE_SOLAR_API_KEY: str

    # Database settings
    DB_HOST: str
    DB_PASSWORD: str
    DB_USERNAME: str
    DB_PORT: str
    DB_NAME: str

    # Prediction buckets
    SAP_PREDICTIONS_BUCKET: str
    CARBON_PREDICTIONS_BUCKET: str
    HEAT_PREDICTIONS_BUCKET: str
    # LIGHTING_COST_PREDICTIONS_BUCKET: str
    # HEATING_COST_PREDICTIONS_BUCKET: str
    # HOT_WATER_COST_PREDICTIONS_BUCKET: str
    HEATING_KWH_PREDICTIONS_BUCKET: str
    HOTWATER_KWH_PREDICTIONS_BUCKET: str

    # Other S3 buckts
    ENERGY_ASSESSMENTS_BUCKET: str

    # Optional AWS creds (only required in local)
    AWS_ACCESS_KEY_ID: Optional[str] = None
    AWS_SECRET_KEY_ID: Optional[str] = None
    AWS_DEFAULT_REGION: Optional[str] = None

    model_config = SettingsConfigDict(
        env_file=resolve_env_file(),
        env_file_encoding="utf-8",
    )


@lru_cache()
def get_settings():
    return Settings()


@lru_cache()
def get_prediction_buckets():
    return {
        "sap_change_predictions": get_settings().SAP_PREDICTIONS_BUCKET,
        "heat_demand_predictions": get_settings().HEAT_PREDICTIONS_BUCKET,
        "carbon_change_predictions": get_settings().CARBON_PREDICTIONS_BUCKET,
        "heating_kwh_predictions": get_settings().HEATING_KWH_PREDICTIONS_BUCKET,
        "hotwater_kwh_predictions": get_settings().HOTWATER_KWH_PREDICTIONS_BUCKET,
    }