mirror of
https://github.com/home-assistant/core.git
synced 2025-07-14 16:57:10 +00:00
Melcloud use NamedTuple (#53234)
Co-authored-by: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
This commit is contained in:
parent
6be30b0289
commit
a14bde8187
@ -1,4 +1,8 @@
|
|||||||
"""Support for MelCloud device sensors."""
|
"""Support for MelCloud device sensors."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from typing import Any, Callable, NamedTuple
|
||||||
|
|
||||||
from pymelcloud import DEVICE_TYPE_ATA, DEVICE_TYPE_ATW
|
from pymelcloud import DEVICE_TYPE_ATA, DEVICE_TYPE_ATW
|
||||||
from pymelcloud.atw_device import Zone
|
from pymelcloud.atw_device import Zone
|
||||||
|
|
||||||
@ -8,83 +12,85 @@ from homeassistant.components.sensor import (
|
|||||||
STATE_CLASS_MEASUREMENT,
|
STATE_CLASS_MEASUREMENT,
|
||||||
SensorEntity,
|
SensorEntity,
|
||||||
)
|
)
|
||||||
from homeassistant.const import (
|
from homeassistant.const import ENERGY_KILO_WATT_HOUR, TEMP_CELSIUS
|
||||||
ATTR_DEVICE_CLASS,
|
|
||||||
ATTR_ICON,
|
|
||||||
ENERGY_KILO_WATT_HOUR,
|
|
||||||
TEMP_CELSIUS,
|
|
||||||
)
|
|
||||||
from homeassistant.util import dt as dt_util
|
from homeassistant.util import dt as dt_util
|
||||||
|
|
||||||
from . import MelCloudDevice
|
from . import MelCloudDevice
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
|
||||||
ATTR_MEASUREMENT_NAME = "measurement_name"
|
|
||||||
ATTR_UNIT = "unit"
|
|
||||||
ATTR_VALUE_FN = "value_fn"
|
|
||||||
ATTR_ENABLED_FN = "enabled"
|
|
||||||
|
|
||||||
ATA_SENSORS = {
|
class SensorMetadata(NamedTuple):
|
||||||
"room_temperature": {
|
"""Metadata for an individual sensor."""
|
||||||
ATTR_MEASUREMENT_NAME: "Room Temperature",
|
|
||||||
ATTR_ICON: "mdi:thermometer",
|
measurement_name: str
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
icon: str
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
unit: str
|
||||||
ATTR_VALUE_FN: lambda x: x.device.room_temperature,
|
device_class: str
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
value_fn: Callable[[Any], float]
|
||||||
},
|
enabled: Callable[[Any], bool]
|
||||||
"energy": {
|
|
||||||
ATTR_MEASUREMENT_NAME: "Energy",
|
|
||||||
ATTR_ICON: "mdi:factory",
|
ATA_SENSORS: dict[str, SensorMetadata] = {
|
||||||
ATTR_UNIT: ENERGY_KILO_WATT_HOUR,
|
"room_temperature": SensorMetadata(
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_ENERGY,
|
"Room Temperature",
|
||||||
ATTR_VALUE_FN: lambda x: x.device.total_energy_consumed,
|
icon="mdi:thermometer",
|
||||||
ATTR_ENABLED_FN: lambda x: x.device.has_energy_consumed_meter,
|
unit=TEMP_CELSIUS,
|
||||||
},
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
|
value_fn=lambda x: x.device.room_temperature,
|
||||||
|
enabled=lambda x: True,
|
||||||
|
),
|
||||||
|
"energy": SensorMetadata(
|
||||||
|
"Energy",
|
||||||
|
icon="mdi:factory",
|
||||||
|
unit=ENERGY_KILO_WATT_HOUR,
|
||||||
|
device_class=DEVICE_CLASS_ENERGY,
|
||||||
|
value_fn=lambda x: x.device.total_energy_consumed,
|
||||||
|
enabled=lambda x: x.device.has_energy_consumed_meter,
|
||||||
|
),
|
||||||
}
|
}
|
||||||
ATW_SENSORS = {
|
ATW_SENSORS: dict[str, SensorMetadata] = {
|
||||||
"outside_temperature": {
|
"outside_temperature": SensorMetadata(
|
||||||
ATTR_MEASUREMENT_NAME: "Outside Temperature",
|
"Outside Temperature",
|
||||||
ATTR_ICON: "mdi:thermometer",
|
icon="mdi:thermometer",
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
unit=TEMP_CELSIUS,
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
ATTR_VALUE_FN: lambda x: x.device.outside_temperature,
|
value_fn=lambda x: x.device.outside_temperature,
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
enabled=lambda x: True,
|
||||||
},
|
),
|
||||||
"tank_temperature": {
|
"tank_temperature": SensorMetadata(
|
||||||
ATTR_MEASUREMENT_NAME: "Tank Temperature",
|
"Tank Temperature",
|
||||||
ATTR_ICON: "mdi:thermometer",
|
icon="mdi:thermometer",
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
unit=TEMP_CELSIUS,
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
ATTR_VALUE_FN: lambda x: x.device.tank_temperature,
|
value_fn=lambda x: x.device.tank_temperature,
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
enabled=lambda x: True,
|
||||||
},
|
),
|
||||||
}
|
}
|
||||||
ATW_ZONE_SENSORS = {
|
ATW_ZONE_SENSORS: dict[str, SensorMetadata] = {
|
||||||
"room_temperature": {
|
"room_temperature": SensorMetadata(
|
||||||
ATTR_MEASUREMENT_NAME: "Room Temperature",
|
"Room Temperature",
|
||||||
ATTR_ICON: "mdi:thermometer",
|
icon="mdi:thermometer",
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
unit=TEMP_CELSIUS,
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
ATTR_VALUE_FN: lambda zone: zone.room_temperature,
|
value_fn=lambda zone: zone.room_temperature,
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
enabled=lambda x: True,
|
||||||
},
|
),
|
||||||
"flow_temperature": {
|
"flow_temperature": SensorMetadata(
|
||||||
ATTR_MEASUREMENT_NAME: "Flow Temperature",
|
"Flow Temperature",
|
||||||
ATTR_ICON: "mdi:thermometer",
|
icon="mdi:thermometer",
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
unit=TEMP_CELSIUS,
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
ATTR_VALUE_FN: lambda zone: zone.flow_temperature,
|
value_fn=lambda zone: zone.flow_temperature,
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
enabled=lambda x: True,
|
||||||
},
|
),
|
||||||
"return_temperature": {
|
"return_temperature": SensorMetadata(
|
||||||
ATTR_MEASUREMENT_NAME: "Flow Return Temperature",
|
"Flow Return Temperature",
|
||||||
ATTR_ICON: "mdi:thermometer",
|
icon="mdi:thermometer",
|
||||||
ATTR_UNIT: TEMP_CELSIUS,
|
unit=TEMP_CELSIUS,
|
||||||
ATTR_DEVICE_CLASS: DEVICE_CLASS_TEMPERATURE,
|
device_class=DEVICE_CLASS_TEMPERATURE,
|
||||||
ATTR_VALUE_FN: lambda zone: zone.return_temperature,
|
value_fn=lambda zone: zone.return_temperature,
|
||||||
ATTR_ENABLED_FN: lambda x: True,
|
enabled=lambda x: True,
|
||||||
},
|
),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -93,23 +99,23 @@ async def async_setup_entry(hass, entry, async_add_entities):
|
|||||||
mel_devices = hass.data[DOMAIN].get(entry.entry_id)
|
mel_devices = hass.data[DOMAIN].get(entry.entry_id)
|
||||||
async_add_entities(
|
async_add_entities(
|
||||||
[
|
[
|
||||||
MelDeviceSensor(mel_device, measurement, definition)
|
MelDeviceSensor(mel_device, measurement, metadata)
|
||||||
for measurement, definition in ATA_SENSORS.items()
|
for measurement, metadata in ATA_SENSORS.items()
|
||||||
for mel_device in mel_devices[DEVICE_TYPE_ATA]
|
for mel_device in mel_devices[DEVICE_TYPE_ATA]
|
||||||
if definition[ATTR_ENABLED_FN](mel_device)
|
if metadata.enabled(mel_device)
|
||||||
]
|
]
|
||||||
+ [
|
+ [
|
||||||
MelDeviceSensor(mel_device, measurement, definition)
|
MelDeviceSensor(mel_device, measurement, metadata)
|
||||||
for measurement, definition in ATW_SENSORS.items()
|
for measurement, metadata in ATW_SENSORS.items()
|
||||||
for mel_device in mel_devices[DEVICE_TYPE_ATW]
|
for mel_device in mel_devices[DEVICE_TYPE_ATW]
|
||||||
if definition[ATTR_ENABLED_FN](mel_device)
|
if metadata.enabled(mel_device)
|
||||||
]
|
]
|
||||||
+ [
|
+ [
|
||||||
AtwZoneSensor(mel_device, zone, measurement, definition)
|
AtwZoneSensor(mel_device, zone, measurement, metadata)
|
||||||
for mel_device in mel_devices[DEVICE_TYPE_ATW]
|
for mel_device in mel_devices[DEVICE_TYPE_ATW]
|
||||||
for zone in mel_device.device.zones
|
for zone in mel_device.device.zones
|
||||||
for measurement, definition, in ATW_ZONE_SENSORS.items()
|
for measurement, metadata, in ATW_ZONE_SENSORS.items()
|
||||||
if definition[ATTR_ENABLED_FN](zone)
|
if metadata.enabled(zone)
|
||||||
],
|
],
|
||||||
True,
|
True,
|
||||||
)
|
)
|
||||||
@ -118,25 +124,25 @@ async def async_setup_entry(hass, entry, async_add_entities):
|
|||||||
class MelDeviceSensor(SensorEntity):
|
class MelDeviceSensor(SensorEntity):
|
||||||
"""Representation of a Sensor."""
|
"""Representation of a Sensor."""
|
||||||
|
|
||||||
def __init__(self, api: MelCloudDevice, measurement, definition):
|
def __init__(self, api: MelCloudDevice, measurement, metadata: SensorMetadata):
|
||||||
"""Initialize the sensor."""
|
"""Initialize the sensor."""
|
||||||
self._api = api
|
self._api = api
|
||||||
self._def = definition
|
self._metadata = metadata
|
||||||
|
|
||||||
self._attr_device_class = definition[ATTR_DEVICE_CLASS]
|
self._attr_device_class = metadata.device_class
|
||||||
self._attr_icon = definition[ATTR_ICON]
|
self._attr_icon = metadata.icon
|
||||||
self._attr_name = f"{api.name} {definition[ATTR_MEASUREMENT_NAME]}"
|
self._attr_name = f"{api.name} {metadata.measurement_name}"
|
||||||
self._attr_unique_id = f"{api.device.serial}-{api.device.mac}-{measurement}"
|
self._attr_unique_id = f"{api.device.serial}-{api.device.mac}-{measurement}"
|
||||||
self._attr_unit_of_measurement = definition[ATTR_UNIT]
|
self._attr_unit_of_measurement = metadata.unit
|
||||||
self._attr_state_class = STATE_CLASS_MEASUREMENT
|
self._attr_state_class = STATE_CLASS_MEASUREMENT
|
||||||
|
|
||||||
if self.device_class == DEVICE_CLASS_ENERGY:
|
if metadata.device_class == DEVICE_CLASS_ENERGY:
|
||||||
self._attr_last_reset = dt_util.utc_from_timestamp(0)
|
self._attr_last_reset = dt_util.utc_from_timestamp(0)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def state(self):
|
def state(self):
|
||||||
"""Return the state of the sensor."""
|
"""Return the state of the sensor."""
|
||||||
return self._def[ATTR_VALUE_FN](self._api)
|
return self._metadata.value_fn(self._api)
|
||||||
|
|
||||||
async def async_update(self):
|
async def async_update(self):
|
||||||
"""Retrieve latest state."""
|
"""Retrieve latest state."""
|
||||||
@ -151,17 +157,19 @@ class MelDeviceSensor(SensorEntity):
|
|||||||
class AtwZoneSensor(MelDeviceSensor):
|
class AtwZoneSensor(MelDeviceSensor):
|
||||||
"""Air-to-Air device sensor."""
|
"""Air-to-Air device sensor."""
|
||||||
|
|
||||||
def __init__(self, api: MelCloudDevice, zone: Zone, measurement, definition):
|
def __init__(
|
||||||
|
self, api: MelCloudDevice, zone: Zone, measurement, metadata: SensorMetadata
|
||||||
|
):
|
||||||
"""Initialize the sensor."""
|
"""Initialize the sensor."""
|
||||||
if zone.zone_index == 1:
|
if zone.zone_index == 1:
|
||||||
full_measurement = measurement
|
full_measurement = measurement
|
||||||
else:
|
else:
|
||||||
full_measurement = f"{measurement}-zone-{zone.zone_index}"
|
full_measurement = f"{measurement}-zone-{zone.zone_index}"
|
||||||
super().__init__(api, full_measurement, definition)
|
super().__init__(api, full_measurement, metadata)
|
||||||
self._zone = zone
|
self._zone = zone
|
||||||
self._attr_name = f"{api.name} {zone.name} {definition[ATTR_MEASUREMENT_NAME]}"
|
self._attr_name = f"{api.name} {zone.name} {metadata.measurement_name}"
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def state(self):
|
def state(self):
|
||||||
"""Return zone based state."""
|
"""Return zone based state."""
|
||||||
return self._def[ATTR_VALUE_FN](self._zone)
|
return self._metadata.value_fn(self._zone)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user