From 793ad568eb0006ba4b0fd10c3e0e95941980ea95 Mon Sep 17 00:00:00 2001
From: Kevin Stillhammer <kevin.stillhammer@gmail.com>
Date: Mon, 23 May 2022 10:47:40 +0200
Subject: [PATCH] here_travel_time: Add unique_id and DeviceInfo (#72352)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
---
 homeassistant/components/here_travel_time/sensor.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/homeassistant/components/here_travel_time/sensor.py b/homeassistant/components/here_travel_time/sensor.py
index 4a09252f068..75c9fd2ea3b 100644
--- a/homeassistant/components/here_travel_time/sensor.py
+++ b/homeassistant/components/here_travel_time/sensor.py
@@ -19,6 +19,8 @@ from homeassistant.const import (
 )
 from homeassistant.core import HomeAssistant
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.device_registry import DeviceEntryType
+from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.start import async_at_start
 from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@@ -144,6 +146,7 @@ async def async_setup_entry(
     async_add_entities(
         [
             HERETravelTimeSensor(
+                config_entry.entry_id,
                 config_entry.data[CONF_NAME],
                 config_entry.options[CONF_TRAFFIC_MODE],
                 hass.data[DOMAIN][config_entry.entry_id],
@@ -157,6 +160,7 @@ class HERETravelTimeSensor(SensorEntity, CoordinatorEntity):
 
     def __init__(
         self,
+        unique_id_prefix: str,
         name: str,
         traffic_mode: str,
         coordinator: HereTravelTimeDataUpdateCoordinator,
@@ -166,6 +170,13 @@ class HERETravelTimeSensor(SensorEntity, CoordinatorEntity):
         self._traffic_mode = traffic_mode == TRAFFIC_MODE_ENABLED
         self._attr_native_unit_of_measurement = TIME_MINUTES
         self._attr_name = name
+        self._attr_unique_id = unique_id_prefix
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, unique_id_prefix)},
+            entry_type=DeviceEntryType.SERVICE,
+            name=name,
+            manufacturer="HERE Technologies",
+        )
 
     async def async_added_to_hass(self) -> None:
         """Wait for start so origin and destination entities can be resolved."""