Catch connection timeout

homeassistant/components/switch/mystrom.py

@@ -10,7 +10,6 @@ import logging
 import requests
 
 from homeassistant.components.switch import SwitchDevice
-from homeassistant.const import STATE_UNKNOWN
 
 DEFAULT_NAME = 'myStrom Switch'
 
@@ -45,7 +44,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
 class MyStromSwitch(SwitchDevice):
     """ Represents a myStrom switch. """
     def __init__(self, name, resource):
-        self._state = STATE_UNKNOWN
+        self._state = False
         self._name = name
         self._resource = resource
         self.consumption = 0
@@ -67,24 +66,32 @@ class MyStromSwitch(SwitchDevice):
 
     def turn_on(self, **kwargs):
         """ Turn the switch on. """
+        try:
             request = requests.get('{}/relay'.format(self._resource),
                                    params={'state': '1'},
                                    timeout=10)
             if request.status_code == 200:
                 self._state = True
             else:
+                raise requests.exceptions.ConnectionError
+
+        except requests.exceptions.ConnectionError:
             _LOGGER.error("Can't turn on %s. Is device offline?",
                           self._resource)
 
     def turn_off(self, **kwargs):
         """ Turn the switch off. """
+        try:
             request = requests.get('{}/relay'.format(self._resource),
                                    params={'state': '0'},
                                    timeout=10)
             if request.status_code == 200:
                 self._state = False
             else:
-            _LOGGER.error("Can't turn off %s. Is device offline?",
+                raise requests.exceptions.ConnectionError
+
+        except requests.exceptions.ConnectionError:
+            _LOGGER.error("Can't turn on %s. Is device offline?",
                           self._resource)
 
     def update(self):