From ca914d8e4f268c1dfd4ebc3e7a1b6fe5f8ad9354 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Manuel=20R=C3=BCger?= Date: Thu, 22 May 2025 11:51:28 +0200 Subject: [PATCH] switchbot_cloud: Add Smart Lock door and calibration state (#143695) * switchbot_cloud: Add Smart Lock door and calibration state * Incorporate review --- .../components/switchbot_cloud/__init__.py | 3 + .../switchbot_cloud/binary_sensor.py | 101 ++++++++++++++++++ 2 files changed, 104 insertions(+) create mode 100644 homeassistant/components/switchbot_cloud/binary_sensor.py diff --git a/homeassistant/components/switchbot_cloud/__init__.py b/homeassistant/components/switchbot_cloud/__init__.py index 6f36739e2fc..8074c882671 100644 --- a/homeassistant/components/switchbot_cloud/__init__.py +++ b/homeassistant/components/switchbot_cloud/__init__.py @@ -16,6 +16,7 @@ from .coordinator import SwitchBotCoordinator _LOGGER = getLogger(__name__) PLATFORMS: list[Platform] = [ + Platform.BINARY_SENSOR, Platform.BUTTON, Platform.CLIMATE, Platform.LOCK, @@ -29,6 +30,7 @@ PLATFORMS: list[Platform] = [ class SwitchbotDevices: """Switchbot devices data.""" + binary_sensors: list[Device] = field(default_factory=list) buttons: list[Device] = field(default_factory=list) climates: list[Remote] = field(default_factory=list) switches: list[Device | Remote] = field(default_factory=list) @@ -141,6 +143,7 @@ async def make_device_data( ) devices_data.locks.append((device, coordinator)) devices_data.sensors.append((device, coordinator)) + devices_data.binary_sensors.append((device, coordinator)) if isinstance(device, Device) and device.device_type in ["Bot"]: coordinator = await coordinator_for_device( diff --git a/homeassistant/components/switchbot_cloud/binary_sensor.py b/homeassistant/components/switchbot_cloud/binary_sensor.py new file mode 100644 index 00000000000..14278072c83 --- /dev/null +++ b/homeassistant/components/switchbot_cloud/binary_sensor.py @@ -0,0 +1,101 @@ +"""Support for SwitchBot Cloud binary sensors.""" + +from dataclasses import dataclass + +from switchbot_api import Device, SwitchBotAPI + +from homeassistant.components.binary_sensor import ( + BinarySensorDeviceClass, + BinarySensorEntity, + BinarySensorEntityDescription, +) +from homeassistant.config_entries import ConfigEntry +from homeassistant.const import EntityCategory +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback + +from . import SwitchbotCloudData +from .const import DOMAIN +from .coordinator import SwitchBotCoordinator +from .entity import SwitchBotCloudEntity + + +@dataclass(frozen=True) +class SwitchBotCloudBinarySensorEntityDescription(BinarySensorEntityDescription): + """Describes a Switchbot Cloud binary sensor.""" + + # Value or values to consider binary sensor to be "on" + on_value: bool | str = True + + +CALIBRATION_DESCRIPTION = SwitchBotCloudBinarySensorEntityDescription( + key="calibrate", + name="Calibration", + translation_key="calibration", + device_class=BinarySensorDeviceClass.PROBLEM, + entity_category=EntityCategory.DIAGNOSTIC, + on_value=False, +) + +DOOR_OPEN_DESCRIPTION = SwitchBotCloudBinarySensorEntityDescription( + key="doorState", + device_class=BinarySensorDeviceClass.DOOR, + on_value="opened", +) + +BINARY_SENSOR_DESCRIPTIONS_BY_DEVICE_TYPES = { + "Smart Lock": ( + CALIBRATION_DESCRIPTION, + DOOR_OPEN_DESCRIPTION, + ), + "Smart Lock Pro": ( + CALIBRATION_DESCRIPTION, + DOOR_OPEN_DESCRIPTION, + ), +} + + +async def async_setup_entry( + hass: HomeAssistant, + config: ConfigEntry, + async_add_entities: AddConfigEntryEntitiesCallback, +) -> None: + """Set up SwitchBot Cloud entry.""" + data: SwitchbotCloudData = hass.data[DOMAIN][config.entry_id] + + async_add_entities( + SwitchBotCloudBinarySensor(data.api, device, coordinator, description) + for device, coordinator in data.devices.binary_sensors + for description in BINARY_SENSOR_DESCRIPTIONS_BY_DEVICE_TYPES[ + device.device_type + ] + ) + + +class SwitchBotCloudBinarySensor(SwitchBotCloudEntity, BinarySensorEntity): + """Representation of a Switchbot binary sensor.""" + + entity_description: SwitchBotCloudBinarySensorEntityDescription + + def __init__( + self, + api: SwitchBotAPI, + device: Device, + coordinator: SwitchBotCoordinator, + description: SwitchBotCloudBinarySensorEntityDescription, + ) -> None: + """Initialize SwitchBot Cloud sensor entity.""" + super().__init__(api, device, coordinator) + self.entity_description = description + self._attr_unique_id = f"{device.device_id}_{description.key}" + + @property + def is_on(self) -> bool | None: + """Set attributes from coordinator data.""" + if not self.coordinator.data: + return None + + return ( + self.coordinator.data.get(self.entity_description.key) + == self.entity_description.on_value + )