Make flexit bacnet switch more generic and prepare for more switches (#…

…109154)

Make switch more generic and prepare for more switches

homeassistant/components/flexit_bacnet/switch.py

@@ -1,6 +1,6 @@
 """The Flexit Nordic (BACnet) integration."""
 import asyncio.exceptions
-from collections.abc import Callable
+from collections.abc import Awaitable, Callable
 from dataclasses import dataclass
 from typing import Any
 
@@ -27,6 +27,8 @@ class FlexitSwitchEntityDescription(SwitchEntityDescription):
     """Describes a Flexit switch entity."""
 
     is_on_fn: Callable[[FlexitBACnet], bool]
+    turn_on_fn: Callable[[FlexitBACnet], Awaitable[None]]
+    turn_off_fn: Callable[[FlexitBACnet], Awaitable[None]]
 
 
 SWITCHES: tuple[FlexitSwitchEntityDescription, ...] = (
@@ -35,6 +37,8 @@ class FlexitSwitchEntityDescription(SwitchEntityDescription):
         translation_key="electric_heater",
         icon="mdi:radiator",
         is_on_fn=lambda data: data.electric_heater,
+        turn_on_fn=lambda data: data.enable_electric_heater(),
+        turn_off_fn=lambda data: data.disable_electric_heater(),
     ),
 )
 
@@ -80,7 +84,7 @@ def is_on(self) -> bool:
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn electric heater on."""
         try:
-            await self.device.enable_electric_heater()
+            await self.entity_description.turn_on_fn(self.coordinator.data)
         except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
             raise HomeAssistantError from exc
         finally:
@@ -89,7 +93,7 @@ async def async_turn_on(self, **kwargs: Any) -> None:
     async def async_turn_off(self, **kwargs: Any) -> None:
         """Turn electric heater off."""
         try:
-            await self.device.disable_electric_heater()
+            await self.entity_description.turn_off_fn(self.coordinator.data)
         except (asyncio.exceptions.TimeoutError, ConnectionError, DecodingError) as exc:
             raise HomeAssistantError from exc
         finally: