implement some features of Venus OS

custom_components/victron_mk2_mqtt/switch.py

@@ -20,7 +20,10 @@ async def async_setup_platform(
 ) -> None:
     """Set up Victron switch entities."""
     bridge: VictronMqttBridge = hass.data[DOMAIN][DATA_BRIDGE]
-    async_add_entities([VictronRemotePanelStandbySwitch(bridge)])
+    entities: list[SwitchEntity] = [VictronRemotePanelStandbySwitch(bridge)]
+    if bridge.venus_guide_compat:
+        entities.append(VictronESSOptimizedModeSwitch(bridge))
+    async_add_entities(entities)
 
 
 class VictronRemotePanelStandbySwitch(VictronMqttEntity, SwitchEntity):
@@ -46,3 +49,28 @@ class VictronRemotePanelStandbySwitch(VictronMqttEntity, SwitchEntity):
 
     async def async_turn_off(self, **kwargs: Any) -> None:
         await self.bridge.async_publish_command({"kind": "standby", "standby": False})
+
+
+class VictronESSOptimizedModeSwitch(VictronMqttEntity, SwitchEntity):
+    """Guide-compatible ESS optimized mode switch."""
+
+    _attr_name = "ESS Optimized Mode"
+    _attr_icon = "mdi:battery-sync"
+
+    def __init__(self, bridge: VictronMqttBridge) -> None:
+        super().__init__(bridge)
+        self._attr_unique_id = f"{bridge.topic_root}_ess_optimized_mode"
+
+    @property
+    def is_on(self) -> bool:
+        return self.bridge.ess_mode == 10
+
+    @property
+    def available(self) -> bool:
+        return bool(self.bridge.command_topic and self.bridge.venus_guide_compat)
+
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        await self.bridge.async_publish_command({"kind": "ess_mode", "value": 10})
+
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        await self.bridge.async_publish_command({"kind": "ess_mode", "value": 9})