core/homeassistant/components/forecast_solar/energy.py

"""Energy platform."""

from __future__ import annotations

from homeassistant.core import HomeAssistant

from .coordinator import ForecastSolarDataUpdateCoordinator


async def async_get_solar_forecast(
    hass: HomeAssistant, config_entry_id: str
) -> dict[str, dict[str, float | int]] | None:
    """Get solar forecast for a config entry ID."""
    if (
        entry := hass.config_entries.async_get_entry(config_entry_id)
    ) is None or not isinstance(entry.runtime_data, ForecastSolarDataUpdateCoordinator):
        return None

    return {
        "wh_hours": {
            timestamp.isoformat(): val
            for timestamp, val in entry.runtime_data.data.wh_period.items()
        }
    }