From 8f3dcd655762c9d9e6358de40ce06d32bb3447f7 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Tue, 18 Jun 2024 18:21:19 -0500 Subject: [PATCH] Cleanup august dataclasses (#119938) --- homeassistant/components/august/config_flow.py | 2 +- homeassistant/components/august/sensor.py | 11 ++--------- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/august/config_flow.py b/homeassistant/components/august/config_flow.py index 75543311fdd..18c15ad61a1 100644 --- a/homeassistant/components/august/config_flow.py +++ b/homeassistant/components/august/config_flow.py @@ -66,7 +66,7 @@ async def async_validate_input( } -@dataclass +@dataclass(slots=True) class ValidateResult: """Result from validation.""" diff --git a/homeassistant/components/august/sensor.py b/homeassistant/components/august/sensor.py index 862117319fa..59f4d0cf33f 100644 --- a/homeassistant/components/august/sensor.py +++ b/homeassistant/components/august/sensor.py @@ -58,20 +58,13 @@ def _retrieve_linked_keypad_battery_state(detail: KeypadDetail) -> int | None: _T = TypeVar("_T", LockDetail, KeypadDetail) -@dataclass(frozen=True) -class AugustRequiredKeysMixin(Generic[_T]): +@dataclass(frozen=True, kw_only=True) +class AugustSensorEntityDescription(SensorEntityDescription, Generic[_T]): """Mixin for required keys.""" value_fn: Callable[[_T], int | None] -@dataclass(frozen=True) -class AugustSensorEntityDescription( - SensorEntityDescription, AugustRequiredKeysMixin[_T] -): - """Describes August sensor entity.""" - - SENSOR_TYPE_DEVICE_BATTERY = AugustSensorEntityDescription[LockDetail]( key="device_battery", entity_category=EntityCategory.DIAGNOSTIC,