From 4a7eab92bb5ce2543b79e14071c42d08f0854a51 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Tue, 5 Mar 2024 18:27:18 +0100 Subject: [PATCH] Remove entity description mixin in Ecobee (#112408) --- homeassistant/components/ecobee/number.py | 13 +++---------- homeassistant/components/ecobee/sensor.py | 13 +++---------- 2 files changed, 6 insertions(+), 20 deletions(-) diff --git a/homeassistant/components/ecobee/number.py b/homeassistant/components/ecobee/number.py index 345ca7b705f..a4106f196a1 100644 --- a/homeassistant/components/ecobee/number.py +++ b/homeassistant/components/ecobee/number.py @@ -18,21 +18,14 @@ from .entity import EcobeeBaseEntity _LOGGER = logging.getLogger(__name__) -@dataclass(frozen=True) -class EcobeeNumberEntityDescriptionBase: - """Required values when describing Ecobee number entities.""" +@dataclass(frozen=True, kw_only=True) +class EcobeeNumberEntityDescription(NumberEntityDescription): + """Class describing Ecobee number entities.""" ecobee_setting_key: str set_fn: Callable[[EcobeeData, int, int], Awaitable] -@dataclass(frozen=True) -class EcobeeNumberEntityDescription( - NumberEntityDescription, EcobeeNumberEntityDescriptionBase -): - """Class describing Ecobee number entities.""" - - VENTILATOR_NUMBERS = ( EcobeeNumberEntityDescription( key="home", diff --git a/homeassistant/components/ecobee/sensor.py b/homeassistant/components/ecobee/sensor.py index 7f0e7b808a8..ce2f0f7beb8 100644 --- a/homeassistant/components/ecobee/sensor.py +++ b/homeassistant/components/ecobee/sensor.py @@ -25,20 +25,13 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import DOMAIN, ECOBEE_MODEL_TO_NAME, MANUFACTURER -@dataclass(frozen=True) -class EcobeeSensorEntityDescriptionMixin: - """Represent the required ecobee entity description attributes.""" +@dataclass(frozen=True, kw_only=True) +class EcobeeSensorEntityDescription(SensorEntityDescription): + """Represent the ecobee sensor entity description.""" runtime_key: str | None -@dataclass(frozen=True) -class EcobeeSensorEntityDescription( - SensorEntityDescription, EcobeeSensorEntityDescriptionMixin -): - """Represent the ecobee sensor entity description.""" - - SENSOR_TYPES: tuple[EcobeeSensorEntityDescription, ...] = ( EcobeeSensorEntityDescription( key="temperature",