142 files

custom_components/maintenance_supporter/websocket/__init__.py

@@ -533,6 +533,7 @@ def async_register_commands(hass: HomeAssistant) -> None:
         ws_battery_fleet_mark_replaced,
         ws_battery_fleet_overview,
         ws_battery_fleet_set_excluded,
+        ws_battery_fleet_set_included,
         ws_battery_fleet_setup,
         ws_battery_fleet_status,
     )
@@ -687,6 +688,7 @@ def async_register_commands(hass: HomeAssistant) -> None:
     websocket_api.async_register_command(hass, ws_battery_fleet_setup)
     websocket_api.async_register_command(hass, ws_battery_fleet_mark_replaced)
     websocket_api.async_register_command(hass, ws_battery_fleet_set_excluded)
+    websocket_api.async_register_command(hass, ws_battery_fleet_set_included)
     websocket_api.async_register_command(hass, ws_discover_integration_setups)
     websocket_api.async_register_command(hass, ws_adopt_integration_setups)
     websocket_api.async_register_command(hass, ws_list_saved_views)

custom_components/maintenance_supporter/websocket/battery_fleet.py

@@ -154,6 +154,27 @@ async def ws_battery_fleet_set_excluded(
     connection.send_result(msg["id"], {"success": True})
 
 
+@websocket_api.websocket_command(
+    {
+        vol.Required("type"): f"{DOMAIN}/battery_fleet/set_included",
+        vol.Required("entity_id"): vol.All(str, vol.Length(max=MAX_ENTITY_ID_LENGTH)),
+        vol.Required("included"): bool,
+    }
+)
+@require_write
+@websocket_api.async_response
+async def ws_battery_fleet_set_included(
+    hass: HomeAssistant, connection: websocket_api.ActiveConnection, msg: dict[str, Any]
+) -> None:
+    """Manually ADD a battery the discovery heuristics miss (#135)."""
+    from ..helpers.battery_fleet_setup import set_battery_included
+
+    if not set_battery_included(hass, msg["entity_id"], msg["included"]):
+        connection.send_error(msg["id"], "not_configured", "Battery Fleet is not set up")
+        return
+    connection.send_result(msg["id"], {"success": True})
+
+
 @websocket_api.websocket_command(
     {
         vol.Required("type"): f"{DOMAIN}/battery_fleet/mark_replaced",