Safe get for backflush status in lamarzocco (#145559)

* Safe get for backflush status in lamarzocco

* add correct default
This commit is contained in:
Josef Zweck
2025-05-25 14:01:15 +02:00
committed by GitHub
parent 6b1484a7f0
commit d0bc71752b
2 changed files with 9 additions and 3 deletions

View File

@@ -61,7 +61,10 @@ ENTITIES: tuple[LaMarzoccoBinarySensorEntityDescription, ...] = (
device_class=BinarySensorDeviceClass.RUNNING,
is_on_fn=(
lambda machine: cast(
BackFlush, machine.dashboard.config[WidgetType.CM_BACK_FLUSH]
BackFlush,
machine.dashboard.config.get(
WidgetType.CM_BACK_FLUSH, BackFlush(status=BackFlushStatus.OFF)
),
).status
is BackFlushStatus.REQUESTED
),

View File

@@ -5,7 +5,7 @@ from dataclasses import dataclass
from datetime import datetime
from typing import cast
from pylamarzocco.const import ModelName, WidgetType
from pylamarzocco.const import BackFlushStatus, ModelName, WidgetType
from pylamarzocco.models import (
BackFlush,
BaseWidgetOutput,
@@ -106,7 +106,10 @@ ENTITIES: tuple[LaMarzoccoSensorEntityDescription, ...] = (
device_class=SensorDeviceClass.TIMESTAMP,
value_fn=(
lambda config: cast(
BackFlush, config[WidgetType.CM_BACK_FLUSH]
BackFlush,
config.get(
WidgetType.CM_BACK_FLUSH, BackFlush(status=BackFlushStatus.OFF)
),
).last_cleaning_start_time
),
entity_category=EntityCategory.DIAGNOSTIC,