With all binary_sensor ok

This commit is contained in:
Jean-Marc Collin
2023-02-28 22:48:07 +01:00
parent 330c3323d1
commit 168568ac5d

View File

@@ -11,7 +11,13 @@ from homeassistant.config_entries import ConfigEntry
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .commons import VersatileThermostatBaseEntity from .commons import VersatileThermostatBaseEntity
from .const import CONF_NAME from .const import (
CONF_NAME,
CONF_USE_POWER_FEATURE,
CONF_USE_PRESENCE_FEATURE,
CONF_USE_MOTION_FEATURE,
CONF_USE_WINDOW_FEATURE,
)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@@ -29,23 +35,22 @@ async def async_setup_entry(
unique_id = entry.entry_id unique_id = entry.entry_id
name = entry.data.get(CONF_NAME) name = entry.data.get(CONF_NAME)
async_add_entities( entities = [SecurityBinarySensor(hass, unique_id, name, entry.data)]
[ if entry.data.get(CONF_USE_MOTION_FEATURE):
SecurityBinarySensor(hass, unique_id, name, entry.data), entities.append(MotionBinarySensor(hass, unique_id, name, entry.data))
OverpoweringBinarySensor(hass, unique_id, name, entry.data), if entry.data.get(CONF_USE_WINDOW_FEATURE):
WindowBinarySensor(hass, unique_id, name, entry.data), entities.append(WindowBinarySensor(hass, unique_id, name, entry.data))
MotionBinarySensor(hass, unique_id, name, entry.data), if entry.data.get(CONF_USE_PRESENCE_FEATURE):
PresenceBinarySensor(hass, unique_id, name, entry.data), entities.append(PresenceBinarySensor(hass, unique_id, name, entry.data))
], if entry.data.get(CONF_USE_POWER_FEATURE):
True, entities.append(OverpoweringBinarySensor(hass, unique_id, name, entry.data))
)
async_add_entities(entities, True)
class SecurityBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity): class SecurityBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
"""Representation of a BinarySensor which exposes the security state""" """Representation of a BinarySensor which exposes the security state"""
_security_state: bool
def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None: def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None:
"""Initialize the SecurityState Binary sensor""" """Initialize the SecurityState Binary sensor"""
super().__init__(hass, unique_id, entry_infos.get(CONF_NAME)) super().__init__(hass, unique_id, entry_infos.get(CONF_NAME))
@@ -73,8 +78,6 @@ class SecurityBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
class OverpoweringBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity): class OverpoweringBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
"""Representation of a BinarySensor which exposes the overpowering state""" """Representation of a BinarySensor which exposes the overpowering state"""
_security_state: bool
def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None: def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None:
"""Initialize the OverpoweringState Binary sensor""" """Initialize the OverpoweringState Binary sensor"""
super().__init__(hass, unique_id, entry_infos.get(CONF_NAME)) super().__init__(hass, unique_id, entry_infos.get(CONF_NAME))
@@ -93,14 +96,15 @@ class OverpoweringBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity
@property @property
def icon(self) -> str | None: def icon(self) -> str | None:
return "mdi:flash-alert" if self._attr_is_on:
return "mdi:flash-alert-outline"
else:
return "mdi:flash-outline"
class WindowBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity): class WindowBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
"""Representation of a BinarySensor which exposes the window state""" """Representation of a BinarySensor which exposes the window state"""
_security_state: bool
def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None: def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None:
"""Initialize the WindowState Binary sensor""" """Initialize the WindowState Binary sensor"""
super().__init__(hass, unique_id, entry_infos.get(CONF_NAME)) super().__init__(hass, unique_id, entry_infos.get(CONF_NAME))
@@ -119,14 +123,15 @@ class WindowBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
@property @property
def icon(self) -> str | None: def icon(self) -> str | None:
return "mdi:window-open-variant" if self._attr_is_on:
return "mdi:window-open-variant"
else:
return "mdi:window-closed-variant"
class MotionBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity): class MotionBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
"""Representation of a BinarySensor which exposes the motion state""" """Representation of a BinarySensor which exposes the motion state"""
_security_state: bool
def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None: def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None:
"""Initialize the MotionState Binary sensor""" """Initialize the MotionState Binary sensor"""
super().__init__(hass, unique_id, entry_infos.get(CONF_NAME)) super().__init__(hass, unique_id, entry_infos.get(CONF_NAME))
@@ -145,14 +150,15 @@ class MotionBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
@property @property
def icon(self) -> str | None: def icon(self) -> str | None:
return "mdi:motion-sensor" if self._attr_is_on:
return "mdi:motion-sensor"
else:
return "mdi:motion-sensor-off"
class PresenceBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity): class PresenceBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
"""Representation of a BinarySensor which exposes the presence state""" """Representation of a BinarySensor which exposes the presence state"""
_security_state: bool
def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None: def __init__(self, hass: HomeAssistant, unique_id, name, entry_infos) -> None:
"""Initialize the PresenceState Binary sensor""" """Initialize the PresenceState Binary sensor"""
super().__init__(hass, unique_id, entry_infos.get(CONF_NAME)) super().__init__(hass, unique_id, entry_infos.get(CONF_NAME))
@@ -162,6 +168,7 @@ class PresenceBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
@callback @callback
async def async_my_climate_changed(self, event: Event): async def async_my_climate_changed(self, event: Event):
"""Called when my climate have change""" """Called when my climate have change"""
_LOGGER.debug("%s - climate state change", event.origin.name) _LOGGER.debug("%s - climate state change", event.origin.name)
old_state = self._attr_is_on old_state = self._attr_is_on
self._attr_is_on = self.my_climate.presence_state == STATE_ON self._attr_is_on = self.my_climate.presence_state == STATE_ON
@@ -171,4 +178,7 @@ class PresenceBinarySensor(VersatileThermostatBaseEntity, BinarySensorEntity):
@property @property
def icon(self) -> str | None: def icon(self) -> str | None:
return "mdi:home-account" if self._attr_is_on:
return "mdi:home-account"
else:
return "mdi:nature-people"