Solve the button wake issue

Teslemetry · Feb 4, 2024 · 77e868c · 77e868c
1 parent ed27909
commit 77e868c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/custom_components/teslemetry/button.py b/custom_components/teslemetry/button.py
@@ -22,12 +22,12 @@
 class TeslemetryButtonEntityDescription(ButtonEntityDescription):
     """Describes a Teslemetry Button entity."""
 
-    func: Callable
+    func: Callable | None
 
 
 DESCRIPTIONS: tuple[TeslemetryButtonEntityDescription, ...] = (
     TeslemetryButtonEntityDescription(
-        key="wake", func=lambda api: api.wake_up()
+        key="wake"
     ),  # Every button also runs wakeup
     TeslemetryButtonEntityDescription(
         key="flash_lights", func=lambda api: api.flash_lights()
@@ -73,4 +73,5 @@ def __init__(
     async def async_press(self) -> None:
         """Press the button."""
         await self.wake_up_if_asleep()
-        await self.entity_description.func(self.api)
+        if self.entity_description.func:
+            await self.entity_description.func(self.api)