From 67fa304b7869d787239e76aede6a3ec61dd41d16 Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke Date: Wed, 1 Nov 2023 18:09:49 +0100 Subject: [PATCH] Use constant instead of plain key name for device info connections in roborock and roomba (#103182) --- homeassistant/components/roborock/coordinator.py | 3 ++- homeassistant/components/roomba/irobot_base.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/roborock/coordinator.py b/homeassistant/components/roborock/coordinator.py index da15a80ce1f..dd4ef9e052f 100644 --- a/homeassistant/components/roborock/coordinator.py +++ b/homeassistant/components/roborock/coordinator.py @@ -10,6 +10,7 @@ from roborock.exceptions import RoborockException from roborock.local_api import RoborockLocalClient from roborock.roborock_typing import DeviceProp +from homeassistant.const import ATTR_CONNECTIONS from homeassistant.core import HomeAssistant from homeassistant.helpers import device_registry as dr from homeassistant.helpers.device_registry import DeviceInfo @@ -54,7 +55,7 @@ class RoborockDataUpdateCoordinator(DataUpdateCoordinator[DeviceProp]): ) if mac := self.roborock_device_info.network_info.mac: - self.device_info["connections"] = {(dr.CONNECTION_NETWORK_MAC, mac)} + self.device_info[ATTR_CONNECTIONS] = {(dr.CONNECTION_NETWORK_MAC, mac)} async def verify_api(self) -> None: """Verify that the api is reachable. If it is not, switch clients.""" diff --git a/homeassistant/components/roomba/irobot_base.py b/homeassistant/components/roomba/irobot_base.py index 62afdf0df6b..e4aa40e34f0 100644 --- a/homeassistant/components/roomba/irobot_base.py +++ b/homeassistant/components/roomba/irobot_base.py @@ -13,7 +13,7 @@ from homeassistant.components.vacuum import ( StateVacuumEntity, VacuumEntityFeature, ) -from homeassistant.const import STATE_IDLE, STATE_PAUSED +from homeassistant.const import ATTR_CONNECTIONS, STATE_IDLE, STATE_PAUSED import homeassistant.helpers.device_registry as dr from homeassistant.helpers.device_registry import DeviceInfo from homeassistant.helpers.entity import Entity @@ -83,7 +83,7 @@ class IRobotEntity(Entity): if mac_address := self.vacuum_state.get("hwPartsRev", {}).get( "wlan0HwAddr", self.vacuum_state.get("mac") ): - self._attr_device_info["connections"] = { + self._attr_device_info[ATTR_CONNECTIONS] = { (dr.CONNECTION_NETWORK_MAC, mac_address) }