Add power binary sensor for SmartThings (#141126)

This commit is contained in:
Joost Lekkerkerker 2025-03-22 17:59:15 +01:00 committed by GitHub
parent fc0dbcd600
commit 765691c84d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 216 additions and 1 deletions

View File

@ -4,7 +4,7 @@ from __future__ import annotations
from dataclasses import dataclass
from pysmartthings import Attribute, Capability, SmartThings
from pysmartthings import Attribute, Capability, Category, SmartThings
from homeassistant.components.automation import automations_with_entity
from homeassistant.components.binary_sensor import (
@ -33,6 +33,7 @@ class SmartThingsBinarySensorEntityDescription(BinarySensorEntityDescription):
"""Describe a SmartThings binary sensor entity."""
is_on_key: str
category: set[Category] | None = None
CAPABILITY_TO_SENSORS: dict[
@ -96,6 +97,14 @@ CAPABILITY_TO_SENSORS: dict[
is_on_key="detected",
)
},
Capability.SWITCH: {
Attribute.SWITCH: SmartThingsBinarySensorEntityDescription(
key=Attribute.SWITCH,
device_class=BinarySensorDeviceClass.POWER,
is_on_key="on",
category={Category.DRYER, Category.WASHER},
)
},
Capability.TAMPER_ALERT: {
Attribute.TAMPER: SmartThingsBinarySensorEntityDescription(
key=Attribute.TAMPER,
@ -122,6 +131,16 @@ CAPABILITY_TO_SENSORS: dict[
}
def get_main_component_category(
device: FullDevice,
) -> Category | str:
"""Get the main component of a device."""
main = next(
component for component in device.device.components if component.id == MAIN
)
return main.user_category or main.manufacturer_category
async def async_setup_entry(
hass: HomeAssistant,
entry: SmartThingsConfigEntry,
@ -141,6 +160,10 @@ async def async_setup_entry(
for capability, attribute_map in CAPABILITY_TO_SENSORS.items()
if capability in device.status[MAIN]
for attribute, description in attribute_map.items()
if (
not description.category
or get_main_component_category(device) in description.category
)
)

View File

@ -614,6 +614,54 @@
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wd_000001][binary_sensor.dryer_power-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': None,
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'binary_sensor',
'entity_category': None,
'entity_id': 'binary_sensor.dryer_power',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <BinarySensorDeviceClass.POWER: 'power'>,
'original_icon': None,
'original_name': 'Power',
'platform': 'smartthings',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': None,
'unique_id': '02f7256e-8353-5bdd-547f-bd5b1647e01b.switch',
'unit_of_measurement': None,
})
# ---
# name: test_all_entities[da_wm_wd_000001][binary_sensor.dryer_power-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'power',
'friendly_name': 'Dryer Power',
}),
'context': <ANY>,
'entity_id': 'binary_sensor.dryer_power',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wd_000001][binary_sensor.dryer_remote_control-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
@ -708,6 +756,54 @@
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wd_000001_1][binary_sensor.seca_roupa_power-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': None,
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'binary_sensor',
'entity_category': None,
'entity_id': 'binary_sensor.seca_roupa_power',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <BinarySensorDeviceClass.POWER: 'power'>,
'original_icon': None,
'original_name': 'Power',
'platform': 'smartthings',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': None,
'unique_id': '3a6c4e05-811d-5041-e956-3d04c424cbcd.switch',
'unit_of_measurement': None,
})
# ---
# name: test_all_entities[da_wm_wd_000001_1][binary_sensor.seca_roupa_power-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'power',
'friendly_name': 'Seca-Roupa Power',
}),
'context': <ANY>,
'entity_id': 'binary_sensor.seca_roupa_power',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wd_000001_1][binary_sensor.seca_roupa_remote_control-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
@ -802,6 +898,54 @@
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wm_000001][binary_sensor.washer_power-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': None,
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'binary_sensor',
'entity_category': None,
'entity_id': 'binary_sensor.washer_power',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <BinarySensorDeviceClass.POWER: 'power'>,
'original_icon': None,
'original_name': 'Power',
'platform': 'smartthings',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': None,
'unique_id': 'f984b91d-f250-9d42-3436-33f09a422a47.switch',
'unit_of_measurement': None,
})
# ---
# name: test_all_entities[da_wm_wm_000001][binary_sensor.washer_power-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'power',
'friendly_name': 'Washer Power',
}),
'context': <ANY>,
'entity_id': 'binary_sensor.washer_power',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wm_000001][binary_sensor.washer_remote_control-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
@ -896,6 +1040,54 @@
'state': 'off',
})
# ---
# name: test_all_entities[da_wm_wm_000001_1][binary_sensor.washing_machine_power-entry]
EntityRegistryEntrySnapshot({
'aliases': set({
}),
'area_id': None,
'capabilities': None,
'config_entry_id': <ANY>,
'config_subentry_id': <ANY>,
'device_class': None,
'device_id': <ANY>,
'disabled_by': None,
'domain': 'binary_sensor',
'entity_category': None,
'entity_id': 'binary_sensor.washing_machine_power',
'has_entity_name': True,
'hidden_by': None,
'icon': None,
'id': <ANY>,
'labels': set({
}),
'name': None,
'options': dict({
}),
'original_device_class': <BinarySensorDeviceClass.POWER: 'power'>,
'original_icon': None,
'original_name': 'Power',
'platform': 'smartthings',
'previous_unique_id': None,
'supported_features': 0,
'translation_key': None,
'unique_id': '63803fae-cbed-f356-a063-2cf148ae3ca7.switch',
'unit_of_measurement': None,
})
# ---
# name: test_all_entities[da_wm_wm_000001_1][binary_sensor.washing_machine_power-state]
StateSnapshot({
'attributes': ReadOnlyDict({
'device_class': 'power',
'friendly_name': 'Washing Machine Power',
}),
'context': <ANY>,
'entity_id': 'binary_sensor.washing_machine_power',
'last_changed': <ANY>,
'last_reported': <ANY>,
'last_updated': <ANY>,
'state': 'on',
})
# ---
# name: test_all_entities[da_wm_wm_000001_1][binary_sensor.washing_machine_remote_control-entry]
EntityRegistryEntrySnapshot({
'aliases': set({