Fix defence mode service

RenierM26 · Feb 21, 2021 · a47623a · a47623a
1 parent 1c9ecdf
commit a47623a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/custom_components/ezviz_cloud/camera.py b/custom_components/ezviz_cloud/camera.py
@@ -108,12 +108,7 @@ async def async_setup_entry(
     platform.async_register_entity_service(
         "ezviz_defence_mode_change",
         {
-            vol.Required(ATTR_SWITCH): vol.In(
-                [
-                    ATTR_HOME,
-                    ATTR_AWAY,
-                ]
-            ),
+            vol.Required(ATTR_SWITCH): vol.In([ATTR_HOME, ATTR_AWAY]),
         },
         "perform_ezviz_defence_mode_change",
     )
@@ -345,10 +340,10 @@ def perform_ezviz_sound_alarm(self, enable):
 
         self.coordinator.ezviz_client.sound_alarm(self._serial, enable)
 
-    def perform_ezviz_defence_mode_change(self, defence_type):
+    def perform_ezviz_defence_mode_change(self, switch):
         """Sound the alarm on a camera."""
-        _LOGGER.debug("EZVIZ Defence mode to %s", type)
-        service_switch = getattr(DefenseModeType, defence_type)
+        _LOGGER.debug("EZVIZ Defence mode to %s", switch)
+        service_switch = getattr(DefenseModeType, switch)
 
         self.coordinator.ezviz_client.api_set_defence_mode(service_switch.value)
 

diff --git a/custom_components/ezviz_cloud/services.yaml b/custom_components/ezviz_cloud/services.yaml
@@ -44,9 +44,12 @@ ezviz_sound_alarm:
 ezviz_defence_mode_change:
   description: 'Change ezviz defence mode to away or home.'
   fields:
-    mode:
+    switch:
       description: 'Enter HOME_MODE or AWAY_MODE.'
       example: 'HOME_MODE'
+    entity_id:
+      description: 'Name(s) of entities to perform action on.'
+      example: 'camera.living_room_camera'
 ezviz_wake_device:
   description: 'This can also be used to wake device from hibernation.'
   fields: