Home Assistant Git Exporter

config/custom_components/apsystems_ecur/binary_sensor.py

@@ -0,0 +1,92 @@
+from datetime import timedelta
+import logging
+
+from homeassistant.components.binary_sensor import (
+    BinarySensorEntity,
+)
+from homeassistant.helpers.entity import EntityCategory
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+)
+
+from .const import (
+    DOMAIN,
+    RELOAD_ICON,
+    CACHE_ICON,
+    RESTART_ICON
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+async def async_setup_entry(hass, config, add_entities, discovery_info=None):
+
+    ecu = hass.data[DOMAIN].get("ecu")
+    coordinator = hass.data[DOMAIN].get("coordinator")
+
+    sensors = [
+        APSystemsECUBinarySensor(coordinator, ecu, "data_from_cache", 
+            label="Using Cached Data", icon=CACHE_ICON),
+        APSystemsECUBinarySensor(coordinator, ecu, "restart_ecu",
+            label="Restart", icon=RESTART_ICON)
+    ]
+    add_entities(sensors)
+
+
+class APSystemsECUBinarySensor(CoordinatorEntity, BinarySensorEntity):
+
+    def __init__(self, coordinator, ecu, field, label=None, devclass=None, icon=None):
+
+        super().__init__(coordinator)
+
+        self.coordinator = coordinator
+
+        self._ecu = ecu
+        self._field = field
+        self._label = label
+        if not label:
+            self._label = field
+        self._icon = icon
+
+        self._name = f"ECU {self._label}"
+        self._state = None
+
+    @property
+    def unique_id(self):
+        return f"{self._ecu.ecu.ecu_id}_{self._field}"
+
+    @property
+    def name(self):
+        return self._name
+
+    @property
+    def is_on(self):
+        return self.coordinator.data.get(self._field)
+
+    @property
+    def icon(self):
+        return self._icon
+
+    @property
+    def extra_state_attributes(self):
+
+        attrs = {
+            "ecu_id" : self._ecu.ecu.ecu_id,
+            "firmware" : self._ecu.ecu.firmware,
+            "timezone" : self._ecu.ecu.timezone,
+            "last_update" : self._ecu.ecu.last_update
+        }
+        return attrs
+
+    @property
+    def entity_category(self):
+        return EntityCategory.DIAGNOSTIC
+
+    @property
+    def device_info(self):
+        parent = f"ecu_{self._ecu.ecu.ecu_id}"
+        return {
+            "identifiers": {
+                (DOMAIN, parent),
+            }
+        }
+