91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
import logging
|
|
from homeassistant.components.light import LightEntity, ColorMode
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity, DataUpdateCoordinator
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = "plejd"
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
devices = hass.data[DOMAIN]["devices"].get(config_entry.entry_id, [])
|
|
|
|
entities = []
|
|
for d in devices:
|
|
dev = devices[d]
|
|
if dev.type == "light":
|
|
coordinator = Coordinator(hass, dev)
|
|
dev.updateCallback = coordinator.async_set_updated_data
|
|
light = PlejdLight(coordinator, dev)
|
|
entities.append(light)
|
|
async_add_entities(entities, False)
|
|
|
|
class Coordinator(DataUpdateCoordinator):
|
|
def __init__(self, hass, device):
|
|
super().__init__(hass, _LOGGER, name="Plejd Coordinator")
|
|
self.device = device
|
|
|
|
class PlejdLight(LightEntity, CoordinatorEntity):
|
|
def __init__(self, coordinator, device):
|
|
CoordinatorEntity.__init__(self, coordinator)
|
|
LightEntity.__init__(self)
|
|
self.device = device
|
|
|
|
@property
|
|
def _data(self):
|
|
return self.coordinator.data or {}
|
|
|
|
@property
|
|
def device_info(self):
|
|
return {
|
|
"identifiers": {(DOMAIN, f"{self.device.BLE_address}:{self.device.address}")},
|
|
"name": self.device.name,
|
|
"manufacturer": "Plejd",
|
|
"model": {self.device.model},
|
|
#"connections": ???,
|
|
"suggested_area": self.device.room,
|
|
"sw_version": f"{self.device.firmware} ({self.device.hardwareId})",
|
|
}
|
|
|
|
@property
|
|
def has_entity_name(self):
|
|
return True
|
|
|
|
@property
|
|
def name(self):
|
|
return None
|
|
|
|
@property
|
|
def unique_id(self):
|
|
return f"{self.device.BLE_address}:{self.device.address}"
|
|
|
|
@property
|
|
def is_on(self):
|
|
return self.device.state
|
|
|
|
@property
|
|
def brightness(self):
|
|
return self.device.dim
|
|
|
|
@property
|
|
def supported_color_modes(self):
|
|
if self.device.dimmable:
|
|
return {ColorMode.BRIGHTNESS}
|
|
return {ColorMode.ONOFF}
|
|
|
|
@property
|
|
def color_mode(self):
|
|
if self.device.dimmable:
|
|
return ColorMode.BRIGHTNESS
|
|
return ColorMode.ONOFF
|
|
|
|
async def async_turn_on(self, brightness=None, **_):
|
|
await self.device.turn_on(brightness)
|
|
pass
|
|
|
|
async def async_turn_off(self, **_):
|
|
await self.device.turn_off()
|
|
pass
|
|
|
|
|