various incl esphome devices

2025-02-15 17:16:41 +13:00
parent 70ed59fef8
commit 92a54f7ae5
12 changed files with 1758 additions and 489 deletions
--- a/custom_zha_quirks/ts0601_smoke_tuya.py
+++ b/custom_zha_quirks/ts0601_smoke_tuya.py
@@ -0,0 +1,120 @@
+"""Smoke Sensor."""
+import logging
+
+import zigpy.profiles.zha
+from zigpy.quirks import CustomCluster, CustomDevice
+import zigpy.types as t
+from zigpy.zcl.clusters.general import Basic, Groups, Ota, Scenes, Time
+from zigpy.zcl.clusters.security import IasZone
+
+from zhaquirks import Bus
+from zhaquirks.const import (
+    DEVICE_TYPE,
+    ENDPOINTS,
+    INPUT_CLUSTERS,
+    MODELS_INFO,
+    OUTPUT_CLUSTERS,
+    PROFILE_ID,
+    ZONE_STATUS,
+    ZONE_TYPE,
+)
+from zhaquirks.tuya import TuyaManufCluster, TuyaManufClusterAttributes
+
+_LOGGER = logging.getLogger(__name__)
+
+TUYA_SMOKE_DETECTED_ATTR = 0x0401  # [0]/[1] [Detected]/[Clear]!
+
+
+class TuyaSmokeDetectorCluster(TuyaManufClusterAttributes):
+    """Manufacturer Specific Cluster of the TS0601 smoke detector."""
+
+    attributes = {
+        TUYA_SMOKE_DETECTED_ATTR: ("smoke_detected", t.uint8_t, True),
+    }
+
+    def _update_attribute(self, attrid, value):
+        super()._update_attribute(attrid, value)
+        if attrid == TUYA_SMOKE_DETECTED_ATTR:
+            if value == 0:
+                self.endpoint.device.ias_bus.listener_event(
+                    "update_zone_status", IasZone.ZoneStatus.Alarm_1
+                )
+            else:
+                self.endpoint.device.ias_bus.listener_event("update_zone_status", 0)
+        else:
+            _LOGGER.warning(
+                "[0x%04x:%s:0x%04x] unhandled attribute: 0x%04x",
+                self.endpoint.device.nwk,
+                self.endpoint.endpoint_id,
+                self.cluster_id,
+                attrid,
+            )
+
+
+class TuyaIasZone(CustomCluster, IasZone):
+    """IAS Zone."""
+
+    _CONSTANT_ATTRIBUTES = {ZONE_TYPE: IasZone.ZoneType.Fire_Sensor}
+
+    def __init__(self, *args, **kwargs):
+        """Init."""
+        super().__init__(*args, **kwargs)
+        self.endpoint.device.ias_bus.add_listener(self)
+
+    def update_zone_status(self, value):
+        """Update IAS status."""
+        super()._update_attribute(ZONE_STATUS, value)
+
+
+class TuyaSmokeDetector0601(CustomDevice):
+    """TS0601 Smoke detector quirk."""
+
+    def __init__(self, *args, **kwargs):
+        """Init."""
+        self.ias_bus = Bus()
+        super().__init__(*args, **kwargs)
+
+    signature = {
+        MODELS_INFO: [
+            ("_TZE200_aycxwiau", "TS0601"),
+            ("_TZE200_ntcy3xu1", "TS0601"),
+            ("_TZE200_vzekyi4c", "TS0601"),
+            ("_TZE200_uebojraa", "TS0601"),
+        ],
+        ENDPOINTS: {
+            1: {
+                PROFILE_ID: zigpy.profiles.zha.PROFILE_ID,
+                DEVICE_TYPE: zigpy.profiles.zha.DeviceType.SMART_PLUG,
+                INPUT_CLUSTERS: [
+                    Basic.cluster_id,
+                    Groups.cluster_id,
+                    Scenes.cluster_id,
+                    TuyaManufCluster.cluster_id,
+                ],
+                OUTPUT_CLUSTERS: [
+                    Time.cluster_id,
+                    Ota.cluster_id,
+                ],
+            },
+        },
+    }
+
+    replacement = {
+        ENDPOINTS: {
+            1: {
+                PROFILE_ID: zigpy.profiles.zha.PROFILE_ID,
+                DEVICE_TYPE: zigpy.profiles.zha.DeviceType.IAS_ZONE,
+                INPUT_CLUSTERS: [
+                    Basic.cluster_id,
+                    Groups.cluster_id,
+                    Scenes.cluster_id,
+                    TuyaIasZone,
+                    TuyaSmokeDetectorCluster,
+                ],
+                OUTPUT_CLUSTERS: [
+                    Time.cluster_id,
+                    Ota.cluster_id,
+                ],
+            },
+        },
+    }