From 51c6029fe55aaccf07f6bf30df9be2ca14a352d0 Mon Sep 17 00:00:00 2001 From: "J.J.Barrancos" Date: Mon, 28 Aug 2017 21:07:40 +0200 Subject: [PATCH] Fixed issue with missing key --- homeassistant/components/switch/rainbird.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/switch/rainbird.py b/homeassistant/components/switch/rainbird.py index 5987f45bf22..f951f257560 100644 --- a/homeassistant/components/switch/rainbird.py +++ b/homeassistant/components/switch/rainbird.py @@ -38,8 +38,8 @@ def setup_platform(hass, config, add_devices, discovery_info=None): """Set up Rain Bird switches over a Rain Bird controller.""" devices = [] rbdevice = hass.data.get("DATA_RAINBIRD") - for key, switch in config.get(CONF_SWITCHES).items(): - devices.append(RainBirdSwitch(rbdevice, switch)) + for dev_id, switch in config.get(CONF_SWITCHES).items(): + devices.append(RainBirdSwitch(rbdevice, switch, dev_id)) add_devices(devices) return True @@ -47,9 +47,10 @@ def setup_platform(hass, config, add_devices, discovery_info=None): class RainBirdSwitch(SwitchDevice): """Representation of a Rain Bird switch.""" - def __init__(self, rb, dev): + def __init__(self, rb, dev, dev_id): """Initialize a Rain Bird Switch Device.""" self._rainbird = rb + self._devid = dev_id self._zone = int(dev.get(CONF_ZONE)) self._name = dev.get(CONF_FRIENDLY_NAME, "Sprinker %s" % self._zone) self._state = self.get_device_status()