mirror of
https://github.com/ChrisSewell/LeosShoes.git
synced 2025-07-01 18:17:26 -04:00
75 lines
2.5 KiB
Python
75 lines
2.5 KiB
Python
"""Configuration management for paw burn risk assessment."""
|
|
|
|
import os
|
|
from dotenv import load_dotenv
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
# Load environment variables
|
|
load_dotenv()
|
|
|
|
@dataclass
|
|
class RiskConfig:
|
|
"""Configuration for risk assessment parameters."""
|
|
temp_threshold_low: float = 80.0
|
|
temp_threshold_med: float = 90.0
|
|
temp_threshold_high: float = 100.0
|
|
|
|
uv_threshold_low: float = 6.0
|
|
uv_threshold_med: float = 8.0
|
|
uv_threshold_high: float = 10.0
|
|
|
|
risk_threshold_shoes: float = 6.0
|
|
rolling_window_hours: int = 2
|
|
surface_recovery_hours: int = 2
|
|
|
|
@classmethod
|
|
def from_env(cls) -> 'RiskConfig':
|
|
"""Create configuration from environment variables."""
|
|
return cls(
|
|
temp_threshold_low=float(os.getenv('TEMP_THRESHOLD_LOW', 80)),
|
|
temp_threshold_med=float(os.getenv('TEMP_THRESHOLD_MED', 90)),
|
|
temp_threshold_high=float(os.getenv('TEMP_THRESHOLD_HIGH', 100)),
|
|
uv_threshold_low=float(os.getenv('UV_THRESHOLD_LOW', 6)),
|
|
uv_threshold_med=float(os.getenv('UV_THRESHOLD_MED', 8)),
|
|
uv_threshold_high=float(os.getenv('UV_THRESHOLD_HIGH', 10)),
|
|
risk_threshold_shoes=float(os.getenv('RISK_THRESHOLD_SHOES', 6)),
|
|
rolling_window_hours=int(os.getenv('ROLLING_WINDOW_HOURS', 2)),
|
|
surface_recovery_hours=int(os.getenv('SURFACE_RECOVERY_HOURS', 2))
|
|
)
|
|
|
|
@dataclass
|
|
class AppConfig:
|
|
"""Main application configuration."""
|
|
weather_api_key: str
|
|
default_location: str = "New York"
|
|
database_path: str = "paw_risk.db"
|
|
risk_config: RiskConfig = None
|
|
|
|
def __post_init__(self):
|
|
if self.risk_config is None:
|
|
self.risk_config = RiskConfig.from_env()
|
|
|
|
@classmethod
|
|
def from_env(cls) -> 'AppConfig':
|
|
"""Create configuration from environment variables."""
|
|
api_key = os.getenv('WEATHER_API_KEY')
|
|
if not api_key:
|
|
raise ValueError("WEATHER_API_KEY environment variable is required")
|
|
|
|
return cls(
|
|
weather_api_key=api_key,
|
|
default_location=os.getenv('DEFAULT_LOCATION', 'New York'),
|
|
database_path=os.getenv('DATABASE_PATH', 'paw_risk.db'),
|
|
risk_config=RiskConfig.from_env()
|
|
)
|
|
|
|
# Global configuration instance
|
|
config = None
|
|
|
|
def get_config() -> AppConfig:
|
|
"""Get the global configuration instance."""
|
|
global config
|
|
if config is None:
|
|
config = AppConfig.from_env()
|
|
return config |