diff --git a/homeassistant/components/vodafone_station/button.py b/homeassistant/components/vodafone_station/button.py index 3bd012734b6..efea011a541 100644 --- a/homeassistant/components/vodafone_station/button.py +++ b/homeassistant/components/vodafone_station/button.py @@ -21,21 +21,14 @@ from .const import _LOGGER, DOMAIN from .coordinator import VodafoneStationRouter -@dataclass(frozen=True) -class VodafoneStationBaseEntityDescriptionMixin: - """Mixin to describe a Button entity.""" +@dataclass(frozen=True, kw_only=True) +class VodafoneStationEntityDescription(ButtonEntityDescription): + """Vodafone Station entity description.""" press_action: Callable[[VodafoneStationRouter], Any] is_suitable: Callable[[dict], bool] -@dataclass(frozen=True) -class VodafoneStationEntityDescription( - ButtonEntityDescription, VodafoneStationBaseEntityDescriptionMixin -): - """Vodafone Station entity description.""" - - BUTTON_TYPES: Final = ( VodafoneStationEntityDescription( key="reboot", diff --git a/homeassistant/components/vodafone_station/sensor.py b/homeassistant/components/vodafone_station/sensor.py index 5bc7465a902..937c0220cbf 100644 --- a/homeassistant/components/vodafone_station/sensor.py +++ b/homeassistant/components/vodafone_station/sensor.py @@ -25,9 +25,9 @@ from .coordinator import VodafoneStationRouter NOT_AVAILABLE: list = ["", "N/A", "0.0.0.0"] -@dataclass(frozen=True) -class VodafoneStationBaseEntityDescription: - """Vodafone Station entity base description.""" +@dataclass(frozen=True, kw_only=True) +class VodafoneStationEntityDescription(SensorEntityDescription): + """Vodafone Station entity description.""" value: Callable[[Any, Any], Any] = ( lambda coordinator, key: coordinator.data.sensors[key] @@ -35,13 +35,6 @@ class VodafoneStationBaseEntityDescription: is_suitable: Callable[[dict], bool] = lambda val: True -@dataclass(frozen=True, kw_only=True) -class VodafoneStationEntityDescription( - VodafoneStationBaseEntityDescription, SensorEntityDescription -): - """Vodafone Station entity description.""" - - def _calculate_uptime(coordinator: VodafoneStationRouter, key: str) -> datetime: """Calculate device uptime."""