Add temperature and humidity offset to Sonoff SNZB-02D (#3989)

Co-authored-by: TheJulianJES <[email protected]>

Author: fgsch

zhaquirks/sonoff/snzb02d.py

@@ -1,7 +1,7 @@
 """Sonoff SNZB-02D - Zigbee LCD smart temperature humidity sensor."""
 
 from zigpy.quirks import CustomCluster
-from zigpy.quirks.v2 import QuirkBuilder
+from zigpy.quirks.v2 import NumberDeviceClass, QuirkBuilder
 from zigpy.quirks.v2.homeassistant import PERCENTAGE, UnitOfTemperature
 import zigpy.types as t
 from zigpy.zcl import foundation
@@ -50,6 +50,16 @@ class AttributeDefs(BaseAttributeDefs):
             zcl_type=DataTypeId.uint16,
         )
 
+        temperature_offset = ZCLAttributeDef(
+            id=0x2003,
+            type=t.int16s,
+        )
+
+        humidity_offset = ZCLAttributeDef(
+            id=0x2004,
+            type=t.int16s,
+        )
+
 
 (
     QuirkBuilder("SONOFF", "SNZB-02D")
@@ -60,6 +70,7 @@ class AttributeDefs(BaseAttributeDefs):
         min_value=-10,
         max_value=60,
         step=0.1,
+        device_class=NumberDeviceClass.TEMPERATURE,
         unit=UnitOfTemperature.CELSIUS,
         multiplier=0.01,
         translation_key="comfort_temperature_min",
@@ -71,6 +82,7 @@ class AttributeDefs(BaseAttributeDefs):
         min_value=-10,
         max_value=60,
         step=0.1,
+        device_class=NumberDeviceClass.TEMPERATURE,
         unit=UnitOfTemperature.CELSIUS,
         multiplier=0.01,
         translation_key="comfort_temperature_max",
@@ -82,6 +94,7 @@ class AttributeDefs(BaseAttributeDefs):
         min_value=5,
         max_value=95,
         step=0.1,
+        device_class=NumberDeviceClass.HUMIDITY,
         unit=PERCENTAGE,
         multiplier=0.01,
         translation_key="comfort_humidity_min",
@@ -93,6 +106,7 @@ class AttributeDefs(BaseAttributeDefs):
         min_value=5,
         max_value=95,
         step=0.1,
+        device_class=NumberDeviceClass.HUMIDITY,
         unit=PERCENTAGE,
         multiplier=0.01,
         translation_key="comfort_humidity_max",
@@ -105,5 +119,29 @@ class AttributeDefs(BaseAttributeDefs):
         translation_key="display_unit",
         fallback_name="Display unit",
     )
+    .number(
+        CustomSonoffCluster.AttributeDefs.temperature_offset.name,
+        CustomSonoffCluster.cluster_id,
+        min_value=-50,
+        max_value=50,
+        step=0.1,
+        device_class=NumberDeviceClass.TEMPERATURE,
+        unit=UnitOfTemperature.CELSIUS,
+        multiplier=0.01,
+        translation_key="temperature_offset",
+        fallback_name="Temperature offset",
+    )
+    .number(
+        CustomSonoffCluster.AttributeDefs.humidity_offset.name,
+        CustomSonoffCluster.cluster_id,
+        min_value=-50,
+        max_value=50,
+        step=0.1,
+        device_class=NumberDeviceClass.HUMIDITY,
+        unit=PERCENTAGE,
+        multiplier=0.01,
+        translation_key="humidity_offset",
+        fallback_name="Humidity offset",
+    )
     .add_to_registry()
 )