From 5501b7e710bed9ec32b2afc195a75f37eab1bf69 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Tue, 13 Sep 2022 16:16:21 -0600 Subject: [PATCH] Fix bug with RainMachine update entity (#78411) * Fix bug with RainMachine update entity * Comment --- homeassistant/components/rainmachine/update.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/rainmachine/update.py b/homeassistant/components/rainmachine/update.py index b191f2695a0..a811894a0c2 100644 --- a/homeassistant/components/rainmachine/update.py +++ b/homeassistant/components/rainmachine/update.py @@ -99,4 +99,11 @@ class RainMachineUpdateEntity(RainMachineEntity, UpdateEntity): UpdateStates.UPGRADING, UpdateStates.REBOOT, ) - self._attr_latest_version = data["packageDetails"]["newVersion"] + + # The RainMachine API docs say that multiple "packages" can be updated, but + # don't give details on what types exist (which makes it impossible to have + # update entities per update type); so, we use the first one (with the idea that + # after it succeeds, the entity will show the next update): + package_details = data["packageDetails"][0] + self._attr_latest_version = package_details["newVersion"] + self._attr_title = package_details["packageName"]