Fix data access in streamlabs water (home-assistant#107062)

* Fix data access in streamlabs water * Fix data access in streamlabs water
cereal2nd · Jan 4, 2024 · dd6c5e8 · dd6c5e8
1 parent 85a1d8b
commit dd6c5e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/homeassistant/components/streamlabswater/coordinator.py b/homeassistant/components/streamlabswater/coordinator.py
@@ -44,7 +44,7 @@ async def _async_update_data(self) -> dict[str, StreamlabsData]:
     def _update_data(self) -> dict[str, StreamlabsData]:
         locations = self.client.get_locations()
         res = {}
-        for location in locations:
+        for location in locations["locations"]:
             location_id = location["locationId"]
             water_usage = self.client.get_water_usage_summary(location_id)
             res[location_id] = StreamlabsData(

diff --git a/homeassistant/components/streamlabswater/sensor.py b/homeassistant/components/streamlabswater/sensor.py
@@ -27,7 +27,7 @@ async def async_setup_entry(
 
     entities = []
 
-    for location_id in coordinator.data.values():
+    for location_id in coordinator.data:
         entities.extend(
             [
                 StreamLabsDailyUsage(coordinator, location_id),