Use attr_xxx fields instead of properties (#27)

custom_components/veolia/entity.py

@@ -12,11 +12,9 @@ def __init__(self, coordinator, config_entry):
         """Initialize the entity."""
         super().__init__(coordinator)
         self.config_entry = config_entry
-
-    @property
-    def unique_id(self):
-        """Return a unique ID to use for this entity."""
-        return f"{self.config_entry.entry_id}_{self.name}"
+        self._attr_unique_id = f"{self.config_entry.entry_id}_{self.name}"
+        self._attr_unit_of_measurement = VOLUME_LITERS
+        self._attr_icon = "mdi:water"
 
     @property
     def device_info(self):
@@ -33,13 +31,3 @@ def device_state_attributes(self):
         return {
             "last_report": self.coordinator.data[LAST_REPORT_TIMESTAMP],
         }
-
-    @property
-    def icon(self) -> str:
-        """Return the usage icon."""
-        return "mdi:water"
-
-    @property
-    def unit_of_measurement(self) -> str:
-        """Return liter as the unit measurement for water."""
-        return VOLUME_LITERS

custom_components/veolia/sensor.py

@@ -22,10 +22,7 @@ async def async_setup_entry(hass, entry, async_add_devices):
 class VeoliaHourlyUsageSensor(VeoliaEntity):
     """Monitors the hourly water usage."""
 
-    @property
-    def name(self):
-        """Return the name of the sensor."""
-        return "veolia_hourly_consumption"
+    _attr_name = "veolia_hourly_consumption"
 
     @property
     def state(self):
@@ -37,10 +34,7 @@ def state(self):
 class VeoliaDailyUsageSensor(VeoliaEntity):
     """Monitors the daily water usage."""
 
-    @property
-    def name(self):
-        """Return the name of the sensor."""
-        return "veolia_daily_consumption"
+    _attr_name = "veolia_daily_consumption"
 
     @property
     def state(self):
@@ -56,10 +50,7 @@ def state(self):
 class VeoliaMonthlyUsageSensor(VeoliaEntity):
     """Monitors the monthly water usage."""
 
-    @property
-    def name(self):
-        """Return the name of the sensor."""
-        return "veolia_monthly_consumption"
+    _attr_name = "veolia_monthly_consumption"
 
     @property
     def state(self):