Add coordinator to Blink (#102536)

homeassistant/components/blink/sensor.py

@@ -1,8 +1,6 @@
 """Support for Blink system camera sensors."""
 from __future__ import annotations
 
-from datetime import date, datetime
-from decimal import Decimal
 import logging
 
 from homeassistant.components.sensor import (
@@ -17,12 +15,13 @@ from homeassistant.const import (
     EntityCategory,
     UnitOfTemperature,
 )
-from homeassistant.core import HomeAssistant
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.typing import StateType
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
 from .const import DEFAULT_BRAND, DOMAIN, TYPE_TEMPERATURE, TYPE_WIFI_STRENGTH
+from .coordinator import BlinkUpdateCoordinator
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -49,26 +48,32 @@ async def async_setup_entry(
     hass: HomeAssistant, config: ConfigEntry, async_add_entities: AddEntitiesCallback
 ) -> None:
     """Initialize a Blink sensor."""
-    data = hass.data[DOMAIN][config.entry_id]
+    coordinator: BlinkUpdateCoordinator = hass.data[DOMAIN][config.entry_id]
     entities = [
-        BlinkSensor(data, camera, description)
-        for camera in data.cameras
+        BlinkSensor(coordinator, camera, description)
+        for camera in coordinator.api.cameras
         for description in SENSOR_TYPES
     ]
 
     async_add_entities(entities)
 
 
-class BlinkSensor(SensorEntity):
+class BlinkSensor(CoordinatorEntity[BlinkUpdateCoordinator], SensorEntity):
     """A Blink camera sensor."""
 
     _attr_has_entity_name = True
 
-    def __init__(self, data, camera, description: SensorEntityDescription) -> None:
+    def __init__(
+        self,
+        coordinator: BlinkUpdateCoordinator,
+        camera,
+        description: SensorEntityDescription,
+    ) -> None:
         """Initialize sensors from Blink camera."""
+        super().__init__(coordinator)
         self.entity_description = description
-        self.data = data
-        self._camera = data.cameras[camera]
+
+        self._camera = coordinator.api.cameras[camera]
         self._attr_unique_id = f"{self._camera.serial}-{description.key}"
         self._sensor_key = (
             "temperature_calibrated"
@@ -81,12 +86,19 @@ class BlinkSensor(SensorEntity):
             manufacturer=DEFAULT_BRAND,
             model=self._camera.camera_type,
         )
+        self._update_attr()
 
-    @property
-    def native_value(self) -> StateType | date | datetime | Decimal:
-        """Retrieve sensor data from the camera."""
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        """Handle coordinator update."""
+        self._update_attr()
+        super()._handle_coordinator_update()
+
+    @callback
+    def _update_attr(self) -> None:
+        """Update attributes for sensor."""
         try:
-            native_value = self._camera.attributes[self._sensor_key]
+            self._attr_native_value = self._camera.attributes[self._sensor_key]
             _LOGGER.debug(
                 "'%s' %s = %s",
                 self._camera.attributes["name"],
@@ -94,8 +106,7 @@ class BlinkSensor(SensorEntity):
                 self._attr_native_value,
             )
         except KeyError:
-            native_value = None
+            self._attr_native_value = None
             _LOGGER.error(
                 "%s not a valid camera attribute. Did the API change?", self._sensor_key
             )
-        return native_value