From 7f34edfd4b2be937d5d77291c892294de938b734 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Tue, 21 Jul 2026 08:22:01 +0200 Subject: [PATCH] Adapt bang_olufsen to new device registry API (#176937) --- homeassistant/components/bang_olufsen/util.py | 6 ++++-- homeassistant/components/bang_olufsen/websocket.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/bang_olufsen/util.py b/homeassistant/components/bang_olufsen/util.py index d869c63efa2a..4203417424d4 100644 --- a/homeassistant/components/bang_olufsen/util.py +++ b/homeassistant/components/bang_olufsen/util.py @@ -21,10 +21,12 @@ from .const import ( ) -def get_device(hass: HomeAssistant, unique_id: str) -> DeviceEntry: +def get_device(hass: HomeAssistant, unique_id: str, entry_id: str) -> DeviceEntry: """Get the device.""" device_registry = dr.async_get(hass) - device = device_registry.async_get_device({(DOMAIN, unique_id)}) + device = device_registry.async_get_device_by_identifier( + (DOMAIN, unique_id), entry_id + ) assert device return device diff --git a/homeassistant/components/bang_olufsen/websocket.py b/homeassistant/components/bang_olufsen/websocket.py index 93e352b47441..0ed29ed916f6 100644 --- a/homeassistant/components/bang_olufsen/websocket.py +++ b/homeassistant/components/bang_olufsen/websocket.py @@ -50,7 +50,7 @@ class BeoWebsocket(BeoBase): BeoBase.__init__(self, entry, client) self.hass = hass - self._device = get_device(hass, self._unique_id) + self._device = get_device(hass, self._unique_id, self.entry.entry_id) # WebSocket callbacks self._client.get_notification_notifications(self.on_notification_notification)