Add support SNZB-06P #1245

AlexxIT · May 3, 2024 · af72518 · af72518
1 parent 20a7051
commit af72518
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 4 deletions.
diff --git a/custom_components/sonoff/binary_sensor.py b/custom_components/sonoff/binary_sensor.py
@@ -71,6 +71,24 @@ def set_state(self, params: dict):
             self._attr_is_on = False
 
 
+class XHumanSensor(XEntity, BinarySensorEntity):
+    param = "human"
+    uid = "occupancy"
+    _attr_device_class = BinarySensorDeviceClass.OCCUPANCY
+
+    def set_state(self, params: dict):
+        self._attr_is_on = params[self.param] == 1
+
+
+class XLightSensor(XEntity, BinarySensorEntity):
+    param = "brState"
+    uid = "light"
+    _attr_device_class = BinarySensorDeviceClass.LIGHT
+
+    def set_state(self, params: dict):
+        self._attr_is_on = params[self.param] == "brighter"
+
+
 # noinspection PyAbstractClass
 class XRemoteSensor(BinarySensorEntity, RestoreEntity):
     _attr_is_on = False

diff --git a/custom_components/sonoff/core/devices.py b/custom_components/sonoff/core/devices.py
@@ -18,7 +18,13 @@
 from homeassistant.components.switch import SwitchEntity
 
 from .ewelink import XDevice
-from ..binary_sensor import XBinarySensor, XWiFiDoor, XZigbeeMotion
+from ..binary_sensor import (
+    XBinarySensor,
+    XWiFiDoor,
+    XZigbeeMotion,
+    XHumanSensor,
+    XLightSensor,
+)
 from ..climate import XClimateNS, XClimateTH, XThermostat
 from ..core.entity import XEntity
 from ..cover import XCover, XCoverDualR3, XZigbeeCover, XCover91
@@ -39,7 +45,7 @@
     XT5Light,
     XZigbeeLight,
 )
-from ..number import XPulseWidth
+from ..number import XPulseWidth, XSensitivity
 from ..remote import XRemote
 from ..sensor import (
     XEnergySensor,
@@ -408,7 +414,7 @@ def spec(cls, base: str = None, enabled: bool = None, **kwargs) -> type:
         spec(XSensor100, param="humidity"),
         Battery,
     ],  # https://github.com/AlexxIT/SonoffLAN/issues/1166
-    7016: [ZRSSI],  # SNZB-06P
+    7016: [XHumanSensor, XLightSensor, XSensitivity, ZRSSI],  # SNZB-06P
 }
 
 

diff --git a/custom_components/sonoff/core/entity.py b/custom_components/sonoff/core/entity.py
@@ -12,9 +12,10 @@
     "battery": EntityCategory.DIAGNOSTIC,
     "battery_voltage": EntityCategory.DIAGNOSTIC,
     "led": EntityCategory.CONFIG,
-    "rssi": EntityCategory.DIAGNOSTIC,
     "pulse": EntityCategory.CONFIG,
     "pulseWidth": EntityCategory.CONFIG,
+    "rssi": EntityCategory.DIAGNOSTIC,
+    "sensitivity": EntityCategory.CONFIG,
 }
 
 ICONS = {

diff --git a/custom_components/sonoff/number.py b/custom_components/sonoff/number.py
@@ -57,3 +57,11 @@ async def async_set_native_value(self, value: float) -> None:
         await self.ewelink.send(
             self.device, {"pulse": "on", "pulseWidth": int(value / 0.5) * 500}
         )
+
+
+class XSensitivity(XNumber):
+    param = "sensitivity"
+
+    _attr_entity_registry_enabled_default = False
+    _attr_native_max_value = 3
+    _attr_native_min_value = 1