jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-22 20:57:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use value_fn in Switcher sensor platform (#143711)

Browse Source
This commit is contained in:
Shay Levy 2025-04-26 14:34:13 +03:00 committed by GitHub
parent f1b3b0c155
commit eee18035cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 35 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
homeassistant/components/switcher_kis/sensor.py
View File

@ -2,7 +2,17 @@
from __future__ import annotations from __future__ import annotations
from aioswitcher.device import DeviceCategory from collections.abc import Callable
from dataclasses import dataclass
from typing import cast
from aioswitcher.device import (
DeviceCategory,
SwitcherBase,
SwitcherPowerBase,
SwitcherThermostatBase,
SwitcherTimedBase,
)
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
SensorDeviceClass, SensorDeviceClass,
@ -21,35 +31,48 @@ from .const import SIGNAL_DEVICE_ADD
from .coordinator import SwitcherDataUpdateCoordinator from .coordinator import SwitcherDataUpdateCoordinator
from .entity import SwitcherEntity from .entity import SwitcherEntity
POWER_SENSORS: list[SensorEntityDescription] = [
SensorEntityDescription( @dataclass(frozen=True, kw_only=True)
class SwitcherSensorEntityDescription(SensorEntityDescription):
"""Class to describe a Switcher sensor entity."""
value_fn: Callable[[SwitcherBase], StateType]
POWER_SENSORS: list[SwitcherSensorEntityDescription] = [
SwitcherSensorEntityDescription(
key="power_consumption", key="power_consumption",
native_unit_of_measurement=UnitOfPower.WATT, native_unit_of_measurement=UnitOfPower.WATT,
device_class=SensorDeviceClass.POWER, device_class=SensorDeviceClass.POWER,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
value_fn=lambda data: cast(SwitcherPowerBase, data).power_consumption,
), ),
SensorEntityDescription( SwitcherSensorEntityDescription(
key="electric_current", key="electric_current",
native_unit_of_measurement=UnitOfElectricCurrent.AMPERE, native_unit_of_measurement=UnitOfElectricCurrent.AMPERE,
device_class=SensorDeviceClass.CURRENT, device_class=SensorDeviceClass.CURRENT,
state_class=SensorStateClass.MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
value_fn=lambda data: cast(SwitcherPowerBase, data).electric_current,
), ),
] ]
TIME_SENSORS: list[SensorEntityDescription] = [ TIME_SENSORS: list[SwitcherSensorEntityDescription] = [
SensorEntityDescription( SwitcherSensorEntityDescription(
key="remaining_time", key="remaining_time",
translation_key="remaining_time", translation_key="remaining_time",
value_fn=lambda data: cast(SwitcherTimedBase, data).remaining_time,
), ),
SensorEntityDescription( SwitcherSensorEntityDescription(
key="auto_off_set", key="auto_off_set",
translation_key="auto_shutdown", translation_key="auto_shutdown",
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
value_fn=lambda data: cast(SwitcherTimedBase, data).auto_shutdown,
), ),
] ]
TEMPERATURE_SENSORS: list[SensorEntityDescription] = [ TEMPERATURE_SENSORS: list[SwitcherSensorEntityDescription] = [
SensorEntityDescription( SwitcherSensorEntityDescription(
key="temperature", key="temperature",
translation_key="temperature", translation_key="temperature",
value_fn=lambda data: cast(SwitcherThermostatBase, data).temperature,
), ),
] ]
@ -95,11 +118,11 @@ class SwitcherSensorEntity(SwitcherEntity, SensorEntity):
def __init__( def __init__(
self, self,
coordinator: SwitcherDataUpdateCoordinator, coordinator: SwitcherDataUpdateCoordinator,
description: SensorEntityDescription, description: SwitcherSensorEntityDescription,
) -> None: ) -> None:
"""Initialize the entity.""" """Initialize the entity."""
super().__init__(coordinator) super().__init__(coordinator)
self.entity_description = description self.entity_description: SwitcherSensorEntityDescription = description
self._attr_unique_id = ( self._attr_unique_id = (
f"{coordinator.device_id}-{coordinator.mac_address}-{description.key}" f"{coordinator.device_id}-{coordinator.mac_address}-{description.key}"
@ -108,4 +131,4 @@ class SwitcherSensorEntity(SwitcherEntity, SensorEntity):
@property @property
def native_value(self) -> StateType: def native_value(self) -> StateType:
"""Return value of sensor.""" """Return value of sensor."""
return getattr(self.coordinator.data, self.entity_description.key) # type: ignore[no-any-return] return self.entity_description.value_fn(self.coordinator.data)