Merge pull request #133 from zweckj/copilot/add-tank-status-to-dashboard

Add tank status to get_dashboard_from_bluetooth

Author: zweckj

pylamarzocco/devices/_machine.py

@@ -26,6 +26,7 @@
     CoffeeBoiler,
     LastCoffeeList,
     MachineStatus,
+    NoWater,
     PrebrewSettingTimes,
     SecondsInOut,
     SteamBoilerLevel,
@@ -196,6 +197,24 @@ async def get_dashboard_from_bluetooth(self) -> None:
                     # Remove level widget if it exists (not applicable for this model)
                     self.dashboard.config.pop(WidgetType.CM_STEAM_BOILER_LEVEL, None)
 
+        # Get tank status and update dashboard
+        try:
+            tank_status = await self._bluetooth_client.get_tank_status()
+        except (BleakError, BluetoothConnectionFailed) as exc:
+            _LOGGER.error("Failed to get tank status from Bluetooth: %s", exc)
+            raise
+
+        # Initialize or update no water widget
+        no_water = cast(
+            NoWater,
+            self.dashboard.config.get(
+                WidgetType.CM_NO_WATER,
+                NoWater(allarm=not tank_status),
+            ),
+        )
+        no_water.allarm = not tank_status
+        self.dashboard.config[WidgetType.CM_NO_WATER] = no_water
+
     async def set_power(self, enabled: bool) -> bool:
         """Set the power of the machine.
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pylamarzocco"
-version = "2.2.0"
+version = "2.2.1"
 license = { text = "MIT" }
 description = "A Python implementation of the La Marzocco API"
 readme = "README.md"

tests/__snapshots__/test_bluetooth_dashboard.ambr

@@ -25,6 +25,9 @@
         'next_status': None,
         'status': 'StandBy',
       }),
+      'CMNoWater': dict({
+        'allarm': False,
+      }),
       'CMSteamBoilerLevel': dict({
         'enabled': False,
         'enabled_supported': True,
@@ -75,6 +78,9 @@
         'next_status': None,
         'status': 'StandBy',
       }),
+      'CMNoWater': dict({
+        'allarm': False,
+      }),
       'CMSteamBoilerLevel': dict({
         'enabled': True,
         'enabled_supported': True,

tests/test_bluetooth_dashboard.py

@@ -25,6 +25,7 @@
     BluetoothMachineCapabilities,
     CoffeeBoiler,
     MachineStatus,
+    NoWater,
     SteamBoilerLevel,
     SteamBoilerTemperature,
 )
@@ -131,13 +132,15 @@ async def test_get_dashboard_from_bluetooth(
             ),
         ]
     )
+    mock_bluetooth_client.get_tank_status = AsyncMock(return_value=True)
 
     await mock_machine_with_dashboard.get_dashboard_from_bluetooth()
 
     # Verify calls - capabilities should NOT be called automatically anymore
     mock_bluetooth_client.get_machine_capabilities.assert_not_called()
     mock_bluetooth_client.get_machine_mode.assert_called_once()
     mock_bluetooth_client.get_boilers.assert_called_once()
+    mock_bluetooth_client.get_tank_status.assert_called_once()
 
     # Snapshot test includes model_name, model_code, and config
     assert mock_machine_with_dashboard.dashboard.to_dict() == snapshot
@@ -169,6 +172,13 @@ async def test_get_dashboard_from_bluetooth(
         not in mock_machine_with_dashboard.dashboard.config
     )
 
+    # Verify tank status widget
+    no_water = cast(
+        NoWater,
+        mock_machine_with_dashboard.dashboard.config[WidgetType.CM_NO_WATER],
+    )
+    assert no_water.allarm is False  # Tank status is True, so allarm should be False
+
 
 async def test_get_dashboard_no_bluetooth(
     mock_machine_with_dashboard: LaMarzoccoMachine,
@@ -251,6 +261,7 @@ async def test_get_dashboard_initializes_missing_widgets(
             ),
         ]
     )
+    mock_bluetooth_client.get_tank_status = AsyncMock(return_value=True)
 
     # First fetch model info explicitly
     await mock_machine_with_dashboard.get_model_info_from_bluetooth()
@@ -317,6 +328,7 @@ async def test_get_dashboard_without_steam_level_support(
             ),
         ]
     )
+    mock_bluetooth_client.get_tank_status = AsyncMock(return_value=True)
 
     # First fetch model info explicitly
     await mock_machine_with_dashboard.get_model_info_from_bluetooth()
@@ -391,6 +403,7 @@ async def test_get_dashboard_mini_original_temperature_only(
             ),
         ]
     )
+    mock_bluetooth_client.get_tank_status = AsyncMock(return_value=True)
 
     # First fetch model info explicitly
     await mock_machine_with_dashboard.get_model_info_from_bluetooth()