From dee3e14df20bb67aa629501ee527e7be34523546 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Tue, 29 Jun 2021 03:21:33 -0500 Subject: [PATCH] Fix values of RainMachine Freeze Protection and Hot Days binary sensors (#52275) * Fix values of RainMachine Freeze Protection and Hot Days binary sensors * Correct place * Fix --- .../components/rainmachine/binary_sensor.py | 20 ++++++------------- 1 file changed, 6 insertions(+), 14 deletions(-) diff --git a/homeassistant/components/rainmachine/binary_sensor.py b/homeassistant/components/rainmachine/binary_sensor.py index 4b89b52befe..171fd26b910 100644 --- a/homeassistant/components/rainmachine/binary_sensor.py +++ b/homeassistant/components/rainmachine/binary_sensor.py @@ -179,18 +179,8 @@ class ProvisionSettingsBinarySensor(RainMachineBinarySensor): @callback def update_from_latest_data(self) -> None: """Update the state.""" - if self._sensor_type == TYPE_FREEZE: - self._state = self.coordinator.data["freeze"] - elif self._sensor_type == TYPE_HOURLY: - self._state = self.coordinator.data["hourly"] - elif self._sensor_type == TYPE_MONTH: - self._state = self.coordinator.data["month"] - elif self._sensor_type == TYPE_RAINDELAY: - self._state = self.coordinator.data["rainDelay"] - elif self._sensor_type == TYPE_RAINSENSOR: - self._state = self.coordinator.data["rainSensor"] - elif self._sensor_type == TYPE_WEEKDAY: - self._state = self.coordinator.data["weekDay"] + if self._sensor_type == TYPE_FLOW_SENSOR: + self._state = self.coordinator.data["system"].get("useFlowSensor") class UniversalRestrictionsBinarySensor(RainMachineBinarySensor): @@ -199,5 +189,7 @@ class UniversalRestrictionsBinarySensor(RainMachineBinarySensor): @callback def update_from_latest_data(self) -> None: """Update the state.""" - if self._sensor_type == TYPE_FLOW_SENSOR: - self._state = self.coordinator.data["system"].get("useFlowSensor") + if self._sensor_type == TYPE_FREEZE_PROTECTION: + self._state = self.coordinator.data["freezeProtectEnabled"] + elif self._sensor_type == TYPE_HOT_DAYS: + self._state = self.coordinator.data["hotDaysExtraWatering"]