112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
"""Base class for deCONZ devices."""
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers.device_registry import CONNECTION_ZIGBEE
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from .const import DOMAIN as DECONZ_DOMAIN
|
|
|
|
|
|
class DeconzBase:
|
|
"""Common base for deconz entities and events."""
|
|
|
|
def __init__(self, device, gateway):
|
|
"""Set up device and add update callback to get data from websocket."""
|
|
self._device = device
|
|
self.gateway = gateway
|
|
self.listeners = []
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return a unique identifier for this device."""
|
|
return self._device.uniqueid
|
|
|
|
@property
|
|
def serial(self):
|
|
"""Return a serial number for this device."""
|
|
if self._device.uniqueid is None or self._device.uniqueid.count(":") != 7:
|
|
return None
|
|
|
|
return self._device.uniqueid.split("-", 1)[0]
|
|
|
|
@property
|
|
def device_info(self):
|
|
"""Return a device description for device registry."""
|
|
if self.serial is None:
|
|
return None
|
|
|
|
bridgeid = self.gateway.api.config.bridgeid
|
|
|
|
return {
|
|
"connections": {(CONNECTION_ZIGBEE, self.serial)},
|
|
"identifiers": {(DECONZ_DOMAIN, self.serial)},
|
|
"manufacturer": self._device.manufacturer,
|
|
"model": self._device.modelid,
|
|
"name": self._device.name,
|
|
"sw_version": self._device.swversion,
|
|
"via_device": (DECONZ_DOMAIN, bridgeid),
|
|
}
|
|
|
|
|
|
class DeconzDevice(DeconzBase, Entity):
|
|
"""Representation of a deCONZ device."""
|
|
|
|
def __init__(self, device, gateway):
|
|
"""Set up device and add update callback to get data from websocket."""
|
|
super().__init__(device, gateway)
|
|
|
|
self.unsub_dispatcher = None
|
|
|
|
@property
|
|
def entity_registry_enabled_default(self):
|
|
"""Return if the entity should be enabled when first added to the entity registry."""
|
|
if not self.gateway.option_allow_clip_sensor and self._device.type.startswith(
|
|
"CLIP"
|
|
):
|
|
return False
|
|
|
|
if (
|
|
not self.gateway.option_allow_deconz_groups
|
|
and self._device.type == "LightGroup"
|
|
):
|
|
return False
|
|
|
|
return True
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Subscribe to device events."""
|
|
self._device.register_async_callback(self.async_update_callback)
|
|
self.gateway.deconz_ids[self.entity_id] = self._device.deconz_id
|
|
self.listeners.append(
|
|
async_dispatcher_connect(
|
|
self.hass, self.gateway.signal_reachable, self.async_update_callback
|
|
)
|
|
)
|
|
|
|
async def async_will_remove_from_hass(self) -> None:
|
|
"""Disconnect device object when removed."""
|
|
self._device.remove_callback(self.async_update_callback)
|
|
del self.gateway.deconz_ids[self.entity_id]
|
|
for unsub_dispatcher in self.listeners:
|
|
unsub_dispatcher()
|
|
|
|
@callback
|
|
def async_update_callback(self, force_update=False):
|
|
"""Update the device's state."""
|
|
self.async_schedule_update_ha_state()
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return True if device is available."""
|
|
return self.gateway.available and self._device.reachable
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the device."""
|
|
return self._device.name
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""No polling needed."""
|
|
return False
|