Use DeviceInfo on zha (#58202)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-10-22 11:32:15 +02:00 committed by GitHub
parent 184e0d7fdf
commit 2e5c9b69d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -92,14 +92,14 @@ class BaseZhaEntity(LogMixin, entity.Entity):
"""Return a device description for device registry.""" """Return a device description for device registry."""
zha_device_info = self._zha_device.device_info zha_device_info = self._zha_device.device_info
ieee = zha_device_info["ieee"] ieee = zha_device_info["ieee"]
return { return entity.DeviceInfo(
"connections": {(CONNECTION_ZIGBEE, ieee)}, connections={(CONNECTION_ZIGBEE, ieee)},
"identifiers": {(DOMAIN, ieee)}, identifiers={(DOMAIN, ieee)},
ATTR_MANUFACTURER: zha_device_info[ATTR_MANUFACTURER], manufacturer=zha_device_info[ATTR_MANUFACTURER],
ATTR_MODEL: zha_device_info[ATTR_MODEL], model=zha_device_info[ATTR_MODEL],
ATTR_NAME: zha_device_info[ATTR_NAME], name=zha_device_info[ATTR_NAME],
"via_device": (DOMAIN, self.hass.data[DATA_ZHA][DATA_ZHA_BRIDGE_ID]), via_device=(DOMAIN, self.hass.data[DATA_ZHA][DATA_ZHA_BRIDGE_ID]),
} )
@callback @callback
def async_state_changed(self) -> None: def async_state_changed(self) -> None: