Merge pull request #126 from natekspencer/dev

Add restart button

Author: natekspencer

custom_components/vivint/__init__.py

@@ -29,6 +29,7 @@
 PLATFORMS = [
     Platform.ALARM_CONTROL_PANEL,
     Platform.BINARY_SENSOR,
+    Platform.BUTTON,
     Platform.CAMERA,
     Platform.CLIMATE,
     Platform.COVER,
@@ -110,7 +111,7 @@ def async_on_device_event(event_type: str, viv_device: VivintDevice) -> None:
                         ),
                     )
                 )
-                if CapabilityCategoryType.DOORBELL in device.capabilities.keys():
+                if CapabilityCategoryType.DOORBELL in device.capabilities:
                     entry.async_on_unload(
                         device.on(
                             DOORBELL_DING,

custom_components/vivint/button.py

@@ -0,0 +1,63 @@
+"""Support for Vivint buttons."""
+from __future__ import annotations
+
+from vivintpy.devices.alarm_panel import AlarmPanel
+from vivintpy.entity import UPDATE
+
+from homeassistant.components.button import (
+    ButtonDeviceClass,
+    ButtonEntity,
+    ButtonEntityDescription,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import DOMAIN
+from .hub import VivintBaseEntity, VivintHub
+
+HUB_REBOOT_ENTITY = ButtonEntityDescription(
+    key="reboot", device_class=ButtonDeviceClass.RESTART
+)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up Vivint button platform."""
+    hub: VivintHub = hass.data[DOMAIN][entry.entry_id]
+    entities = [
+        VivintButtonEntity(
+            device=alarm_panel, hub=hub, entity_description=HUB_REBOOT_ENTITY
+        )
+        for system in hub.account.systems
+        if system.is_admin
+        for alarm_panel in system.alarm_panels
+    ]
+    async_add_entities(entities)
+
+
+class VivintButtonEntity(VivintBaseEntity, ButtonEntity):
+    """A class that describes device button entities."""
+
+    device: AlarmPanel
+
+    @property
+    def available(self) -> bool:
+        """Return if entity is available."""
+        return super().available and self.device._AlarmPanel__panel.data["can_reboot"]
+
+    async def async_press(self) -> None:
+        """Handle the button press."""
+        await self.device.reboot()
+
+    async def async_added_to_hass(self) -> None:
+        """Set up a listener for the entity."""
+        await super().async_added_to_hass()
+        self.async_on_remove(
+            self.device._AlarmPanel__panel.on(
+                UPDATE, lambda _: self.async_write_ha_state()
+            )
+        )

custom_components/vivint/update.py

@@ -56,7 +56,10 @@ class VivintUpdateEntity(VivintBaseEntity, UpdateEntity):
     @property
     def in_progress(self) -> bool:
         """Update installation progress."""
-        return self.device._AlarmPanel__panel.data["sus"] != "Idle"
+        return self.device._AlarmPanel__panel.data["sus"] not in (
+            "Idle",
+            "Reboot Pending",
+        )
 
     @property
     def installed_version(self) -> str: