From 2e5c9b69d4ca3a023631c9efdf8f8320a1d8efa1 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 22 Oct 2021 11:32:15 +0200 Subject: [PATCH] Use DeviceInfo on zha (#58202) Co-authored-by: epenet --- homeassistant/components/zha/entity.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/zha/entity.py b/homeassistant/components/zha/entity.py index 6fd68056025..0ba75a99306 100644 --- a/homeassistant/components/zha/entity.py +++ b/homeassistant/components/zha/entity.py @@ -92,14 +92,14 @@ class BaseZhaEntity(LogMixin, entity.Entity): """Return a device description for device registry.""" zha_device_info = self._zha_device.device_info ieee = zha_device_info["ieee"] - return { - "connections": {(CONNECTION_ZIGBEE, ieee)}, - "identifiers": {(DOMAIN, ieee)}, - ATTR_MANUFACTURER: zha_device_info[ATTR_MANUFACTURER], - ATTR_MODEL: zha_device_info[ATTR_MODEL], - ATTR_NAME: zha_device_info[ATTR_NAME], - "via_device": (DOMAIN, self.hass.data[DATA_ZHA][DATA_ZHA_BRIDGE_ID]), - } + return entity.DeviceInfo( + connections={(CONNECTION_ZIGBEE, ieee)}, + identifiers={(DOMAIN, ieee)}, + manufacturer=zha_device_info[ATTR_MANUFACTURER], + model=zha_device_info[ATTR_MODEL], + name=zha_device_info[ATTR_NAME], + via_device=(DOMAIN, self.hass.data[DATA_ZHA][DATA_ZHA_BRIDGE_ID]), + ) @callback def async_state_changed(self) -> None: