evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add value_fn to switch entity description in pyLoad (#120569)

Browse source
This commit is contained in:
Mr. Bubbles 2024-06-26 15:09:42 +02:00 • committed by GitHub
parent 43d686e0f1
commit af9b4b98ca
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
homeassistant/components/pyload/switch.py
View file

@ -18,6 +18,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import PyLoadConfigEntry
from .coordinator import pyLoadData
from .entity import BasePyLoadEntity
@ -35,6 +36,7 @@ class PyLoadSwitchEntityDescription(SwitchEntityDescription):
turn_on_fn: Callable[[PyLoadAPI], Awaitable[Any]]
turn_off_fn: Callable[[PyLoadAPI], Awaitable[Any]]
toggle_fn: Callable[[PyLoadAPI], Awaitable[Any]]
value_fn: Callable[[pyLoadData], bool]
SENSOR_DESCRIPTIONS: tuple[PyLoadSwitchEntityDescription, ...] = (
@ -45,6 +47,7 @@ SENSOR_DESCRIPTIONS: tuple[PyLoadSwitchEntityDescription, ...] = (
turn_on_fn=lambda api: api.unpause(),
turn_off_fn=lambda api: api.pause(),
toggle_fn=lambda api: api.toggle_pause(),
value_fn=lambda data: data.download,
),
PyLoadSwitchEntityDescription(
key=PyLoadSwitch.RECONNECT,
@ -53,6 +56,7 @@ SENSOR_DESCRIPTIONS: tuple[PyLoadSwitchEntityDescription, ...] = (
turn_on_fn=lambda api: api.toggle_reconnect(),
turn_off_fn=lambda api: api.toggle_reconnect(),
toggle_fn=lambda api: api.toggle_reconnect(),
value_fn=lambda data: data.reconnect,
),
)
@ -80,7 +84,9 @@ class PyLoadSwitchEntity(BasePyLoadEntity, SwitchEntity):
@property
def is_on(self) -> bool | None:
"""Return the state of the device."""
return getattr(self.coordinator.data, self.entity_description.key)
return self.entity_description.value_fn(
self.coordinator.data,
)
async def async_turn_on(self, **kwargs: Any) -> None:
"""Turn the entity on."""