From d0bc71752b69d667874fc3d2f7146994ad221976 Mon Sep 17 00:00:00 2001 From: Josef Zweck Date: Sun, 25 May 2025 14:01:15 +0200 Subject: [PATCH] Safe get for backflush status in lamarzocco (#145559) * Safe get for backflush status in lamarzocco * add correct default --- homeassistant/components/lamarzocco/binary_sensor.py | 5 ++++- homeassistant/components/lamarzocco/sensor.py | 7 +++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/lamarzocco/binary_sensor.py b/homeassistant/components/lamarzocco/binary_sensor.py index c108bdb02d8..aacfca929ad 100644 --- a/homeassistant/components/lamarzocco/binary_sensor.py +++ b/homeassistant/components/lamarzocco/binary_sensor.py @@ -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 ), diff --git a/homeassistant/components/lamarzocco/sensor.py b/homeassistant/components/lamarzocco/sensor.py index aecb2ff7f04..afe34005108 100644 --- a/homeassistant/components/lamarzocco/sensor.py +++ b/homeassistant/components/lamarzocco/sensor.py @@ -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,