From 856db4d122f4881bb73b6457abd319cea0e053a4 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Tue, 7 Jan 2020 10:43:12 -0700 Subject: [PATCH] Fix incorrect Rainmachine dict lookups (#30550) --- homeassistant/components/rainmachine/binary_sensor.py | 4 +++- homeassistant/components/rainmachine/sensor.py | 10 +++++----- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/rainmachine/binary_sensor.py b/homeassistant/components/rainmachine/binary_sensor.py index 4e847b86f70..1fe98482211 100644 --- a/homeassistant/components/rainmachine/binary_sensor.py +++ b/homeassistant/components/rainmachine/binary_sensor.py @@ -96,7 +96,9 @@ class RainMachineBinarySensor(RainMachineEntity, BinarySensorDevice): async def async_update(self): """Update the state.""" if self._sensor_type == TYPE_FLOW_SENSOR: - self._state = self.rainmachine.data[PROVISION_SETTINGS].get("useFlowSensor") + self._state = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( + "useFlowSensor" + ) elif self._sensor_type == TYPE_FREEZE: self._state = self.rainmachine.data[RESTRICTIONS_CURRENT]["freeze"] elif self._sensor_type == TYPE_FREEZE_PROTECTION: diff --git a/homeassistant/components/rainmachine/sensor.py b/homeassistant/components/rainmachine/sensor.py index f0fb6cd1264..399e86b7db1 100644 --- a/homeassistant/components/rainmachine/sensor.py +++ b/homeassistant/components/rainmachine/sensor.py @@ -97,14 +97,14 @@ class RainMachineSensor(RainMachineEntity): async def async_update(self): """Update the sensor's state.""" if self._sensor_type == TYPE_FLOW_SENSOR_CLICK_M3: - self._state = self.rainmachine.data[PROVISION_SETTINGS].get( + self._state = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( "flowSensorClicksPerCubicMeter" ) elif self._sensor_type == TYPE_FLOW_SENSOR_CONSUMED_LITERS: - clicks = self.rainmachine.data[PROVISION_SETTINGS].get( + clicks = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( "flowSensorWateringClicks" ) - clicks_per_m3 = self.rainmachine.data[PROVISION_SETTINGS].get( + clicks_per_m3 = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( "flowSensorClicksPerCubicMeter" ) @@ -113,11 +113,11 @@ class RainMachineSensor(RainMachineEntity): else: self._state = None elif self._sensor_type == TYPE_FLOW_SENSOR_START_INDEX: - self._state = self.rainmachine.data[PROVISION_SETTINGS].get( + self._state = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( "flowSensorStartIndex" ) elif self._sensor_type == TYPE_FLOW_SENSOR_WATERING_CLICKS: - self._state = self.rainmachine.data[PROVISION_SETTINGS].get( + self._state = self.rainmachine.data[PROVISION_SETTINGS]["system"].get( "flowSensorWateringClicks" ) elif self._sensor_type == TYPE_FREEZE_TEMP: