core/homeassistant/components/omie/util.py

"""Utility functions for OMIE - Spain and Portugal electricity prices integration."""

import datetime as dt
from typing import Final
from zoneinfo import ZoneInfo

from pyomie.model import OMIEResults, SpotData
from pyomie.util import localize_quarter_hourly_data

CET: Final = ZoneInfo("CET")


def current_quarter_hour_cet(current_time: dt.datetime) -> dt.datetime:
    """Return the start of the quarter-hour for the passed in time in CET."""
    current_quarter_begin = current_time.minute // 15 * 15
    return current_time.replace(
        minute=current_quarter_begin, second=0, microsecond=0
    ).astimezone(CET)


def pick_series_cet(
    res: OMIEResults[SpotData] | None,
    series_name: str,
) -> dict[dt.datetime, float]:
    """Pick values for this series from market data, keyed by CET."""
    if res is None:
        return {}

    market_date = res.market_date
    series_data = getattr(res.contents, series_name, [])

    return {
        dt.datetime.fromisoformat(dt_str).astimezone(CET): series_values
        for dt_str, series_values in localize_quarter_hourly_data(
            market_date, series_data
        ).items()
    }