Add Vallox bypass locking switch entity (#75857)

* Switch entity

* adding missing function for tests

* Apply suggestions from code review

Co-authored-by: Sebastian Lövdahl <slovdahl@hibox.fi>

* Review

* fix

* Update homeassistant/components/vallox/switch.py

Co-authored-by: Andre Richter <andre-richter@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* Adding a separate parameter for the expected state

Co-authored-by: Sebastian Lövdahl <slovdahl@hibox.fi>
Co-authored-by: Andre Richter <andre-richter@users.noreply.github.com>
Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

homeassistant/components/vallox/switch.py

@@ -0,0 +1,105 @@
+"""Support for Vallox ventilation unit switches."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any
+
+from vallox_websocket_api import Vallox
+
+from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from . import ValloxDataUpdateCoordinator, ValloxEntity
+from .const import DOMAIN
+
+
+class ValloxSwitchEntity(ValloxEntity, SwitchEntity):
+    """Representation of a Vallox switch."""
+
+    entity_description: ValloxSwitchEntityDescription
+    _attr_entity_category = EntityCategory.CONFIG
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        name: str,
+        coordinator: ValloxDataUpdateCoordinator,
+        description: ValloxSwitchEntityDescription,
+        client: Vallox,
+    ) -> None:
+        """Initialize the Vallox switch."""
+        super().__init__(name, coordinator)
+
+        self.entity_description = description
+
+        self._attr_unique_id = f"{self._device_uuid}-{description.key}"
+        self._client = client
+
+    @property
+    def is_on(self) -> bool | None:
+        """Return true if the switch is on."""
+        if (
+            value := self.coordinator.data.get_metric(
+                self.entity_description.metric_key
+            )
+        ) is None:
+            return None
+        return value == 1
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Turn on."""
+        await self._set_value(True)
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off."""
+        await self._set_value(False)
+
+    async def _set_value(self, value: bool) -> None:
+        """Update the current value."""
+        metric_key = self.entity_description.metric_key
+        await self._client.set_values({metric_key: 1 if value else 0})
+        await self.coordinator.async_request_refresh()
+
+
+@dataclass
+class ValloxMetricKeyMixin:
+    """Dataclass to allow defining metric_key without a default value."""
+
+    metric_key: str
+
+
+@dataclass
+class ValloxSwitchEntityDescription(SwitchEntityDescription, ValloxMetricKeyMixin):
+    """Describes Vallox switch entity."""
+
+
+SWITCH_ENTITIES: tuple[ValloxSwitchEntityDescription, ...] = (
+    ValloxSwitchEntityDescription(
+        key="bypass_locked",
+        name="Bypass locked",
+        icon="mdi:arrow-horizontal-lock",
+        metric_key="A_CYC_BYPASS_LOCKED",
+    ),
+)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up the switches."""
+
+    data = hass.data[DOMAIN][entry.entry_id]
+    client = data["client"]
+    client.set_settable_address("A_CYC_BYPASS_LOCKED", int)
+
+    async_add_entities(
+        [
+            ValloxSwitchEntity(data["name"], data["coordinator"], description, client)
+            for description in SWITCH_ENTITIES
+        ]
+    )