Added High Water Alarm check

theOrakle · Feb 23, 2024 · 0cf24b1 · 0cf24b1
1 parent 1e8f7ba
commit 0cf24b1
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 18 deletions.
diff --git a/custom_components/glentronics/api.py b/custom_components/glentronics/api.py
@@ -7,6 +7,8 @@
 import aiohttp
 import async_timeout
 
+from .const import LOGGER
+
 API_URL = "https://api.glentronicsconnect.com"
 API_USERNAME = "Glentronics"
 API_PASSWORD = "API201622@"
@@ -91,6 +93,8 @@ async def async_get_data(self) -> any:
             data[proxy] = details.get("StatusList")[0]
             data[proxy]["StatusFields"] = details.get("StatusFields")
             data[proxy]["Proxy"] = details.get("Location")
+            data[proxy]["HasWaterAlarm"] = details.get("UsedAsHighWaterAlarm")
+        LOGGER.debug(data)
         return data
 
     async def _api_wrapper(

diff --git a/custom_components/glentronics/binary_sensor.py b/custom_components/glentronics/binary_sensor.py
@@ -13,19 +13,12 @@
 
 ENTITY_DESCRIPTIONS = (
     BinarySensorEntityDescription(
-        key="Alarm Status",
+        key="Alarm Status (USB)",
         translation_key="FieldStatusOK",
         name="Status",
         icon="mdi:usb",
         device_class=BinarySensorDeviceClass.PROBLEM,
     ),
-    BinarySensorEntityDescription(
-        key="High Water Detector Status",
-        translation_key="FieldStatusOK",
-        name="High Water",
-        icon="mdi:home-flood",
-        device_class=BinarySensorDeviceClass.MOISTURE,
-    ),
     BinarySensorEntityDescription(
         key="WiFi Module Status",
         translation_key="FieldStatusOK",
@@ -52,16 +45,42 @@
 
 async def async_setup_entry(hass, entry, async_add_devices):
     """Set up the binary_sensor platform."""
+    devices = []
     coordinator = hass.data[DOMAIN][entry.entry_id]
-    async_add_devices(
-        GlentronicsBinarySensor(
-            coordinator=coordinator,
-            entity_description=entity_description,
-            proxy=proxy,
+    for proxy in coordinator.data:
+        water_alarm = coordinator.data[proxy].get("HasWaterAlarm")
+        if water_alarm:
+            entity_description = BinarySensorEntityDescription(
+                key="High Water Detector Status",
+                translation_key="FieldStatusOK",
+                name="High Water",
+                icon="mdi:home-flood",
+                device_class=BinarySensorDeviceClass.MOISTURE,
+            )
+        else:
+            entity_description = BinarySensorEntityDescription(
+                key="Alarm Status (Remote Terminals)",
+                translation_key="FieldStatusOK",
+                name="Remote Terminals",
+                icon="mdi:chip",
+                device_class=BinarySensorDeviceClass.PROBLEM,
+            )
+        devices.append(
+            GlentronicsBinarySensor(
+                coordinator=coordinator,
+                entity_description=entity_description,
+                proxy=proxy,
+            )
         )
-        for entity_description in ENTITY_DESCRIPTIONS
-        for proxy in coordinator.data
-    )
+        for entity_description in ENTITY_DESCRIPTIONS:
+            devices.append(
+                GlentronicsBinarySensor(
+                    coordinator=coordinator,
+                    entity_description=entity_description,
+                    proxy=proxy,
+                )
+            )
+    async_add_devices(devices)
 
 
 class GlentronicsBinarySensor(GlentronicsEntity, BinarySensorEntity):

diff --git a/custom_components/glentronics/manifest.json b/custom_components/glentronics/manifest.json
@@ -8,7 +8,7 @@
   "issue_tracker": "https://github.com/theOrakle/glentronics/issues",
   "loggers": ["custom_components.glentronics"],
   "requirements": [],
-  "version": "1.0.4",
+  "version": "1.0.5",
   "dependencies": [],
   "codeowners": ["@theOrakle"]
 }
diff --git a/hacs.json b/hacs.json
@@ -1,6 +1,6 @@
 {
   "name": "Glentronics",
-  "domains": ["binary_sensor", "sensor"],
+  "domains": ["binary_sensor"],
   "homeassistant": "2023.1.0",
   "iot_class": "Cloud Polling",
   "render_readme": true