Removed 'stop' support, and add 'closing' and 'opening' status.

corporategoth · May 10, 2023 · 14306fe · 14306fe
1 parent 899eae7
commit 14306fe
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 12 deletions.
diff --git a/custom_components/powerpetdoor/const.py b/custom_components/powerpetdoor/const.py
@@ -6,9 +6,6 @@
     CONF_PORT,
     CONF_TIMEOUT,
     ATTR_ENTITY_ID,
-    SERVICE_TOGGLE,
-    SERVICE_OPEN,
-    SERVICE_CLOSE,
 )
 
 DOMAIN = "powerpetdoor"

diff --git a/custom_components/powerpetdoor/cover.py b/custom_components/powerpetdoor/cover.py
@@ -6,8 +6,7 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, CoordinatorEntity
 from homeassistant.config_entries import ConfigEntry, SOURCE_IMPORT
-from homeassistant.components.cover import CoverEntity, CoverDeviceClass
-from homeassistant.helpers import entity_platform
+from homeassistant.components.cover import CoverEntity, CoverDeviceClass, SUPPORT_CLOSE, SUPPORT_OPEN
 from .client import PowerPetDoorClient
 
 from .const import (
@@ -31,9 +30,6 @@
     CMD_CLOSE,
     STATE_LAST_CHANGE,
     FIELD_DOOR_STATUS,
-    SERVICE_OPEN,
-    SERVICE_CLOSE,
-    SERVICE_TOGGLE,
 )
 
 import logging
@@ -42,6 +38,7 @@
 
 class PetDoor(CoordinatorEntity, CoverEntity):
     _attr_device_class = CoverDeviceClass.SHUTTER
+    _attr_supported_features = (SUPPORT_CLOSE | SUPPORT_OPEN)
     _attr_position = None
 
     last_change = None
@@ -100,6 +97,16 @@ def current_cover_position(self) -> int | None:
         elif self.coordinator.data in (DOOR_STATE_RISING, DOOR_STATE_CLOSING_MID_OPEN):
             return 33
 
+    @property
+    def is_opening(self) -> bool | None:
+        """Return True if entity is on."""
+        return (self.coordinator.data in (DOOR_STATE_RISING, DOOR_STATE_SLOWING))
+
+    @property
+    def is_closing(self) -> bool | None:
+        """Return True if entity is on."""
+        return (self.coordinator.data in (DOOR_STATE_CLOSING_TOP_OPEN, DOOR_STATE_CLOSING_MID_OPEN))
+
     @property
     def is_closed(self) -> bool | None:
         """Return True if entity is on."""

diff --git a/custom_components/powerpetdoor/manifest.json b/custom_components/powerpetdoor/manifest.json
@@ -9,5 +9,5 @@
   "issue_tracker": "https://github.com/corporategoth/ha-powerpetdoor/issues",
   "quality_scale": "gold",
   "requirements": [],
-  "version": "0.3.0"
+  "version": "0.3.1"
 }
diff --git a/custom_components/powerpetdoor/switch.py b/custom_components/powerpetdoor/switch.py
@@ -31,9 +31,6 @@
     FIELD_OUTSIDE,
     FIELD_POWER,
     FIELD_AUTO,
-    SERVICE_OPEN,
-    SERVICE_CLOSE,
-    SERVICE_TOGGLE,
 )
 
 import logging