ha-core/homeassistant/components/elvia/__init__.py

50 lines
1.4 KiB
Python

"""The Elvia integration."""
from __future__ import annotations
from datetime import datetime, timedelta
from typing import TYPE_CHECKING
from elvia import error as ElviaError
from homeassistant.const import CONF_API_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.helpers.event import async_track_time_interval
from .const import CONF_METERING_POINT_ID, LOGGER
from .importer import ElviaImporter
if TYPE_CHECKING:
from homeassistant.config_entries import ConfigEntry
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Elvia from a config entry."""
importer = ElviaImporter(
hass=hass,
api_token=entry.data[CONF_API_TOKEN],
metering_point_id=entry.data[CONF_METERING_POINT_ID],
)
async def _import_meter_values(_: datetime | None = None) -> None:
"""Import meter values."""
try:
await importer.import_meter_values()
except ElviaError.ElviaException as exception:
LOGGER.exception("Unknown error %s", exception)
try:
await importer.import_meter_values()
except ElviaError.ElviaException as exception:
LOGGER.exception("Unknown error %s", exception)
return False
entry.async_on_unload(
async_track_time_interval(
hass,
_import_meter_values,
timedelta(minutes=60),
)
)
return True