coordinator.py

Go to the documentation of this file.

"""DROP device data update coordinator object."""
 
from __future__ import annotations
 
import logging
 
from dropmqttapi.mqttapi import DropAPI
 
from homeassistant.components import mqtt
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
from .const import CONF_COMMAND_TOPIC, DOMAIN
 
_LOGGER = logging.getLogger(__name__)
 
 
class DROPDeviceDataUpdateCoordinator(DataUpdateCoordinator[None]):
    """DROP device object."""
 
    config_entry: ConfigEntry
 
    def __init__(self, hass: HomeAssistant, unique_id: str) -> None:
        """Initialize the device."""
        super().__init__(hass, _LOGGER, name=f"{DOMAIN}-{unique_id}")
        self.drop_apidrop_api = DropAPI()
 
    async def set_water(self, value: int) -> None:
        """Change water supply state."""
        payload = self.drop_apidrop_api.set_water_message(value)
        await mqtt.async_publish(
            self.hasshass,
            self.config_entryconfig_entry.data[CONF_COMMAND_TOPIC],
            payload,
        )
 
    async def set_bypass(self, value: int) -> None:
        """Change water bypass state."""
        payload = self.drop_apidrop_api.set_bypass_message(value)
        await mqtt.async_publish(
            self.hasshass,
            self.config_entryconfig_entry.data[CONF_COMMAND_TOPIC],
            payload,
        )
 
    async def set_protect_mode(self, value: str) -> None:
        """Change protect mode state."""
        payload = self.drop_apidrop_api.set_protect_mode_message(value)
        await mqtt.async_publish(
            self.hasshass,
            self.config_entryconfig_entry.data[CONF_COMMAND_TOPIC],
            payload,
        )