Only add PTZ buttons for PTZ cameras

RenierM26 · May 28, 2023 · 35a58d0 · 35a58d0
1 parent 141ac65
commit 35a58d0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/custom_components/ezviz_cloud/button.py b/custom_components/ezviz_cloud/button.py
@@ -25,6 +25,7 @@ class EzvizButtonEntityDescriptionMixin:
     """Mixin values for EZVIZ button entities."""
 
     method: Callable[[Any, Any, Any], Any]
+    supported_ext: str
 
 
 @dataclass
@@ -37,39 +38,39 @@ class EzvizButtonEntityDescription(
 BUTTON_ENTITIES = (
     EzvizButtonEntityDescription(
         key="ptz_up",
-        entity_registry_enabled_default=False,
         name="PTZ up",
         icon="mdi:pan",
         method=lambda pyezviz_client, serial, run: pyezviz_client.ptz_control(
             "UP", serial, run
         ),
+        supported_ext="154",
     ),
     EzvizButtonEntityDescription(
         key="ptz_down",
-        entity_registry_enabled_default=False,
         name="PTZ down",
         icon="mdi:pan",
         method=lambda pyezviz_client, serial, run: pyezviz_client.ptz_control(
             "DOWN", serial, run
         ),
+        supported_ext="154",
     ),
     EzvizButtonEntityDescription(
         key="ptz_left",
-        entity_registry_enabled_default=False,
         name="PTZ left",
         icon="mdi:pan",
         method=lambda pyezviz_client, serial, run: pyezviz_client.ptz_control(
             "LEFT", serial, run
         ),
+        supported_ext="154",
     ),
     EzvizButtonEntityDescription(
         key="ptz_right",
-        entity_registry_enabled_default=False,
         name="PTZ right",
         icon="mdi:pan",
         method=lambda pyezviz_client, serial, run: pyezviz_client.ptz_control(
             "RIGHT", serial, run
         ),
+        supported_ext="154",
     ),
 )
 
@@ -85,7 +86,10 @@ async def async_setup_entry(
     async_add_entities(
         EzvizButton(coordinator, camera, entity_description)
         for camera in coordinator.data
+        for capibility, value in coordinator.data[camera]["supportExt"].items()
         for entity_description in BUTTON_ENTITIES
+        if capibility == entity_description.supported_ext
+        if value == "1"
     )
 
 

diff --git a/custom_components/ezviz_cloud/manifest.json b/custom_components/ezviz_cloud/manifest.json
@@ -7,6 +7,6 @@
   "documentation": "https://www.home-assistant.io/integrations/ezviz",
   "iot_class": "cloud_polling",
   "loggers": ["paho_mqtt", "pyezviz"],
-  "requirements": ["pyezviz==0.2.0.12"],
-  "version": "0.1.0.11"
+  "requirements": ["pyezviz==0.2.0.14"],
+  "version": "0.1.0.12"
 }