Handle HTTPError on nuki integration (#80801)

fix(nuki): handle requests errors

homeassistant/components/nuki/config_flow.py

@@ -8,13 +8,13 @@ from pynuki.bridge import InvalidCredentialsException
 from requests.exceptions import RequestException
 import voluptuous as vol
 
-from homeassistant import config_entries, exceptions
+from homeassistant import config_entries
 from homeassistant.components import dhcp
 from homeassistant.const import CONF_HOST, CONF_PORT, CONF_TOKEN
 from homeassistant.data_entry_flow import FlowResult
 
 from .const import DEFAULT_PORT, DEFAULT_TIMEOUT, DOMAIN
-from .helpers import parse_id
+from .helpers import CannotConnect, InvalidAuth, parse_id
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -153,11 +153,3 @@ class NukiConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
         return self.async_show_form(
             step_id="user", data_schema=data_schema, errors=errors
         )
-
-
-class CannotConnect(exceptions.HomeAssistantError):
-    """Error to indicate we cannot connect."""
-
-
-class InvalidAuth(exceptions.HomeAssistantError):
-    """Error to indicate there is invalid auth."""

homeassistant/components/nuki/helpers.py

@@ -1,6 +1,15 @@
 """nuki integration helpers."""
+from homeassistant import exceptions
 
 
 def parse_id(hardware_id):
     """Parse Nuki ID."""
     return hex(hardware_id).split("x")[-1].upper()
+
+
+class CannotConnect(exceptions.HomeAssistantError):
+    """Error to indicate we cannot connect."""
+
+
+class InvalidAuth(exceptions.HomeAssistantError):
+    """Error to indicate there is invalid auth."""

homeassistant/components/nuki/lock.py

@@ -6,6 +6,7 @@ from typing import Any
 
 from pynuki import NukiLock, NukiOpener
 from pynuki.constants import MODE_OPENER_CONTINUOUS
+from requests.exceptions import RequestException
 import voluptuous as vol
 
 from homeassistant.components.lock import LockEntity, LockEntityFeature
@@ -26,6 +27,7 @@ from .const import (
     DOMAIN as NUKI_DOMAIN,
     ERROR_STATES,
 )
+from .helpers import CannotConnect
 
 
 async def async_setup_entry(
@@ -114,15 +116,24 @@ class NukiLockEntity(NukiDeviceEntity):
 
     def lock(self, **kwargs: Any) -> None:
         """Lock the device."""
+        try:
             self._nuki_device.lock()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def unlock(self, **kwargs: Any) -> None:
         """Unlock the device."""
+        try:
             self._nuki_device.unlock()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def open(self, **kwargs: Any) -> None:
         """Open the door latch."""
+        try:
             self._nuki_device.unlatch()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def lock_n_go(self, unlatch: bool) -> None:
         """Lock and go.
@@ -130,7 +141,10 @@ class NukiLockEntity(NukiDeviceEntity):
         This will first unlock the door, then wait for 20 seconds (or another
         amount of time depending on the lock settings) and relock.
         """
+        try:
             self._nuki_device.lock_n_go(unlatch)
+        except RequestException as err:
+            raise CannotConnect from err
 
 
 class NukiOpenerEntity(NukiDeviceEntity):
@@ -148,15 +162,24 @@ class NukiOpenerEntity(NukiDeviceEntity):
 
     def lock(self, **kwargs: Any) -> None:
         """Disable ring-to-open."""
+        try:
             self._nuki_device.deactivate_rto()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def unlock(self, **kwargs: Any) -> None:
         """Enable ring-to-open."""
+        try:
             self._nuki_device.activate_rto()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def open(self, **kwargs: Any) -> None:
         """Buzz open the door."""
+        try:
             self._nuki_device.electric_strike_actuation()
+        except RequestException as err:
+            raise CannotConnect from err
 
     def lock_n_go(self, unlatch: bool) -> None:
         """Stub service."""
@@ -168,7 +191,10 @@ class NukiOpenerEntity(NukiDeviceEntity):
         rings the bell. This is similar to ring-to-open, except that it does
         not automatically deactivate
         """
+        try:
             if enable:
                 self._nuki_device.activate_continuous_mode()
             else:
                 self._nuki_device.deactivate_continuous_mode()
+        except RequestException as err:
+            raise CannotConnect from err