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

Fix meater sensor (#71283) 

* Fix meater sensor

* Cleanup MeaterEntityDescription

* Apply suggestions from code review

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* Update sensor.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
pull/71297/head
Erik Montnemery 2022-05-04 16:13:09 +02:00 committed by Franck Nijhof
parent bf17bd55fd
commit 44d17a80c3
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
homeassistant/components/meater/sensor.py
View File

@ -27,15 +27,18 @@ from .const import DOMAIN
@dataclass
class MeaterSensorEntityDescription(SensorEntityDescription):
"""Describes meater sensor entity."""
class MeaterSensorEntityDescriptionMixin:
"""Mixin for MeaterSensorEntityDescription."""
available: Callable[
[MeaterProbe | None], bool | type[NotImplementedError]
] = lambda x: NotImplementedError
value: Callable[
[MeaterProbe], datetime | float | str | None | type[NotImplementedError]
] = lambda x: NotImplementedError
available: Callable[[MeaterProbe | None], bool]
value: Callable[[MeaterProbe], datetime | float | str | None]
@dataclass
class MeaterSensorEntityDescription(
SensorEntityDescription, MeaterSensorEntityDescriptionMixin
):
"""Describes meater sensor entity."""
def _elapsed_time_to_timestamp(probe: MeaterProbe) -> datetime | None:
@ -108,7 +111,8 @@ SENSOR_TYPES = (
available=lambda probe: probe is not None and probe.cook is not None,
value=lambda probe: probe.cook.peak_temperature if probe.cook else None,
),
# Time since the start of cook in seconds. Default: 0.
# Remaining time in seconds. When unknown/calculating default is used. Default: -1
# Exposed as a TIMESTAMP sensor where the timestamp is current time + remaining time.
MeaterSensorEntityDescription(
key="cook_time_remaining",
device_class=SensorDeviceClass.TIMESTAMP,
@ -116,7 +120,8 @@ SENSOR_TYPES = (
available=lambda probe: probe is not None and probe.cook is not None,
value=_remaining_time_to_timestamp,
),
# Remaining time in seconds. When unknown/calculating default is used. Default: -1
# Time since the start of cook in seconds. Default: 0. Exposed as a TIMESTAMP sensor
# where the timestamp is current time - elapsed time.
MeaterSensorEntityDescription(
key="cook_time_elapsed",
device_class=SensorDeviceClass.TIMESTAMP,
@ -147,7 +152,7 @@ async def async_setup_entry(
# Add entities for temperature probes which we've not yet seen
for dev in devices:
if dev in known_probes:
if dev.id in known_probes:
continue
entities.extend(
@ -156,7 +161,7 @@ async def async_setup_entry(
for sensor_description in SENSOR_TYPES
]
)
known_probes.add(dev)
known_probes.add(dev.id)
async_add_entities(entities)