56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
"""Services for the Tado integration."""
|
|
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.core import HomeAssistant, ServiceCall, callback
|
|
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
|
|
from homeassistant.helpers import selector
|
|
|
|
from .const import (
|
|
ATTR_MESSAGE,
|
|
CONF_CONFIG_ENTRY,
|
|
CONF_READING,
|
|
DOMAIN,
|
|
SERVICE_ADD_METER_READING,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
SCHEMA_ADD_METER_READING = vol.Schema(
|
|
{
|
|
vol.Required(CONF_CONFIG_ENTRY): selector.ConfigEntrySelector(
|
|
{
|
|
"integration": DOMAIN,
|
|
}
|
|
),
|
|
vol.Required(CONF_READING): vol.Coerce(int),
|
|
}
|
|
)
|
|
|
|
|
|
async def _add_meter_reading(call: ServiceCall) -> None:
|
|
"""Send meter reading to Tado."""
|
|
entry_id: str = call.data[CONF_CONFIG_ENTRY]
|
|
reading: int = call.data[CONF_READING]
|
|
_LOGGER.debug("Add meter reading %s", reading)
|
|
|
|
entry = call.hass.config_entries.async_get_entry(entry_id)
|
|
if entry is None:
|
|
raise ServiceValidationError("Config entry not found")
|
|
|
|
coordinator = entry.runtime_data.coordinator
|
|
response: dict = await coordinator.set_meter_reading(call.data[CONF_READING])
|
|
|
|
if ATTR_MESSAGE in response:
|
|
raise HomeAssistantError(response[ATTR_MESSAGE])
|
|
|
|
|
|
@callback
|
|
def async_setup_services(hass: HomeAssistant) -> None:
|
|
"""Set up the services for the Tado integration."""
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_ADD_METER_READING, _add_meter_reading, SCHEMA_ADD_METER_READING
|
|
)
|