Show Plugwise configuration-link on gateway only (home-assistant#158094)

Author: bouwew

homeassistant/components/plugwise/entity.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from plugwise.constants import GwEntityData
+from plugwise import GwEntityData
 
 from homeassistant.const import ATTR_NAME, ATTR_VIA_DEVICE, CONF_HOST
 from homeassistant.helpers.device_registry import (
@@ -30,37 +30,43 @@ def __init__(
         super().__init__(coordinator)
         self._dev_id = device_id
 
-        configuration_url: str | None = None
-        if entry := self.coordinator.config_entry:
-            configuration_url = f"http://{entry.data[CONF_HOST]}"
+        api = coordinator.api
+        gateway_id = api.gateway_id
+        entry = coordinator.config_entry
 
-        data = coordinator.data[device_id]
+        # Link configuration-URL for the gateway device
+        configuration_url = (
+            f"http://{entry.data[CONF_HOST]}"
+            if device_id == gateway_id and entry
+            else None
+        )
+
+        # Build connections set
         connections = set()
-        if mac := data.get("mac_address"):
+        if mac := self.device.get("mac_address"):
             connections.add((CONNECTION_NETWORK_MAC, mac))
-        if mac := data.get("zigbee_mac_address"):
-            connections.add((CONNECTION_ZIGBEE, mac))
+        if zigbee_mac := self.device.get("zigbee_mac_address"):
+            connections.add((CONNECTION_ZIGBEE, zigbee_mac))
 
+        # Set base device info
         self._attr_device_info = DeviceInfo(
             configuration_url=configuration_url,
             identifiers={(DOMAIN, device_id)},
             connections=connections,
-            manufacturer=data.get("vendor"),
-            model=data.get("model"),
-            model_id=data.get("model_id"),
-            name=coordinator.api.smile.name,
-            sw_version=data.get("firmware"),
-            hw_version=data.get("hardware"),
+            manufacturer=self.device.get("vendor"),
+            model=self.device.get("model"),
+            model_id=self.device.get("model_id"),
+            name=api.smile.name,
+            sw_version=self.device.get("firmware"),
+            hw_version=self.device.get("hardware"),
         )
 
-        if device_id != coordinator.api.gateway_id:
+        # Add extra info if not the gateway device
+        if device_id != gateway_id:
             self._attr_device_info.update(
                 {
-                    ATTR_NAME: data.get(ATTR_NAME),
-                    ATTR_VIA_DEVICE: (
-                        DOMAIN,
-                        str(self.coordinator.api.gateway_id),
-                    ),
+                    ATTR_NAME: self.device.get(ATTR_NAME),
+                    ATTR_VIA_DEVICE: (DOMAIN, gateway_id),
                 }
             )