From 1ab47b5d2b22cfbf53675e84f4f2df3a36c3b36c Mon Sep 17 00:00:00 2001 From: Fabian Affolter Date: Thu, 16 Mar 2017 19:59:34 +0100 Subject: [PATCH] Check if droplet exists (#6663) * Check if droplet exists * Add droplet to message and remove else --- homeassistant/components/binary_sensor/digital_ocean.py | 3 +++ homeassistant/components/switch/digital_ocean.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/homeassistant/components/binary_sensor/digital_ocean.py b/homeassistant/components/binary_sensor/digital_ocean.py index 31f29597f06..ea02196f3eb 100644 --- a/homeassistant/components/binary_sensor/digital_ocean.py +++ b/homeassistant/components/binary_sensor/digital_ocean.py @@ -36,6 +36,9 @@ def setup_platform(hass, config, add_devices, discovery_info=None): dev = [] for droplet in droplets: droplet_id = digital_ocean.DIGITAL_OCEAN.get_droplet_id(droplet) + if droplet_id is None: + _LOGGER.error("Droplet %s is not available", droplet) + return False dev.append(DigitalOceanBinarySensor( digital_ocean.DIGITAL_OCEAN, droplet_id)) diff --git a/homeassistant/components/switch/digital_ocean.py b/homeassistant/components/switch/digital_ocean.py index 11414ce96c5..066a356ba7b 100644 --- a/homeassistant/components/switch/digital_ocean.py +++ b/homeassistant/components/switch/digital_ocean.py @@ -35,6 +35,9 @@ def setup_platform(hass, config, add_devices, discovery_info=None): dev = [] for droplet in droplets: droplet_id = digital_ocean.DIGITAL_OCEAN.get_droplet_id(droplet) + if droplet_id is None: + _LOGGER.error("Droplet %s is not available", droplet) + return False dev.append(DigitalOceanSwitch( digital_ocean.DIGITAL_OCEAN, droplet_id))