From 935242e64e2e201efe65540b55117d7e805f780f Mon Sep 17 00:00:00 2001 From: Jan-Philipp Benecke Date: Mon, 3 Jul 2023 14:04:17 +0200 Subject: [PATCH] Use device info object for Discovergy (#95764) --- homeassistant/components/discovergy/sensor.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/homeassistant/components/discovergy/sensor.py b/homeassistant/components/discovergy/sensor.py index 35955a6b189..3f4069752f2 100644 --- a/homeassistant/components/discovergy/sensor.py +++ b/homeassistant/components/discovergy/sensor.py @@ -11,16 +11,13 @@ from homeassistant.components.sensor import ( ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( - ATTR_IDENTIFIERS, - ATTR_MANUFACTURER, - ATTR_MODEL, - ATTR_NAME, UnitOfElectricPotential, UnitOfEnergy, UnitOfPower, UnitOfVolume, ) from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import StateType from homeassistant.helpers.update_coordinator import CoordinatorEntity @@ -198,12 +195,12 @@ class DiscovergySensor(CoordinatorEntity[DiscovergyUpdateCoordinator], SensorEnt self.entity_description = description self._attr_unique_id = f"{meter.full_serial_number}-{data_key}" - self._attr_device_info = { - ATTR_IDENTIFIERS: {(DOMAIN, meter.get_meter_id())}, - ATTR_NAME: f"{meter.measurement_type.capitalize()} {meter.location.street} {meter.location.street_number}", - ATTR_MODEL: f"{meter.type} {meter.full_serial_number}", - ATTR_MANUFACTURER: MANUFACTURER, - } + self._attr_device_info = DeviceInfo( + identifiers={(DOMAIN, meter.get_meter_id())}, + name=f"{meter.measurement_type.capitalize()} {meter.location.street} {meter.location.street_number}", + model=f"{meter.type} {meter.full_serial_number}", + manufacturer=MANUFACTURER, + ) @property def native_value(self) -> StateType: