77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
"""Support for Deebot Sensor."""
|
|
from typing import Optional, Dict, Any
|
|
|
|
from deebotozmo import *
|
|
from homeassistant.components.binary_sensor import BinarySensorEntity
|
|
|
|
from .const import DOMAIN
|
|
from .helpers import get_device_info
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_devices):
|
|
"""Add binary_sensor for passed config_entry in HA."""
|
|
hub = hass.data[DOMAIN][config_entry.entry_id]
|
|
|
|
new_devices = []
|
|
for vacbot in hub.vacbots:
|
|
new_devices.append(DeebotMopAttachedBinarySensor(vacbot, "mop_attached"))
|
|
|
|
if new_devices:
|
|
async_add_devices(new_devices)
|
|
|
|
|
|
class DeebotMopAttachedBinarySensor(BinarySensorEntity):
|
|
"""Deebot mop attached binary sensor"""
|
|
|
|
def __init__(self, vacbot: VacBot, device_id: str):
|
|
"""Initialize the Sensor."""
|
|
self._vacbot = vacbot
|
|
self._id = device_id
|
|
|
|
if self._vacbot.vacuum.get("nick", None) is not None:
|
|
self._vacbot_name = "{}".format(self._vacbot.vacuum["nick"])
|
|
else:
|
|
# In case there is no nickname defined, use the device id
|
|
self._vacbot_name = "{}".format(self._vacbot.vacuum["did"])
|
|
|
|
self._name = self._vacbot_name + "_" + device_id
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return an unique ID."""
|
|
return self._vacbot.vacuum.get("did", None) + "_" + self._id
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the device."""
|
|
return self._name
|
|
|
|
@property
|
|
def should_poll(self) -> bool:
|
|
return False
|
|
|
|
@property
|
|
def is_on(self):
|
|
return self._vacbot.mop_attached
|
|
|
|
@property
|
|
def icon(self) -> Optional[str]:
|
|
"""Return the icon to use in the frontend, if any."""
|
|
return "mdi:water" if self.is_on else "mdi:water-off"
|
|
|
|
@property
|
|
def entity_registry_enabled_default(self) -> bool:
|
|
"""Return if the entity should be enabled when first added to the entity registry."""
|
|
return True
|
|
|
|
@property
|
|
def device_info(self) -> Optional[Dict[str, Any]]:
|
|
return get_device_info(self._vacbot)
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Set up the event listeners now that hass is ready."""
|
|
listener: EventListener = self._vacbot.waterEvents.subscribe(lambda _: self.schedule_update_ha_state())
|
|
self.async_on_remove(listener.unsubscribe)
|