Merge de8d2a7b0c into 99d25bfd56

custom_components/sws12500/utils.py

@@ -20,7 +20,7 @@ from homeassistant.helpers.translation import async_get_translations
 
 from .const import (
     AZIMUT,
-    BATTERY_LEVEL,
+    BATLEVEL,
     DATABASE_PATH,
     DEV_DBG,
     OUTSIDE_HUMIDITY,
@@ -30,7 +30,6 @@ from .const import (
     SENSORS_TO_LOAD,
     WIND_SPEED,
     UnitOfDir,
-    UnitOfBat,
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -182,33 +181,19 @@ def wind_dir_to_text(deg: float) -> UnitOfDir | None:
 
     return None
 
-
-def battery_level_to_text(battery: int) -> UnitOfBat:
+def battery_level_to_text(battery: int) -> str:
     """Return battery level in text representation.
 
-    Returns UnitOfBat
+    Returns UnitOfBat or None
     """
+    if battery is None:
+        return "unknown"
 
-    return {
-        0: UnitOfBat.LOW,
-        1: UnitOfBat.NORMAL,
-    }.get(battery, UnitOfBat.UNKNOWN)
-
-
-def battery_level_to_icon(battery: UnitOfBat) -> str:
-    """Return battery level in icon representation.
-
-    Returns str
-    """
-
-    icons = {
-        UnitOfBat.LOW: "mdi:battery-alert",
-        UnitOfBat.NORMAL: "mdi:battery",
-    }
-
-    return icons.get(battery, "mdi:battery-unknown")
-
-
+    if battery is 0:
+        return BATLEVEL[battery]
+    elif battery is 1:
+        return BATLEVEL[battery]
+    
 def fahrenheit_to_celsius(fahrenheit: float) -> float:
     """Convert Fahrenheit to Celsius."""
     return (fahrenheit - 32) * 5.0 / 9.0
@@ -295,12 +280,10 @@ def long_term_units_in_statistics_meta():
     db = conn.cursor()
 
     try:
-        db.execute(
-            """
+        db.execute("""
             SELECT statistic_id, unit_of_measurement from statistics_meta
             WHERE statistic_id LIKE 'sensor.weather_station_sws%'
-         """
-        )
+         """)
         rows = db.fetchall()
         sensor_units = {
             statistic_id: f"{statistic_id} ({unit})" for statistic_id, unit in rows
@@ -316,8 +299,8 @@ def long_term_units_in_statistics_meta():
 
 async def migrate_data(hass: HomeAssistant, sensor_id: str | None = None) -> bool:
     """Migrate data from mm/d to mm."""
-
-    _LOGGER.debug("Sensor %s is required for data migration", sensor_id)
+    
+    _LOGGER.debug("Sensor %s is required for data migration", sensor_id) 
     updated_rows = 0
 
     if not Path(DATABASE_PATH).exists():