From 2d1af86a59b498034b4c1bc0408df05e8c864351 Mon Sep 17 00:00:00 2001
From: figorr <77553975+figorr@users.noreply.github.com>
Date: Fri, 5 Jan 2024 08:18:01 +0100
Subject: [PATCH] Update sensor.py to solve deprecated constant

To solve log error after updating HA Core 2024.1

Logger: homeassistant.const
Source: helpers/deprecation.py:204

VOLUME_LITERS was used from ecowater_softener, this is a deprecated constant which will be removed in HA Core 2025.1. Use UnitOfVolume.LITERS instead, please create a bug report at https://github.com/barleybobs/homeassistant-ecowater-softener/issues
VOLUME_GALLONS was used from ecowater_softener, this is a deprecated constant which will be removed in HA Core 2025.1. Use UnitOfVolume.GALLONS instead, please create a bug report at https://github.com/barleybobs/homeassistant-ecowater-softener/issues
TIME_DAYS was used from ecowater_softener, this is a deprecated constant which will be removed in HA Core 2025.1. Use UnitOfTime.DAYS instead, please create a bug report at https://github.com/barleybobs/homeassistant-ecowater-softener/issues
---
 custom_components/ecowater_softener/sensor.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/custom_components/ecowater_softener/sensor.py b/custom_components/ecowater_softener/sensor.py
index cacb9f9..d7b9a95 100644
--- a/custom_components/ecowater_softener/sensor.py
+++ b/custom_components/ecowater_softener/sensor.py
@@ -14,9 +14,8 @@
 from homeassistant.helpers.typing import StateType
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.const import (
-    VOLUME_LITERS,
-    VOLUME_GALLONS,
-    TIME_DAYS,
+    UnitOfVolume,
+    UnitOfTime,
     PERCENTAGE,
 )
 
@@ -80,7 +79,7 @@ class EcowaterSensorEntityDescription(SensorEntityDescription):
         key=DAYS_UNTIL_OUT_OF_SALT,
         name="Days until out of salt",
         icon="mdi:calendar",
-        native_unit_of_measurement=TIME_DAYS,
+        native_unit_of_measurement=UnitOfTime.DAYS,
     ),
     EcowaterSensorEntityDescription(
         key=RECHARGE_ENABLED,
@@ -137,9 +136,9 @@ def __init__(
     def native_unit_of_measurement(self) -> StateType:
         if self.entity_description.key.startswith('water'):
             if self.coordinator.data['water_units'].lower() == 'liters':
-                return VOLUME_LITERS
+                return UnitOfVolume.LITERS
             elif self.coordinator.data['water_units'].lower() == 'gallons':
-                return VOLUME_GALLONS
+                return UnitOfVolume.GALLONS
         elif self.entity_description.native_unit_of_measurement != None:
             return self.entity_description.native_unit_of_measurement