Refactor goalzero (#72398)

homeassistant/components/goalzero/binary_sensor.py

@@ -3,22 +3,18 @@ from __future__ import annotations
 
 from typing import cast
 
-from goalzero import Yeti
-
 from homeassistant.components.binary_sensor import (
     BinarySensorDeviceClass,
     BinarySensorEntity,
     BinarySensorEntityDescription,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CONF_NAME
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
-from . import YetiEntity
-from .const import DATA_KEY_API, DATA_KEY_COORDINATOR, DOMAIN
+from .const import DOMAIN
+from .entity import GoalZeroEntity
 
 PARALLEL_UPDATES = 0
 
@@ -51,38 +47,19 @@ async def async_setup_entry(
     hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
 ) -> None:
     """Set up the Goal Zero Yeti sensor."""
-    name = entry.data[CONF_NAME]
-    goalzero_data = hass.data[DOMAIN][entry.entry_id]
     async_add_entities(
-        YetiBinarySensor(
-            goalzero_data[DATA_KEY_API],
-            goalzero_data[DATA_KEY_COORDINATOR],
-            name,
+        GoalZeroBinarySensor(
+            hass.data[DOMAIN][entry.entry_id],
             description,
-            entry.entry_id,
         )
         for description in BINARY_SENSOR_TYPES
     )
 
 
-class YetiBinarySensor(YetiEntity, BinarySensorEntity):
+class GoalZeroBinarySensor(GoalZeroEntity, BinarySensorEntity):
     """Representation of a Goal Zero Yeti sensor."""
 
-    def __init__(
-        self,
-        api: Yeti,
-        coordinator: DataUpdateCoordinator,
-        name: str,
-        description: BinarySensorEntityDescription,
-        server_unique_id: str,
-    ) -> None:
-        """Initialize a Goal Zero Yeti sensor."""
-        super().__init__(api, coordinator, name, server_unique_id)
-        self.entity_description = description
-        self._attr_name = f"{name} {description.name}"
-        self._attr_unique_id = f"{server_unique_id}/{description.key}"
-
     @property
     def is_on(self) -> bool:
         """Return True if the service is on."""
-        return cast(bool, self.api.data[self.entity_description.key] == 1)
+        return cast(bool, self._api.data[self.entity_description.key] == 1)