Add Duty binary_sensor platform to FireServiceRota integration (#43638)

homeassistant/components/fireservicerota/binary_sensor.py

@@ -0,0 +1,93 @@
+"""Binary Sensor platform for FireServiceRota integration."""
+import logging
+
+from homeassistant.components.binary_sensor import BinarySensorEntity
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.helpers.typing import HomeAssistantType
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+    DataUpdateCoordinator,
+)
+
+from .const import DATA_COORDINATOR, DOMAIN as FIRESERVICEROTA_DOMAIN
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistantType, entry: ConfigEntry, async_add_entities
+) -> None:
+    """Set up FireServiceRota binary sensor based on a config entry."""
+
+    coordinator: DataUpdateCoordinator = hass.data[FIRESERVICEROTA_DOMAIN][
+        entry.entry_id
+    ][DATA_COORDINATOR]
+
+    async_add_entities([ResponseBinarySensor(coordinator, entry)])
+
+
+class ResponseBinarySensor(CoordinatorEntity, BinarySensorEntity):
+    """Representation of an FireServiceRota sensor."""
+
+    def __init__(self, coordinator: DataUpdateCoordinator, entry):
+        """Initialize."""
+        super().__init__(coordinator)
+        self._unique_id = f"{entry.unique_id}_Duty"
+
+        self._state = None
+
+    @property
+    def name(self) -> str:
+        """Return the name of the sensor."""
+        return "Duty"
+
+    @property
+    def icon(self) -> str:
+        """Return the icon to use in the frontend."""
+        return "mdi:calendar"
+
+    @property
+    def unique_id(self) -> str:
+        """Return the unique ID for this binary sensor."""
+        return self._unique_id
+
+    @property
+    def is_on(self):
+        """Return the state of the binary sensor."""
+        if not self.coordinator.data:
+            return
+
+        data = self.coordinator.data
+        if "available" in data and data["available"]:
+            self._state = True
+        else:
+            self._state = False
+
+        _LOGGER.debug("Set state of entity 'Duty Binary Sensor' to '%s'", self._state)
+        return self._state
+
+    @property
+    def device_state_attributes(self):
+        """Return available attributes for binary sensor."""
+        attr = {}
+        if not self.coordinator.data:
+            return attr
+
+        data = self.coordinator.data
+        attr = {
+            key: data[key]
+            for key in (
+                "start_time",
+                "end_time",
+                "available",
+                "active",
+                "assigned_function_ids",
+                "skill_ids",
+                "type",
+                "assigned_function",
+            )
+            if key in data
+        }
+
+        _LOGGER.debug("Set attributes of entity 'Duty Binary Sensor' to '%s'", attr)
+        return attr