254 lines
8.9 KiB
Python
254 lines
8.9 KiB
Python
"""Platform for light integration."""
|
|
from __future__ import annotations
|
|
from typing import Any
|
|
from homeassistant.components.light import (ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN, ATTR_RGB_COLOR, ColorMode, LightEntity)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.entity import DeviceInfo
|
|
from .const import DOMAIN
|
|
import logging
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback
|
|
) -> None:
|
|
hub = hass.data[DOMAIN][config_entry.entry_id]
|
|
|
|
new_devices = []
|
|
for room in hub.cync_rooms:
|
|
if not hub.cync_rooms[room]._update_callback and (room in config_entry.options["rooms"] or room in config_entry.options["subgroups"]):
|
|
new_devices.append(CyncRoomEntity(hub.cync_rooms[room]))
|
|
|
|
for switch_id in hub.cync_switches:
|
|
if not hub.cync_switches[switch_id]._update_callback and not hub.cync_switches[switch_id].plug and not hub.cync_switches[switch_id].fan and switch_id in config_entry.options["switches"]:
|
|
new_devices.append(CyncSwitchEntity(hub.cync_switches[switch_id]))
|
|
|
|
if new_devices:
|
|
async_add_entities(new_devices)
|
|
|
|
|
|
class CyncRoomEntity(LightEntity):
|
|
"""Representation of a Cync Room Light Entity."""
|
|
|
|
should_poll = False
|
|
|
|
def __init__(self, room) -> None:
|
|
"""Initialize the light."""
|
|
self.room = room
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Run when this Entity has been added to HA."""
|
|
self.room.register(self.schedule_update_ha_state)
|
|
|
|
async def async_will_remove_from_hass(self) -> None:
|
|
"""Entity being removed from hass."""
|
|
self.room.reset()
|
|
|
|
@property
|
|
def device_info(self) -> DeviceInfo:
|
|
"""Return device registry information for this entity."""
|
|
return DeviceInfo(
|
|
identifiers = {(DOMAIN, f"{self.room.parent_room if self.room.is_subgroup else self.room.name} ({self.room.home_name})")},
|
|
manufacturer = "Cync by Savant",
|
|
name = f"{self.room.parent_room if self.room.is_subgroup else self.room.name} ({self.room.home_name})",
|
|
suggested_area = f"{self.room.parent_room if self.room.is_subgroup else self.room.name}",
|
|
)
|
|
|
|
@property
|
|
def icon(self) -> str | None:
|
|
"""Icon of the entity."""
|
|
if self.room.is_subgroup:
|
|
return "mdi:lightbulb-group-outline"
|
|
else:
|
|
return "mdi:lightbulb-group"
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return Unique ID string."""
|
|
uid = 'cync_room_' + '-'.join(self.room.switches) + '_' + '-'.join(self.room.subgroups)
|
|
return uid
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Return the name of the room."""
|
|
return self.room.name
|
|
|
|
@property
|
|
def is_on(self) -> bool | None:
|
|
"""Return true if light is on."""
|
|
return self.room.power_state
|
|
|
|
@property
|
|
def brightness(self) -> int | None:
|
|
"""Return the brightness of this room between 0..255."""
|
|
return round(self.room.brightness*255/100)
|
|
|
|
@property
|
|
def max_color_temp_kelvin(self) -> int:
|
|
"""Return maximum supported color temperature."""
|
|
return self.room.max_color_temp_kelvin
|
|
|
|
@property
|
|
def min_color_temp_kelvin(self) -> int:
|
|
"""Return minimum supported color temperature."""
|
|
return self.room.min_color_temp_kelvin
|
|
|
|
@property
|
|
def color_temp_kelvin(self) -> int:
|
|
"""Return color temperature in kelvin."""
|
|
return self.room.color_temp_kelvin
|
|
|
|
@property
|
|
def rgb_color(self) -> tuple[int, int, int] | None:
|
|
"""Return the RGB color tuple of this light switch"""
|
|
return (self.room.rgb['r'],self.room.rgb['g'],self.room.rgb['b'])
|
|
|
|
@property
|
|
def supported_color_modes(self) -> set[str] | None:
|
|
"""Return list of available color modes."""
|
|
|
|
modes: set[ColorMode | str] = set()
|
|
|
|
if self.room.support_color_temp:
|
|
modes.add(ColorMode.COLOR_TEMP)
|
|
if self.room.support_rgb:
|
|
modes.add(ColorMode.RGB)
|
|
if self.room.support_brightness:
|
|
modes.add(ColorMode.BRIGHTNESS)
|
|
if not modes:
|
|
modes.add(ColorMode.ONOFF)
|
|
|
|
return modes
|
|
|
|
@property
|
|
def color_mode(self) -> str | None:
|
|
"""Return the active color mode."""
|
|
|
|
if self.room.support_color_temp:
|
|
if self.room.support_rgb and self.room.rgb['active']:
|
|
return ColorMode.RGB
|
|
else:
|
|
return ColorMode.COLOR_TEMP
|
|
if self.room.support_brightness:
|
|
return ColorMode.BRIGHTNESS
|
|
else:
|
|
return ColorMode.ONOFF
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the light."""
|
|
await self.room.turn_on(kwargs.get(ATTR_RGB_COLOR),kwargs.get(ATTR_BRIGHTNESS),kwargs.get(ATTR_COLOR_TEMP_KELVIN))
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the light."""
|
|
await self.room.turn_off()
|
|
|
|
class CyncSwitchEntity(LightEntity):
|
|
"""Representation of a Cync Switch Light Entity."""
|
|
|
|
should_poll = False
|
|
|
|
def __init__(self, cync_switch) -> None:
|
|
"""Initialize the light."""
|
|
self.cync_switch = cync_switch
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
"""Run when this Entity has been added to HA."""
|
|
self.cync_switch.register(self.schedule_update_ha_state)
|
|
|
|
async def async_will_remove_from_hass(self) -> None:
|
|
"""Entity being removed from hass."""
|
|
self.cync_switch.reset()
|
|
|
|
@property
|
|
def device_info(self) -> DeviceInfo:
|
|
"""Return device registry information for this entity."""
|
|
return DeviceInfo(
|
|
identifiers = {(DOMAIN, f"{self.cync_switch.room.name} ({self.cync_switch.home_name})")},
|
|
manufacturer = "Cync by Savant",
|
|
name = f"{self.cync_switch.room.name} ({self.cync_switch.home_name})",
|
|
suggested_area = f"{self.cync_switch.room.name}",
|
|
)
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return Unique ID string."""
|
|
return 'cync_switch_' + self.cync_switch.device_id
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Return the name of the switch."""
|
|
return self.cync_switch.name
|
|
|
|
@property
|
|
def is_on(self) -> bool | None:
|
|
"""Return true if light is on."""
|
|
return self.cync_switch.power_state
|
|
|
|
@property
|
|
def brightness(self) -> int | None:
|
|
"""Return the brightness of this switch between 0..255."""
|
|
return round(self.cync_switch.brightness*255/100)
|
|
|
|
@property
|
|
def max_color_temp_kelvin(self) -> int:
|
|
"""Return maximum supported color temperature."""
|
|
return self.cync_switch.max_color_temp_kelvin
|
|
|
|
@property
|
|
def min_color_temp_kelvin(self) -> int:
|
|
"""Return minimum supported color temperature."""
|
|
return self.cync_switch.min_color_temp_kelvin
|
|
|
|
@property
|
|
def color_temp_kelvin(self) -> int | None:
|
|
"""Return the color temperature of this light for HA."""
|
|
return self.cync_switch.color_temp_kelvin
|
|
|
|
@property
|
|
def rgb_color(self) -> tuple[int, int, int] | None:
|
|
"""Return the RGB color tuple of this light switch"""
|
|
return (self.cync_switch.rgb['r'],self.cync_switch.rgb['g'],self.cync_switch.rgb['b'])
|
|
|
|
@property
|
|
def supported_color_modes(self) -> set[str] | None:
|
|
"""Return list of available color modes."""
|
|
|
|
modes: set[ColorMode | str] = set()
|
|
|
|
if self.cync_switch.support_color_temp:
|
|
modes.add(ColorMode.COLOR_TEMP)
|
|
if self.cync_switch.support_rgb:
|
|
modes.add(ColorMode.RGB)
|
|
if self.cync_switch.support_brightness:
|
|
modes.add(ColorMode.BRIGHTNESS)
|
|
if not modes:
|
|
modes.add(ColorMode.ONOFF)
|
|
|
|
return modes
|
|
|
|
@property
|
|
def color_mode(self) -> str | None:
|
|
"""Return the active color mode."""
|
|
|
|
if self.cync_switch.support_color_temp:
|
|
if self.cync_switch.support_rgb and self.cync_switch.rgb['active']:
|
|
return ColorMode.RGB
|
|
else:
|
|
return ColorMode.COLOR_TEMP
|
|
if self.cync_switch.support_brightness:
|
|
return ColorMode.BRIGHTNESS
|
|
else:
|
|
return ColorMode.ONOFF
|
|
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn on the light."""
|
|
await self.cync_switch.turn_on(kwargs.get(ATTR_RGB_COLOR),kwargs.get(ATTR_BRIGHTNESS),kwargs.get(ATTR_COLOR_TEMP_KELVIN))
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn off the light."""
|
|
await self.cync_switch.turn_off()
|