Integrate to homeassistant

custom_components/midea_auto_codec/binary_sensor.py

@@ -5,58 +5,115 @@ from homeassistant.components.binary_sensor import (
 from homeassistant.const import (
     Platform,
     CONF_DEVICE_ID,
-    CONF_DEVICE,
-    CONF_ENTITIES
+    CONF_ENTITIES, CONF_DEVICE
 )
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from .const import (
     DOMAIN,
     DEVICES
 )
-from .midea_entities import MideaBinaryBaseEntity
+from .midea_entity import MideaEntity
 
 
-async def async_setup_entry(hass, config_entry, async_add_entities):
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up binary sensor entities for Midea devices."""
     device_id = config_entry.data.get(CONF_DEVICE_ID)
-    device = hass.data[DOMAIN][DEVICES][device_id].get(CONF_DEVICE)
-    manufacturer = hass.data[DOMAIN][DEVICES][device_id].get("manufacturer")
-    rationale = hass.data[DOMAIN][DEVICES][device_id].get("rationale")
-    entities = hass.data[DOMAIN][DEVICES][device_id].get(CONF_ENTITIES).get(Platform.BINARY_SENSOR)
-    devs = [MideaDeviceStatusSensorEntity(device, manufacturer, rationale,"Status", {})]
-    if entities is not None:
+    device_data = hass.data[DOMAIN][DEVICES][device_id]
+    coordinator = device_data.get("coordinator")
+    device = device_data.get(CONF_DEVICE)
+    manufacturer = device_data.get("manufacturer")
+    rationale = device_data.get("rationale")
+    entities = device_data.get(CONF_ENTITIES, {}).get(Platform.BINARY_SENSOR, {})
+    
+    devs = [MideaDeviceStatusSensorEntity(coordinator, device, manufacturer, rationale, "Status", {})]
+    if entities:
         for entity_key, config in entities.items():
-            devs.append(MideaBinarySensorEntity(device, manufacturer, rationale, entity_key, config))
+            devs.append(MideaBinarySensorEntity(
+                coordinator, device, manufacturer, rationale, entity_key, config
+            ))
     async_add_entities(devs)
 
 
-class MideaDeviceStatusSensorEntity(MideaBinaryBaseEntity, BinarySensorEntity):
+class MideaDeviceStatusSensorEntity(MideaEntity, BinarySensorEntity):
+    """Device status binary sensor."""
+
+    def __init__(self, coordinator, device, manufacturer, rationale, entity_key, config):
+        super().__init__(
+            coordinator,
+            device.device_id,
+            device.device_name,
+            f"T0x{device.device_type:02X}",
+            device.sn,
+            device.sn8,
+            device.model,
+        )
+        self._device = device
+        self._manufacturer = manufacturer
+        self._rationale = rationale
+        self._entity_key = entity_key
+        self._config = config
+
+    @property
+    def entity_id_suffix(self) -> str:
+        """Return the suffix for entity ID."""
+        return "status"
 
     @property
     def device_class(self):
+        """Return the device class."""
         return BinarySensorDeviceClass.CONNECTIVITY
 
     @property
     def icon(self):
+        """Return the icon."""
         return "mdi:devices"
 
     @property
     def is_on(self):
-        return self._device.connected
-
-    @property
-    def available(self):
-        return True
+        """Return if the device is connected."""
+        return self.coordinator.data.connected
 
     @property
     def extra_state_attributes(self) -> dict:
-        return self._device.attributes
-
-    def update_state(self, status):
-        try:
-            self.schedule_update_ha_state()
-        except Exception as e:
-            pass
+        """Return extra state attributes."""
+        return self.device_attributes
 
 
-class MideaBinarySensorEntity(MideaBinaryBaseEntity, BinarySensorEntity):
-    pass
+class MideaBinarySensorEntity(MideaEntity, BinarySensorEntity):
+    """Generic binary sensor entity."""
+
+    def __init__(self, coordinator, device, manufacturer, rationale, entity_key, config):
+        super().__init__(
+            coordinator,
+            device.device_id,
+            device.device_name,
+            f"T0x{device.device_type:02X}",
+            device.sn,
+            device.sn8,
+            device.model,
+        )
+        self._device = device
+        self._manufacturer = manufacturer
+        self._rationale = rationale
+        self._entity_key = entity_key
+        self._config = config
+
+    @property
+    def entity_id_suffix(self) -> str:
+        """Return the suffix for entity ID."""
+        return f"binary_sensor_{self._entity_key}"
+
+    @property
+    def is_on(self):
+        """Return if the binary sensor is on."""
+        value = self.device_attributes.get(self._entity_key)
+        if isinstance(value, bool):
+            return value
+        return value == 1 or value == "on" or value == "true"