Mill connection

Signed-off-by: Daniel Hjelseth Høyer <github@dahoiv.net>

homeassistant/components/mill/climate.py

@@ -113,6 +113,7 @@ class MillHeater(CoordinatorEntity[MillDataUpdateCoordinator], ClimateEntity):
         self._id = heater.device_id
         self._attr_unique_id = heater.device_id
         self._attr_device_info = DeviceInfo(
+            connections={(CONNECTION_NETWORK_MAC, heater.mac_address)},
             identifiers={(DOMAIN, heater.device_id)},
             manufacturer=MANUFACTURER,
             model=heater.model,
@@ -197,16 +198,16 @@ class LocalMillHeater(CoordinatorEntity[MillDataUpdateCoordinator], ClimateEntit
     def __init__(self, coordinator: MillDataUpdateCoordinator) -> None:
         """Initialize the thermostat."""
         super().__init__(coordinator)
-        if mac := coordinator.mill_data_connection.mac_address:
-            self._attr_unique_id = mac
-            self._attr_device_info = DeviceInfo(
-                connections={(CONNECTION_NETWORK_MAC, mac)},
-                configuration_url=self.coordinator.mill_data_connection.url,
-                manufacturer=MANUFACTURER,
-                model="Generation 3",
-                name=coordinator.mill_data_connection.name,
-                sw_version=coordinator.mill_data_connection.version,
-            )
+        mac = coordinator.mill_data_connection.mac_address
+        self._attr_unique_id = mac
+        self._attr_device_info = DeviceInfo(
+            connections={(CONNECTION_NETWORK_MAC, mac)},
+            configuration_url=self.coordinator.mill_data_connection.url,
+            manufacturer=MANUFACTURER,
+            model="Generation 3",
+            name=coordinator.mill_data_connection.name,
+            sw_version=coordinator.mill_data_connection.version,
+        )
 
         self._update_attr()
 

homeassistant/components/mill/sensor.py

@@ -193,6 +193,7 @@ class MillSensor(CoordinatorEntity, SensorEntity):
         self._available = False
         self._attr_unique_id = f"{mill_device.device_id}_{entity_description.key}"
         self._attr_device_info = DeviceInfo(
+            connections={(CONNECTION_NETWORK_MAC, mill_device.mac_address)},
             identifiers={(DOMAIN, mill_device.device_id)},
             name=mill_device.name,
             manufacturer=MANUFACTURER,
@@ -227,16 +228,17 @@ class LocalMillSensor(CoordinatorEntity, SensorEntity):
         super().__init__(coordinator)
 
         self.entity_description = entity_description
-        if mac := coordinator.mill_data_connection.mac_address:
-            self._attr_unique_id = f"{mac}_{entity_description.key}"
-            self._attr_device_info = DeviceInfo(
-                connections={(CONNECTION_NETWORK_MAC, mac)},
-                configuration_url=self.coordinator.mill_data_connection.url,
-                manufacturer=MANUFACTURER,
-                model="Generation 3",
-                name=coordinator.mill_data_connection.name,
-                sw_version=coordinator.mill_data_connection.version,
-            )
+        mac = coordinator.mill_data_connection.mac_address
+        self._attr_unique_id = f"{mac}_{entity_description.key}"
+        self._attr_device_info = DeviceInfo(
+            connections={(CONNECTION_NETWORK_MAC, mac)},
+            configuration_url=self.coordinator.mill_data_connection.url,
+            identifiers={(DOMAIN, mac)},
+            manufacturer=MANUFACTURER,
+            model="Generation 3",
+            name=coordinator.mill_data_connection.name,
+            sw_version=coordinator.mill_data_connection.version,
+        )
 
     @property
     def native_value(self):