Add support for SwitchBot Relay Switch 1 and Relay Switch 1PM (#132327)

homeassistant/components/switchbot_cloud/__init__.py

@@ -75,9 +75,11 @@ def make_device_data(
             )
         if (
             isinstance(device, Device)
-            and device.device_type.startswith("Plug")
-            or isinstance(device, Remote)
-        ):
+            and (
+                device.device_type.startswith("Plug")
+                or device.device_type in ["Relay Switch 1PM", "Relay Switch 1"]
+            )
+        ) or isinstance(device, Remote):
             devices_data.switches.append(
                 prepare_device(hass, api, device, coordinators_by_id)
             )
@@ -88,6 +90,7 @@ def make_device_data(
             "Hub 2",
             "MeterPro",
             "MeterPro(CO2)",
+            "Relay Switch 1PM",
         ]:
             devices_data.sensors.append(
                 prepare_device(hass, api, device, coordinators_by_id)

homeassistant/components/switchbot_cloud/sensor.py

@@ -12,6 +12,9 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import (
     CONCENTRATION_PARTS_PER_MILLION,
     PERCENTAGE,
+    UnitOfElectricCurrent,
+    UnitOfElectricPotential,
+    UnitOfPower,
     UnitOfTemperature,
 )
 from homeassistant.core import HomeAssistant, callback
@@ -26,38 +29,97 @@ SENSOR_TYPE_TEMPERATURE = "temperature"
 SENSOR_TYPE_HUMIDITY = "humidity"
 SENSOR_TYPE_BATTERY = "battery"
 SENSOR_TYPE_CO2 = "CO2"
+SENSOR_TYPE_POWER = "power"
+SENSOR_TYPE_VOLTAGE = "voltage"
+SENSOR_TYPE_CURRENT = "electricCurrent"
 
-METER_PLUS_SENSOR_DESCRIPTIONS = (
-    SensorEntityDescription(
+TEMPERATURE_DESCRIPTION = SensorEntityDescription(
     key=SENSOR_TYPE_TEMPERATURE,
     device_class=SensorDeviceClass.TEMPERATURE,
     state_class=SensorStateClass.MEASUREMENT,
     native_unit_of_measurement=UnitOfTemperature.CELSIUS,
-    ),
-    SensorEntityDescription(
+)
+
+HUMIDITY_DESCRIPTION = SensorEntityDescription(
     key=SENSOR_TYPE_HUMIDITY,
     device_class=SensorDeviceClass.HUMIDITY,
     state_class=SensorStateClass.MEASUREMENT,
     native_unit_of_measurement=PERCENTAGE,
-    ),
-    SensorEntityDescription(
+)
+
+BATTERY_DESCRIPTION = SensorEntityDescription(
     key=SENSOR_TYPE_BATTERY,
     device_class=SensorDeviceClass.BATTERY,
     state_class=SensorStateClass.MEASUREMENT,
     native_unit_of_measurement=PERCENTAGE,
-    ),
 )
 
-METER_PRO_CO2_SENSOR_DESCRIPTIONS = (
-    *METER_PLUS_SENSOR_DESCRIPTIONS,
-    SensorEntityDescription(
-        key=SENSOR_TYPE_CO2,
-        native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+POWER_DESCRIPTION = SensorEntityDescription(
+    key=SENSOR_TYPE_POWER,
+    device_class=SensorDeviceClass.POWER,
     state_class=SensorStateClass.MEASUREMENT,
-        device_class=SensorDeviceClass.CO2,
-    ),
+    native_unit_of_measurement=UnitOfPower.WATT,
 )
 
+VOLATGE_DESCRIPTION = SensorEntityDescription(
+    key=SENSOR_TYPE_VOLTAGE,
+    device_class=SensorDeviceClass.VOLTAGE,
+    state_class=SensorStateClass.MEASUREMENT,
+    native_unit_of_measurement=UnitOfElectricPotential.VOLT,
+)
+
+CURRENT_DESCRIPTION = SensorEntityDescription(
+    key=SENSOR_TYPE_CURRENT,
+    device_class=SensorDeviceClass.CURRENT,
+    state_class=SensorStateClass.MEASUREMENT,
+    native_unit_of_measurement=UnitOfElectricCurrent.MILLIAMPERE,
+)
+
+CO2_DESCRIPTION = SensorEntityDescription(
+    key=SENSOR_TYPE_CO2,
+    device_class=SensorDeviceClass.CO2,
+    state_class=SensorStateClass.MEASUREMENT,
+    native_unit_of_measurement=CONCENTRATION_PARTS_PER_MILLION,
+)
+
+SENSOR_DESCRIPTIONS_BY_DEVICE_TYPES = {
+    "Meter": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+        BATTERY_DESCRIPTION,
+    ),
+    "MeterPlus": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+        BATTERY_DESCRIPTION,
+    ),
+    "WoIOSensor": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+        BATTERY_DESCRIPTION,
+    ),
+    "Relay Switch 1PM": (
+        POWER_DESCRIPTION,
+        VOLATGE_DESCRIPTION,
+        CURRENT_DESCRIPTION,
+    ),
+    "Hub 2": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+    ),
+    "MeterPro": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+        BATTERY_DESCRIPTION,
+    ),
+    "MeterPro(CO2)": (
+        TEMPERATURE_DESCRIPTION,
+        HUMIDITY_DESCRIPTION,
+        BATTERY_DESCRIPTION,
+        CO2_DESCRIPTION,
+    ),
+}
+
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -70,11 +132,7 @@ async def async_setup_entry(
     async_add_entities(
         SwitchBotCloudSensor(data.api, device, coordinator, description)
         for device, coordinator in data.devices.sensors
-        for description in (
-            METER_PRO_CO2_SENSOR_DESCRIPTIONS
-            if device.device_type == "MeterPro(CO2)"
-            else METER_PLUS_SENSOR_DESCRIPTIONS
-        )
+        for description in SENSOR_DESCRIPTIONS_BY_DEVICE_TYPES[device.device_type]
     )
 
 

homeassistant/components/switchbot_cloud/switch.py

@@ -69,6 +69,18 @@ class SwitchBotCloudPlugSwitch(SwitchBotCloudSwitch):
     _attr_device_class = SwitchDeviceClass.OUTLET
 
 
+class SwitchBotCloudRelaySwitchSwitch(SwitchBotCloudSwitch):
+    """Representation of a SwitchBot relay switch."""
+
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        """Handle updated data from the coordinator."""
+        if not self.coordinator.data:
+            return
+        self._attr_is_on = self.coordinator.data.get("switchStatus") == 1
+        self.async_write_ha_state()
+
+
 @callback
 def _async_make_entity(
     api: SwitchBotAPI, device: Device | Remote, coordinator: SwitchBotCoordinator
@@ -78,4 +90,9 @@ def _async_make_entity(
         return SwitchBotCloudRemoteSwitch(api, device, coordinator)
     if "Plug" in device.device_type:
         return SwitchBotCloudPlugSwitch(api, device, coordinator)
+    if device.device_type in [
+        "Relay Switch 1PM",
+        "Relay Switch 1",
+    ]:
+        return SwitchBotCloudRelaySwitchSwitch(api, device, coordinator)
     raise NotImplementedError(f"Unsupported device type: {device.device_type}")

tests/components/switchbot_cloud/test_switch.py

@@ -0,0 +1,56 @@
+"""Test for the switchbot_cloud relay switch."""
+
+from unittest.mock import patch
+
+from switchbot_api import Device
+
+from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
+from homeassistant.components.switchbot_cloud import SwitchBotAPI
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.const import (
+    ATTR_ENTITY_ID,
+    SERVICE_TURN_OFF,
+    SERVICE_TURN_ON,
+    STATE_OFF,
+    STATE_ON,
+)
+from homeassistant.core import HomeAssistant
+
+from . import configure_integration
+
+
+async def test_relay_switch(
+    hass: HomeAssistant, mock_list_devices, mock_get_status
+) -> None:
+    """Test turn on and turn off."""
+    mock_list_devices.return_value = [
+        Device(
+            deviceId="relay-switch-id-1",
+            deviceName="relay-switch-1",
+            deviceType="Relay Switch 1",
+            hubDeviceId="test-hub-id",
+        ),
+    ]
+
+    mock_get_status.return_value = {"switchStatus": 0}
+
+    entry = configure_integration(hass)
+    await hass.config_entries.async_setup(entry.entry_id)
+    await hass.async_block_till_done()
+
+    assert entry.state is ConfigEntryState.LOADED
+
+    entity_id = "switch.relay_switch_1"
+    assert hass.states.get(entity_id).state == STATE_OFF
+
+    with patch.object(SwitchBotAPI, "send_command"):
+        await hass.services.async_call(
+            SWITCH_DOMAIN, SERVICE_TURN_ON, {ATTR_ENTITY_ID: entity_id}, blocking=True
+        )
+    assert hass.states.get(entity_id).state == STATE_ON
+
+    with patch.object(SwitchBotAPI, "send_command"):
+        await hass.services.async_call(
+            SWITCH_DOMAIN, SERVICE_TURN_OFF, {ATTR_ENTITY_ID: entity_id}, blocking=True
+        )
+    assert hass.states.get(entity_id).state == STATE_OFF