zha: catch the exception from bellows if a device isn't available. (#13314)

* catch the exception from bellows if a device isn't available.

* fix import of zigpy exception

* fix lint import

homeassistant/components/light/zha.py

@@ -5,7 +5,6 @@ For more details on this platform, please refer to the documentation
 at https://home-assistant.io/components/light.zha/
 """
 import logging
-
 from homeassistant.components import light, zha
 from homeassistant.const import STATE_UNKNOWN
 import homeassistant.util.color as color_util
@@ -112,14 +111,25 @@ class Light(zha.Entity, light.Light):
             self._state = 1
             self.async_schedule_update_ha_state()
             return
-
+        from zigpy.exceptions import DeliveryError
+        try:
             await self._endpoint.on_off.on()
+        except DeliveryError as ex:
+            _LOGGER.error("Unable to turn the light on: %s", ex)
+            return
+
         self._state = 1
         self.async_schedule_update_ha_state()
 
     async def async_turn_off(self, **kwargs):
         """Turn the entity off."""
+        from zigpy.exceptions import DeliveryError
+        try:
             await self._endpoint.on_off.off()
+        except DeliveryError as ex:
+            _LOGGER.error("Unable to turn the light off: %s", ex)
+            return
+
         self._state = 0
         self.async_schedule_update_ha_state()
 

homeassistant/components/switch/zha.py

@@ -57,12 +57,24 @@ class Switch(zha.Entity, SwitchDevice):
 
     async def async_turn_on(self, **kwargs):
         """Turn the entity on."""
+        from zigpy.exceptions import DeliveryError
+        try:
             await self._endpoint.on_off.on()
+        except DeliveryError as ex:
+            _LOGGER.error("Unable to turn the switch on: %s", ex)
+            return
+
         self._state = 1
 
     async def async_turn_off(self, **kwargs):
         """Turn the entity off."""
+        from zigpy.exceptions import DeliveryError
+        try:
             await self._endpoint.on_off.off()
+        except DeliveryError as ex:
+            _LOGGER.error("Unable to turn the switch off: %s", ex)
+            return
+
         self._state = 0
 
     async def async_update(self):