diff --git a/homeassistant/components/linky/sensor.py b/homeassistant/components/linky/sensor.py index bd2d38735d6..5ff04c5ee70 100644 --- a/homeassistant/components/linky/sensor.py +++ b/homeassistant/components/linky/sensor.py @@ -18,6 +18,8 @@ from homeassistant.helpers.entity import Entity from homeassistant.helpers.event import async_track_time_interval from homeassistant.helpers.typing import HomeAssistantType +from .const import DOMAIN + _LOGGER = logging.getLogger(__name__) SCAN_INTERVAL = timedelta(hours=4) @@ -28,17 +30,6 @@ INDEX_CURRENT = -1 INDEX_LAST = -2 ATTRIBUTION = "Data provided by Enedis" -SENSORS = { - "yesterday": ("Linky yesterday", DAILY, INDEX_LAST), - "current_month": ("Linky current month", MONTHLY, INDEX_CURRENT), - "last_month": ("Linky last month", MONTHLY, INDEX_LAST), - "current_year": ("Linky current year", YEARLY, INDEX_CURRENT), - "last_year": ("Linky last year", YEARLY, INDEX_LAST), -} -SENSORS_INDEX_LABEL = 0 -SENSORS_INDEX_SCALE = 1 -SENSORS_INDEX_WHEN = 2 - async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Old way of setting up the Linky platform.""" @@ -114,6 +105,12 @@ class LinkySensor(Entity): self._username = account.username self._time = None self._consumption = None + self._unique_id = f"{self._username}_{scale}_{when}" + + @property + def unique_id(self): + """Return a unique ID.""" + return self._unique_id @property def name(self): @@ -144,6 +141,15 @@ class LinkySensor(Entity): CONF_USERNAME: self._username, } + @property + def device_info(self): + """Return device information.""" + return { + "identifiers": {(DOMAIN, self.unique_id)}, + "name": self.name, + "manufacturer": "Enedis", + } + async def async_update(self) -> None: """Retrieve the new data for the sensor.""" data = self._account.data[self._scale][self._when]