Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

Make switch more generic and prepare for more switches
pull/108090/head^2
Jonas Fors Lellky 2024-01-31 13:19:23 +01:00 committed by GitHub
parent 4bad88b42c
commit a61b18155b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/flexit_bacnet/switch.py
View File

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