Merge pull request #29 from Jezza34000/develop

♿ add fastpolling for switch action
Jezza34000 · Feb 11, 2025 · 328dff6 · 328dff6
2 parents a03b18a + 9428959
commit 328dff6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -39,7 +39,7 @@ repos:
       - id: black
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.9.4
+    rev: v0.9.5
     hooks:
       - id: ruff
         args:

diff --git a/custom_components/petkit/button.py b/custom_components/petkit/button.py
@@ -99,6 +99,7 @@ class PetKitButtonDesc(PetKitDescSensorBase, ButtonEntityDescription):
                 {DeviceAction.START: LBCommand.CLEANING},
             ),
             only_for_types=DEVICES_LITTER_BOX,
+            is_available=lambda device: device.state.work_state is None,
         ),
         PetKitButtonDesc(
             key="Maintenance mode",
@@ -109,6 +110,7 @@ class PetKitButtonDesc(PetKitDescSensorBase, ButtonEntityDescription):
                 {DeviceAction.START: LBCommand.MAINTENANCE},
             ),
             only_for_types=[T4],
+            is_available=lambda device: device.state.work_state is None,
         ),
         PetKitButtonDesc(
             key="Exit maintenance mode",
@@ -119,6 +121,8 @@ class PetKitButtonDesc(PetKitDescSensorBase, ButtonEntityDescription):
                 {DeviceAction.END: LBCommand.MAINTENANCE},
             ),
             only_for_types=[T4],
+            is_available=lambda device: device.state.work_state is not None
+            and device.state.work_state.work_mode == 9,
         ),
         PetKitButtonDesc(
             key="Dump litter",
@@ -129,6 +133,7 @@ class PetKitButtonDesc(PetKitDescSensorBase, ButtonEntityDescription):
                 {DeviceAction.START: LBCommand.DUMPING},
             ),
             only_for_types=DEVICES_LITTER_BOX,
+            is_available=lambda device: device.state.work_state is None,
         ),
         PetKitButtonDesc(
             key="Pause",

diff --git a/custom_components/petkit/switch.py b/custom_components/petkit/switch.py
@@ -5,6 +5,7 @@
 import asyncio
 from collections.abc import Callable
 from dataclasses import dataclass
+from datetime import timedelta
 from typing import TYPE_CHECKING, Any
 
 from pypetkitapi import (
@@ -22,7 +23,7 @@
 from homeassistant.components.switch import SwitchEntity, SwitchEntityDescription
 from homeassistant.const import EntityCategory
 
-from .const import LOGGER, POWER_ONLINE_STATE
+from .const import LOGGER, MIN_SCAN_INTERVAL, POWER_ONLINE_STATE
 from .entity import PetKitDescSensorBase, PetkitEntity
 
 if TYPE_CHECKING:
@@ -722,5 +723,7 @@ async def async_turn_off(self, **_: Any) -> None:
 
     async def _update_coordinator_data(self, result: bool) -> None:
         """Update the coordinator data based on the result."""
+        self.coordinator.update_interval = timedelta(seconds=MIN_SCAN_INTERVAL)
+        self.coordinator.fast_poll_tic = 3
         await asyncio.sleep(1)
         await self.coordinator.async_request_refresh()