From 1c0d5f86373d1892687257edcfa2e5f5def5cde2 Mon Sep 17 00:00:00 2001 From: Joost Lekkerkerker Date: Mon, 28 Aug 2023 17:59:53 +0200 Subject: [PATCH] Clean up Balboa entity (#99203) --- homeassistant/components/balboa/entity.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/balboa/entity.py b/homeassistant/components/balboa/entity.py index 2b845b65496..3b4f7d08fff 100644 --- a/homeassistant/components/balboa/entity.py +++ b/homeassistant/components/balboa/entity.py @@ -9,18 +9,18 @@ from homeassistant.helpers.entity import Entity from .const import DOMAIN -class BalboaBaseEntity(Entity): +class BalboaEntity(Entity): """Balboa base entity.""" + _attr_should_poll = False + _attr_has_entity_name = True + def __init__(self, client: SpaClient, name: str | None = None) -> None: """Initialize the control.""" mac = client.mac_address model = client.model - - self._attr_should_poll = False self._attr_unique_id = f'{model}-{name}-{mac.replace(":","")[-6:]}' self._attr_name = name - self._attr_has_entity_name = True self._attr_device_info = DeviceInfo( identifiers={(DOMAIN, mac)}, name=model, @@ -36,10 +36,6 @@ class BalboaBaseEntity(Entity): """Return whether the state is based on actual reading from device.""" return not self._client.available - -class BalboaEntity(BalboaBaseEntity): - """Balboa entity.""" - async def async_added_to_hass(self) -> None: """Run when entity about to be added to hass.""" self.async_on_remove(self._client.on(EVENT_UPDATE, self.async_write_ha_state))