Ezviz battery camera work mode

homeassistant/components/ezviz/select.py

@@ -2,9 +2,15 @@
 
 from __future__ import annotations
 
+from collections.abc import Callable
 from dataclasses import dataclass
 
-from pyezviz.constants import DeviceSwitchType, SoundMode
+from pyezviz.constants import (
+    BatteryCameraWorkMode,
+    DeviceCatagories,
+    DeviceSwitchType,
+    SoundMode,
+)
 from pyezviz.exceptions import HTTPError, PyEzvizError
 
 from homeassistant.components.select import SelectEntity, SelectEntityDescription
@@ -21,19 +27,92 @@
 PARALLEL_UPDATES = 1
 
 
+class EzvizSelectEntityActionBase:
+    """Base class to handle retrieval and selection of a EzvizSelectEntity."""
+
+    current_option: Callable[[EzvizSelect], str | None]
+    select_option: Callable[[EzvizSelect, str, str], None]
+
+
+class AlarmSoundModeAction(EzvizSelectEntityActionBase):
+    """Class dedicated to Alarm Sound Mode."""
+
+    @staticmethod
+    def current_option(ezvizSelect: EzvizSelect) -> str | None:
+        """Return the selected entity option to represent the entity state."""
+        sound_mode_value = getattr(
+            SoundMode, ezvizSelect.data[ezvizSelect.entity_description.key]
+        ).value
+        if sound_mode_value in [0, 1, 2]:
+            return ezvizSelect.options[sound_mode_value]
+
+        return None
+
+    @staticmethod
+    def select_option(ezvizSelect: EzvizSelect, serial: str, option: str) -> None:
+        """Change the selected option."""
+        sound_mode_value = ezvizSelect.options.index(option)
+
+        ezvizSelect.coordinator.ezviz_client.alarm_sound(serial, sound_mode_value, 1)
+
+
+class BatteryWorkModeAction(EzvizSelectEntityActionBase):
+    """Class dedicated to Battery Work Mode."""
+
+    @staticmethod
+    def current_option(ezvizSelect: EzvizSelect) -> str | None:
+        """Return the selected entity option to represent the entity state."""
+        battery_work_mode = getattr(
+            BatteryCameraWorkMode,
+            ezvizSelect.data[ezvizSelect.entity_description.key],
+            BatteryCameraWorkMode.UNKNOWN,
+        )
+        if battery_work_mode == BatteryCameraWorkMode.UNKNOWN:
+            return None
+
+        return battery_work_mode.name.lower()
+
+    @staticmethod
+    def select_option(ezvizSelect: EzvizSelect, serial: str, option: str) -> None:
+        """Change the selected option."""
+        battery_work_mode = getattr(BatteryCameraWorkMode, option.upper())
+
+        ezvizSelect.coordinator.ezviz_client.set_battery_camera_work_mode(
+            serial, battery_work_mode.value
+        )
+
+
 @dataclass(frozen=True, kw_only=True)
 class EzvizSelectEntityDescription(SelectEntityDescription):
     """Describe a EZVIZ Select entity."""
 
     supported_switch: int
+    action_handler: type[EzvizSelectEntityActionBase]
 
 
-SELECT_TYPE = EzvizSelectEntityDescription(
+ALARM_SOUND_MODE_SELECT_TYPE = EzvizSelectEntityDescription(
     key="alarm_sound_mod",
     translation_key="alarm_sound_mode",
     entity_category=EntityCategory.CONFIG,
     options=["soft", "intensive", "silent"],
     supported_switch=DeviceSwitchType.ALARM_TONE.value,
+    action_handler=AlarmSoundModeAction,
+)
+
+BATTERY_WORK_MODE_SELECT_TYPE = EzvizSelectEntityDescription(
+    key="battery_camera_work_mode",
+    translation_key="battery_camera_work_mode",
+    icon="mdi:battery-sync",
+    entity_category=EntityCategory.CONFIG,
+    options=[
+        "plugged_in",
+        "high_performance",
+        "power_save",
+        "super_power_save",
+        "custom",
+    ],
+    supported_switch=-1,
+    action_handler=BatteryWorkModeAction,
 )
 
 
@@ -45,46 +124,54 @@
         DATA_COORDINATOR
     ]
 
-    async_add_entities(
-        EzvizSelect(coordinator, camera)
+    entities_to_add: list[EzvizSelect] = []
+
+    entities_to_add.extend(
+        EzvizSelect(coordinator, camera, ALARM_SOUND_MODE_SELECT_TYPE)
         for camera in coordinator.data
         for switch in coordinator.data[camera]["switches"]
-        if switch == SELECT_TYPE.supported_switch
+        if switch == ALARM_SOUND_MODE_SELECT_TYPE.supported_switch
+    )
+
+    entities_to_add.extend(
+        EzvizSelect(coordinator, camera, BATTERY_WORK_MODE_SELECT_TYPE)
+        for camera in coordinator.data
+        if coordinator.data[camera]["device_category"]
+        == DeviceCatagories.BATTERY_CAMERA_DEVICE_CATEGORY.value
     )
 
+    async_add_entities(entities_to_add)
+
 
 class EzvizSelect(EzvizEntity, SelectEntity):
     """Representation of a EZVIZ select entity."""
 
+    entity_description: EzvizSelectEntityDescription
+
     def __init__(
         self,
         coordinator: EzvizDataUpdateCoordinator,
         serial: str,
+        description: EzvizSelectEntityDescription,
     ) -> None:
         """Initialize the sensor."""
         super().__init__(coordinator, serial)
-        self._attr_unique_id = f"{serial}_{SELECT_TYPE.key}"
-        self.entity_description = SELECT_TYPE
+        self._attr_unique_id = f"{serial}_{description.key}"
+        self.entity_description = description
 
     @property
     def current_option(self) -> str | None:
         """Return the selected entity option to represent the entity state."""
-        sound_mode_value = getattr(
-            SoundMode, self.data[self.entity_description.key]
-        ).value
-        if sound_mode_value in [0, 1, 2]:
-            return self.options[sound_mode_value]
-
-        return None
+        return self.entity_description.action_handler.current_option(self)
 
     def select_option(self, option: str) -> None:
         """Change the selected option."""
-        sound_mode_value = self.options.index(option)
-
         try:
-            self.coordinator.ezviz_client.alarm_sound(self._serial, sound_mode_value, 1)
+            return self.entity_description.action_handler.select_option(
+                self, self._serial, option
+            )
 
         except (HTTPError, PyEzvizError) as err:
             raise HomeAssistantError(
-                f"Cannot set Warning sound level for {self.entity_id}"
+                f"Cannot select option for {self.entity_description.key}"
             ) from err

homeassistant/components/ezviz/strings.json

@@ -68,6 +68,16 @@
           "intensive": "Intensive",
           "silent": "Silent"
         }
+      },
+      "battery_camera_work_mode": {
+        "name": "Battery work mode",
+        "state": {
+          "plugged_in": "Plugged in",
+          "high_performance": "High performance",
+          "power_save": "Power save",
+          "super_power_save": "Super power saving",
+          "custom": "Custom"
+        }
       }
     },
     "image": {