first commit

2024-01-11 15:10:51 +03:00
commit eddf07c652
602 changed files with 67154 additions and 0 deletions
@@ -0,0 +1,58 @@
+from homeassistant.components.alarm_control_panel import \
+    SUPPORT_ALARM_ARM_AWAY, SUPPORT_ALARM_ARM_HOME, SUPPORT_ALARM_ARM_NIGHT, \
+    SUPPORT_ALARM_TRIGGER, AlarmControlPanelEntity
+from homeassistant.const import STATE_ALARM_TRIGGERED
+from homeassistant.core import callback
+
+from . import DOMAIN
+from .core.converters import Converter
+from .core.device import XDevice
+from .core.entity import XEntity
+from .core.gateway import XGateway
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    def setup(gateway: XGateway, device: XDevice, conv: Converter):
+        if conv.attr in device.entities:
+            entity: XEntity = device.entities[conv.attr]
+            entity.gw = gateway
+        else:
+            entity = XiaomiAlarm(gateway, device, conv)
+        async_add_entities([entity])
+
+    gw: XGateway = hass.data[DOMAIN][config_entry.entry_id]
+    gw.add_setup(__name__, setup)
+
+
+# noinspection PyAbstractClass
+class XiaomiAlarm(XEntity, AlarmControlPanelEntity):
+    _attr_code_arm_required = False
+    _attr_supported_features = (
+            SUPPORT_ALARM_ARM_HOME | SUPPORT_ALARM_ARM_AWAY |
+            SUPPORT_ALARM_ARM_NIGHT | SUPPORT_ALARM_TRIGGER
+    )
+
+    @callback
+    def async_set_state(self, data: dict):
+        if self.attr in data:
+            self._attr_state = data[self.attr]
+        if data.get("alarm_trigger") is True:
+            self._attr_state = STATE_ALARM_TRIGGERED
+
+    async def async_alarm_disarm(self, code=None):
+        await self.device_send({self.attr: "disarmed"})
+
+    async def async_alarm_arm_home(self, code=None):
+        await self.device_send({self.attr: "armed_home"})
+
+    async def async_alarm_arm_away(self, code=None):
+        await self.device_send({self.attr: "armed_away"})
+
+    async def async_alarm_arm_night(self, code=None):
+        await self.device_send({self.attr: "armed_night"})
+
+    async def async_alarm_trigger(self, code=None):
+        await self.device_send({"alarm_trigger": True})
+
+    async def async_update(self):
+        await self.device_read(self.subscribed_attrs)