fix unloading errors, and bump to use pychlorinator v0.2.7

DanielNagy · Jan 30, 2024 · 949695e · 949695e
1 parent 008de06
commit 949695e
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/custom_components/astralpool_halo_chlorinator/__init__.py b/custom_components/astralpool_halo_chlorinator/__init__.py
@@ -56,5 +56,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Unload a config entry."""
-    await hass.config_entries.async_forward_entry_unload(entry, PLATFORMS)
-    return True
+    unload_ok = True
+    for platform in PLATFORMS:
+        if not await hass.config_entries.async_forward_entry_unload(entry, platform):
+            unload_ok = False
+
+    return unload_ok
diff --git a/custom_components/astralpool_halo_chlorinator/coordinator.py b/custom_components/astralpool_halo_chlorinator/coordinator.py
@@ -15,7 +15,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 
-class ChlorinatorDataUpdateCoordinator(DataUpdateCoordinator):
+class ChlorinatorDataUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
     """Data coordinator for getting Chlorinator updates."""
 
     def __init__(self, hass: HomeAssistant, chlorinator: HaloChlorinatorAPI) -> None:

diff --git a/custom_components/astralpool_halo_chlorinator/manifest.json b/custom_components/astralpool_halo_chlorinator/manifest.json
@@ -19,7 +19,7 @@
   "iot_class": "local_polling",
   "requirements": [
     "bluetooth-data-tools>=0.4.0",
-    "pychlorinator>=0.2.6"
+    "pychlorinator>=0.2.7"
   ],
   "version": "0.1.0"
 }
diff --git a/custom_components/astralpool_halo_chlorinator/sensor.py b/custom_components/astralpool_halo_chlorinator/sensor.py
@@ -9,6 +9,7 @@
     SensorEntity,
     SensorEntityDescription,
     SensorStateClass,
+    EntityCategory,
 )
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity import DeviceInfo
@@ -109,6 +110,7 @@
         native_unit_of_measurement="L",
         device_class=None,
         state_class=SensorStateClass.MEASUREMENT,
+        entity_category=EntityCategory.DIAGNOSTIC,
     ),
     "DosingPumpSecs": SensorEntityDescription(
         key="DosingPumpSecs",
@@ -133,6 +135,7 @@
         native_unit_of_measurement="mA",
         device_class=None,
         state_class=SensorStateClass.MEASUREMENT,
+        entity_category=EntityCategory.DIAGNOSTIC,
     ),
     "RealCelllevel": SensorEntityDescription(
         key="RealCelllevel",
@@ -141,6 +144,7 @@
         native_unit_of_measurement=None,
         device_class=None,
         state_class=SensorStateClass.MEASUREMENT,
+        entity_category=EntityCategory.DIAGNOSTIC,
     ),
     "PreviousDaysCellLoad": SensorEntityDescription(
         key="PreviousDaysCellLoad",