add device_info property

finity69x2 · Mar 16, 2023 · fec9b8e · fec9b8e
1 parent 9679ab0
commit fec9b8e
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 47 deletions.
diff --git a/custom_components/nws_alerts/__init__.py b/custom_components/nws_alerts/__init__.py
@@ -6,30 +6,18 @@
 import aiohttp
 from async_timeout import timeout
 from homeassistant import config_entries
-from homeassistant.const import CONF_NAME
 from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_NAME
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_registry import (
-    async_entries_for_config_entry,
-    async_get,
-)
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
-
-from .const import (
-    API_ENDPOINT,
-    CONF_GPS_LOC,
-    CONF_INTERVAL,
-    CONF_TIMEOUT,
-    CONF_ZONE_ID,
-    COORDINATOR,
-    DEFAULT_INTERVAL,
-    DEFAULT_TIMEOUT,
-    DOMAIN,
-    ISSUE_URL,
-    PLATFORMS,
-    USER_AGENT,
-    VERSION,
-)
+    async_entries_for_config_entry, async_get)
+from homeassistant.helpers.update_coordinator import (DataUpdateCoordinator,
+                                                      UpdateFailed)
+
+from .const import (API_ENDPOINT, CONF_GPS_LOC, CONF_INTERVAL, CONF_TIMEOUT,
+                    CONF_ZONE_ID, COORDINATOR, DEFAULT_INTERVAL,
+                    DEFAULT_TIMEOUT, DOMAIN, ISSUE_URL, PLATFORMS, USER_AGENT,
+                    VERSION)
 
 _LOGGER = logging.getLogger(__name__)
 

diff --git a/custom_components/nws_alerts/config_flow.py b/custom_components/nws_alerts/config_flow.py
@@ -11,18 +11,9 @@
 from homeassistant.core import callback
 from homeassistant.data_entry_flow import FlowResult
 
-from .const import (
-    API_ENDPOINT,
-    CONF_GPS_LOC,
-    CONF_INTERVAL,
-    CONF_TIMEOUT,
-    CONF_ZONE_ID,
-    DEFAULT_INTERVAL,
-    DEFAULT_NAME,
-    DEFAULT_TIMEOUT,
-    DOMAIN,
-    USER_AGENT,
-)
+from .const import (API_ENDPOINT, CONF_GPS_LOC, CONF_INTERVAL, CONF_TIMEOUT,
+                    CONF_ZONE_ID, DEFAULT_INTERVAL, DEFAULT_NAME,
+                    DEFAULT_TIMEOUT, DOMAIN, USER_AGENT)
 
 JSON_FEATURES = "features"
 JSON_PROPERTIES = "properties"

diff --git a/custom_components/nws_alerts/sensor.py b/custom_components/nws_alerts/sensor.py
@@ -7,23 +7,15 @@
 from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.device_registry import DeviceEntryType
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.util import slugify
-from . import AlertsDataUpdateCoordinator
 
-from .const import (
-    ATTRIBUTION,
-    CONF_GPS_LOC,
-    CONF_INTERVAL,
-    CONF_TIMEOUT,
-    CONF_ZONE_ID,
-    COORDINATOR,
-    DEFAULT_ICON,
-    DEFAULT_INTERVAL,
-    DEFAULT_NAME,
-    DEFAULT_TIMEOUT,
-    DOMAIN,
-)
+from . import AlertsDataUpdateCoordinator
+from .const import (ATTRIBUTION, CONF_GPS_LOC, CONF_INTERVAL, CONF_TIMEOUT,
+                    CONF_ZONE_ID, COORDINATOR, DEFAULT_ICON, DEFAULT_INTERVAL,
+                    DEFAULT_NAME, DEFAULT_TIMEOUT, DOMAIN)
 
 # ---------------------------------------------------------
 # API Documentation
@@ -148,3 +140,13 @@ def extra_state_attributes(self):
     def available(self) -> bool:
         """Return if entity is available."""
         return self.coordinator.last_update_success
+
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return device registry information."""
+        return DeviceInfo(
+            entry_type=DeviceEntryType.SERVICE,
+            identifiers={(DOMAIN, self._config.entry_id)},
+            manufacturer="NWS",
+            name="NWS Alerts",
+        )