Add support new SNZB-02WD #1612

Author: AlexxIT

custom_components/sonoff/core/devices.py

@@ -57,12 +57,13 @@
     XEnergySensorPOWR3,
     XEnergyTotal,
     XHexVoltageTRVZB,
+    XHumCorrection,
     XHumidityTH,
     XOutdoorTempNS,
     XRemoteButton,
     XSensor,
     XT5Action,
-    XTemperatureNS,
+    XTempCorrection,
     XTemperatureTH,
     XTodayWaterUsage,
     XUnknown,
@@ -322,7 +323,7 @@ def spec(cls, base: str = None, enabled: bool = None, **kwargs) -> type:
         Switch1,
         Switch2,
         XClimateNS,
-        XTemperatureNS,
+        XTempCorrection,
         XOutdoorTempNS,
     ],  # Sonoff NS Panel
     # https://github.com/AlexxIT/SonoffLAN/issues/1026
@@ -573,6 +574,8 @@ def spec(cls, base: str = None, enabled: bool = None, **kwargs) -> type:
         spec(XSensor100, param="current"),
         spec(XSensor100, param="voltage"),
     ],
+    # SNZB-02WD https://github.com/AlexxIT/SonoffLAN/issues/1612
+    7033: [XTempCorrection, XHumCorrection, Battery, ZRSSI],
 }
 
 

custom_components/sonoff/sensor.py

@@ -267,19 +267,40 @@ class XEnergyTotal(XSensor):
     _attr_state_class = SensorStateClass.TOTAL_INCREASING
 
 
-class XTemperatureNS(XSensor):
+def parse_float(v: int | float | str):
+    return float(v) if isinstance(v, str) else v
+
+
+class XTempCorrection(XSensor):
     params = {"temperature", "tempCorrection"}
     uid = "temperature"
 
     def set_state(self, params: dict = None, value: float = None):
-        if params:
-            # cache updated in XClimateNS entity
-            cache = self.device["params"]
-            value = cache["temperature"]
-            # fix str https://github.com/AlexxIT/SonoffLAN/issues/1628
+        try:
+            if (cache := self.device["params"]) != params:
+                cache.update(params)
+            value = parse_float(cache["temperature"])
             if v := cache.get("tempCorrection"):
-                value += float(v)
-        XSensor.set_state(self, value=value)
+                value += parse_float(v)
+            XSensor.set_state(self, value=value)
+        except:
+            pass
+
+
+class XHumCorrection(XSensor):
+    params = {"humidity", "humCorrection"}
+    uid = "humidity"
+
+    def set_state(self, params: dict = None, value: float = None):
+        try:
+            if (cache := self.device["params"]) != params:
+                cache.update(params)
+            value = parse_float(cache["humidity"])
+            if v := cache.get("humCorrection"):
+                value += parse_float(v)
+            XSensor.set_state(self, value=value)
+        except:
+            pass
 
 
 class XOutdoorTempNS(XSensor):

tests/test_entity.py

@@ -53,7 +53,7 @@
     XRemoteButton,
     XSensor,
     XT5Action,
-    XTemperatureNS,
+    XTempCorrection,
     XUnknown,
 )
 from custom_components.sonoff.switch import (
@@ -1207,7 +1207,7 @@ def test_ns_panel():
     for uid in ("1", "2"):
         assert any(e.uid == uid for e in entities)
 
-    temp = next(e for e in entities if isinstance(e, XTemperatureNS))
+    temp = next(e for e in entities if isinstance(e, XTempCorrection))
     state = temp.hass.states.get(temp.entity_id)
     assert state.state == "20"
     assert state.attributes == {