add light platform to fressnapf_tracker (home-assistant#157865)

Author: eifinger

homeassistant/components/fressnapf_tracker/__init__.py

@@ -15,6 +15,7 @@
 PLATFORMS: list[Platform] = [
     Platform.BINARY_SENSOR,
     Platform.DEVICE_TRACKER,
+    Platform.LIGHT,
     Platform.SENSOR,
 ]
 

homeassistant/components/fressnapf_tracker/light.py

@@ -0,0 +1,93 @@
+"""Light platform for fressnapf_tracker."""
+
+from typing import TYPE_CHECKING, Any
+
+from homeassistant.components.light import (
+    ATTR_BRIGHTNESS,
+    ColorMode,
+    LightEntity,
+    LightEntityDescription,
+)
+from homeassistant.const import EntityCategory
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import HomeAssistantError
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from . import FressnapfTrackerConfigEntry
+from .const import DOMAIN
+from .entity import FressnapfTrackerEntity
+
+LIGHT_ENTITY_DESCRIPTION = LightEntityDescription(
+    translation_key="led",
+    entity_category=EntityCategory.CONFIG,
+    key="led_brightness_value",
+)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: FressnapfTrackerConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up the Fressnapf Tracker lights."""
+
+    async_add_entities(
+        FressnapfTrackerLight(coordinator, LIGHT_ENTITY_DESCRIPTION)
+        for coordinator in entry.runtime_data
+        if coordinator.data.led_activatable is not None
+        and coordinator.data.led_activatable.has_led
+        and coordinator.data.tracker_settings.features.flash_light
+    )
+
+
+class FressnapfTrackerLight(FressnapfTrackerEntity, LightEntity):
+    """Fressnapf Tracker light."""
+
+    _attr_color_mode: ColorMode = ColorMode.BRIGHTNESS
+    _attr_supported_color_modes: set[ColorMode] = {ColorMode.BRIGHTNESS}
+
+    @property
+    def brightness(self) -> int:
+        """Return the brightness of this light between 0..255."""
+        if TYPE_CHECKING:
+            # The entity is not created if led_brightness_value is None
+            assert self.coordinator.data.led_brightness_value is not None
+        return int(round((self.coordinator.data.led_brightness_value / 100) * 255))
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Turn on the device."""
+        self.raise_if_not_activatable()
+        brightness = kwargs.get(ATTR_BRIGHTNESS, 255)
+        brightness = int((brightness / 255) * 100)
+        await self.coordinator.client.set_led_brightness(brightness)
+        await self.coordinator.async_request_refresh()
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off the device."""
+        await self.coordinator.client.set_led_brightness(0)
+        await self.coordinator.async_request_refresh()
+
+    def raise_if_not_activatable(self) -> None:
+        """Raise error with reasoning if light is not activatable."""
+        if TYPE_CHECKING:
+            # The entity is not created if led_activatable is None
+            assert self.coordinator.data.led_activatable is not None
+        error_type: str | None = None
+        if not self.coordinator.data.led_activatable.seen_recently:
+            error_type = "not_seen_recently"
+        elif not self.coordinator.data.led_activatable.not_charging:
+            error_type = "charging"
+        elif not self.coordinator.data.led_activatable.nonempty_battery:
+            error_type = "low_battery"
+        if error_type is not None:
+            raise HomeAssistantError(
+                translation_domain=DOMAIN,
+                translation_key=error_type,
+            )
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if device is on."""
+        if self.coordinator.data.led_brightness_value is not None:
+            return self.coordinator.data.led_brightness_value > 0
+        return False

homeassistant/components/fressnapf_tracker/strings.json

@@ -45,5 +45,23 @@
         }
       }
     }
+  },
+  "entity": {
+    "light": {
+      "led": {
+        "name": "Flashlight"
+      }
+    }
+  },
+  "exceptions": {
+    "charging": {
+      "message": "The flashlight cannot be activated while charging."
+    },
+    "low_battery": {
+      "message": "The flashlight cannot be activated due to low battery."
+    },
+    "not_seen_recently": {
+      "message": "The flashlight cannot be activated when the tracker has not moved recently."
+    }
   }
 }

tests/components/fressnapf_tracker/conftest.py

