Add reboot button to NSPanel Pro #1723

Author: AlexxIT

custom_components/sonoff/button.py

@@ -1,5 +1,6 @@
-from homeassistant.components.button import ButtonEntity
+from homeassistant.components.button import ButtonDeviceClass, ButtonEntity
 from homeassistant.components.script import ATTR_LAST_TRIGGERED
+from homeassistant.const import EntityCategory
 from homeassistant.helpers.entity import DeviceInfo
 
 from .core.const import DOMAIN
@@ -47,5 +48,12 @@ async def async_press(self):
 class XButton(XEntity, ButtonEntity):
     value = None
 
+    def __init__(self, ewelink: XRegistry, device: dict):
+        XEntity.__init__(self, ewelink, device)
+
+        if self.params == "reboot":
+            self._attr_device_class = ButtonDeviceClass.RESTART
+            self._attr_entity_category = EntityCategory.CONFIG
+
     async def async_press(self):
         await self.ewelink.send(self.device, {self.param: self.value})

custom_components/sonoff/core/devices.py

@@ -143,7 +143,15 @@ def spec(cls, base: str = None, enabled: bool = None, **kwargs) -> type:
 SPEC_2CH = [Switch1, Switch2, LED, RSSI]
 SPEC_3CH = [Switch1, Switch2, Switch3, LED, RSSI]
 SPEC_4CH = [Switch1, Switch2, Switch3, Switch4, LED, RSSI]
-SPEC_NSP = [XTemperatureTH, XPanelAlarm, XPanelBuzzer, XPanelScreen, XCPUTemperature]
+
+SPEC_NSP = [
+    XTemperatureTH,
+    XPanelAlarm,
+    XPanelBuzzer,
+    XPanelScreen,
+    XCPUTemperature,
+    spec(XButton, param="reboot", value=True),
+]
 
 Current1 = spec(XSensor100, param="current_00", uid="current_1")
 Current2 = spec(XSensor100, param="current_01", uid="current_2")

tests/test_entity.py

@@ -2189,3 +2189,9 @@ def test_nspanel():
 
     cpu_temp: XSensor = next(e for e in entities if e.uid == "cpu_temperature")
     assert cpu_temp.state == 42.727
+
+    reboot = next(e for e in entities if e.uid == "reboot")
+
+    registry: DummyRegistry = reboot.ewelink
+    result = registry.call(reboot.async_press())
+    assert result[1] == {"reboot": True}