Add AlarmControlPanel to NSPanel Pro #1648

Author: AlexxIT

custom_components/sonoff/__init__.py

@@ -43,6 +43,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 PLATFORMS = [
+    "alarm_control_panel",
     "binary_sensor",
     "button",
     "climate",

custom_components/sonoff/alarm_control_panel.py

@@ -0,0 +1,57 @@
+from homeassistant.components.alarm_control_panel import (
+    AlarmControlPanelEntity,
+    AlarmControlPanelEntityFeature,
+    AlarmControlPanelState,
+)
+
+from .core.const import DOMAIN
+from .core.entity import XEntity
+from .core.ewelink import SIGNAL_ADD_ENTITIES, XRegistry
+
+PARALLEL_UPDATES = 0  # fix entity_platform parallel_updates Semaphore
+
+
+async def async_setup_entry(hass, config_entry, add_entities):
+    ewelink: XRegistry = hass.data[DOMAIN][config_entry.entry_id]
+    ewelink.dispatcher_connect(
+        SIGNAL_ADD_ENTITIES,
+        lambda x: add_entities(
+            [e for e in x if isinstance(e, AlarmControlPanelEntity)]
+        ),
+    )
+
+
+STATES = {
+    0: AlarmControlPanelState.DISARMED,
+    1: AlarmControlPanelState.ARMED_HOME,
+    2: AlarmControlPanelState.ARMED_AWAY,
+    3: AlarmControlPanelState.ARMED_NIGHT,
+}
+
+
+class XAlarmPanel(XEntity, AlarmControlPanelEntity):
+    param = "securityType"
+    uid = "alarm"
+
+    _attr_code_arm_required = False
+    _attr_supported_features = (
+        AlarmControlPanelEntityFeature.ARM_HOME
+        | AlarmControlPanelEntityFeature.ARM_AWAY
+        | AlarmControlPanelEntityFeature.ARM_NIGHT
+    )
+
+    def set_state(self, params: dict):
+        if self.param in params:
+            self._attr_alarm_state = STATES.get(params[self.param])
+
+    async def async_alarm_disarm(self, code=None):
+        await self.ewelink.send(self.device, {self.param: 0})
+
+    async def async_alarm_arm_home(self, code=None):
+        await self.ewelink.send(self.device, {self.param: 1, "currentType": 1})
+
+    async def async_alarm_arm_away(self, code=None):
+        await self.ewelink.send(self.device, {self.param: 2, "currentType": 2})
+
+    async def async_alarm_arm_night(self, code=None):
+        await self.ewelink.send(self.device, {self.param: 3, "currentType": 3})

custom_components/sonoff/core/devices.py

@@ -18,6 +18,7 @@
 from homeassistant.components.switch import SwitchEntity
 
 from .ewelink import XDevice
+from ..alarm_control_panel import XAlarmPanel
 from ..binary_sensor import (
     XBinarySensor,
     XHumanSensor,
@@ -439,7 +440,7 @@ def spec(cls, base: str = None, enabled: bool = None, **kwargs) -> type:
         ),
     ],
     # NSPanel Pro, https://github.com/AlexxIT/SonoffLAN/issues/984
-    195: [XTemperatureTH],
+    195: [XTemperatureTH, XAlarmPanel],
     # Sonoff TX ULTIMATE T5-1C-86, https://github.com/AlexxIT/SonoffLAN/issues/1183
     209: [Switch1, Startup1, XT5Light, XT5Action, XT5Alarm, XT5Bell],
     # Sonoff TX ULTIMATE T5-2C-86