@@ -5,6 +5,8 @@
 
 from fressnapftracker import (
     Device,
+    LedActivatable,
+    LedBrightness,
     PhoneVerificationResponse,
     Position,
     SmsCodeResponse,
@@ -30,6 +32,30 @@
 MOCK_ACCESS_TOKEN = "mock_access_token"
 MOCK_SERIAL_NUMBER = "ABC123456"
 MOCK_DEVICE_TOKEN = "mock_device_token"
+MOCK_TRACKER = Tracker(
+    name="Fluffy",
+    battery=85,
+    charging=False,
+    position=Position(
+        lat=52.520008,
+        lng=13.404954,
+        accuracy=10,
+        timestamp="2024-01-15T12:00:00Z",
+    ),
+    tracker_settings=TrackerSettings(
+        generation="GPS Tracker 2.0",
+        features=TrackerFeatures(flash_light=True, live_tracking=True),
+    ),
+    led_brightness=LedBrightness(value=50),
+    deep_sleep=None,
+    led_activatable=LedActivatable(
+        has_led=True,
+        seen_recently=True,
+        nonempty_battery=True,
+        not_charging=True,
+        overall=True,
+    ),
+)
 
 
 @pytest.fixture
@@ -42,29 +68,6 @@ def mock_setup_entry() -> Generator[AsyncMock]:
         yield mock_setup_entry
 
 
-@pytest.fixture
-def mock_tracker() -> Tracker:
-    """Create a mock Tracker object."""
-    return Tracker(
-        name="Fluffy",
-        battery=85,
-        charging=False,
-        position=Position(
-            lat=52.520008,
-            lng=13.404954,
-            accuracy=10,
-            timestamp="2024-01-15T12:00:00Z",
-        ),
-        tracker_settings=TrackerSettings(
-            generation="GPS Tracker 2.0",
-            features=TrackerFeatures(live_tracking=True),
-        ),
-        led_brightness=None,
-        deep_sleep=None,
-        led_activatable=None,
-    )
-
-
 @pytest.fixture
 def mock_tracker_no_position() -> Tracker:
     """Create a mock Tracker object without position."""
@@ -122,13 +125,14 @@ def mock_auth_client(mock_device: Device) -> Generator[MagicMock]:
 
 
 @pytest.fixture
-def mock_api_client(mock_tracker: Tracker) -> Generator[MagicMock]:
+def mock_api_client() -> Generator[MagicMock]:
     """Mock the ApiClient."""
     with patch(
         "homeassistant.components.fressnapf_tracker.coordinator.ApiClient"
     ) as mock_api_client:
         client = mock_api_client.return_value
-        client.get_tracker = AsyncMock(return_value=mock_tracker)
+        client.get_tracker = AsyncMock(return_value=MOCK_TRACKER)
+        client.set_led_brightness = AsyncMock(return_value=None)
         yield client
 
 

tests/components/fressnapf_tracker/snapshots/test_light.ambr

@@ -0,0 +1,59 @@
+# serializer version: 1
+# name: test_state_entity_device_snapshots[light.fluffy_flashlight-entry]
+  EntityRegistryEntrySnapshot({
+    'aliases': set({
+    }),
+    'area_id': None,
+    'capabilities': dict({
+      'supported_color_modes': list([
+        <ColorMode.BRIGHTNESS: 'brightness'>,
+      ]),
+    }),
+    'config_entry_id': <ANY>,
+    'config_subentry_id': <ANY>,
+    'device_class': None,
+    'device_id': <ANY>,
+    'disabled_by': None,
+    'domain': 'light',
+    'entity_category': <EntityCategory.CONFIG: 'config'>,
+    'entity_id': 'light.fluffy_flashlight',
+    'has_entity_name': True,
+    'hidden_by': None,
+    'icon': None,
+    'id': <ANY>,
+    'labels': set({
+    }),
+    'name': None,
+    'options': dict({
+    }),
+    'original_device_class': None,
+    'original_icon': None,
+    'original_name': 'Flashlight',
+    'platform': 'fressnapf_tracker',
+    'previous_unique_id': None,
+    'suggested_object_id': None,
+    'supported_features': 0,
+    'translation_key': 'led',
+    'unique_id': 'ABC123456_led_brightness_value',
+    'unit_of_measurement': None,
+  })
+# ---
+# name: test_state_entity_device_snapshots[light.fluffy_flashlight-state]
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'brightness': 128,
+      'color_mode': <ColorMode.BRIGHTNESS: 'brightness'>,
+      'friendly_name': 'Fluffy Flashlight',
+      'supported_color_modes': list([
+        <ColorMode.BRIGHTNESS: 'brightness'>,
+      ]),
+      'supported_features': <LightEntityFeature: 0>,
+    }),
+    'context': <ANY>,
+    'entity_id': 'light.fluffy_flashlight',
+    'last_changed': <ANY>,
+    'last_reported': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'on',
+  })
+# ---