Fix issue #94 (#118)

iMicknl · Mar 23, 2024 · 5465e77 · 5465e77
1 parent 62ce373
commit 5465e77
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/sagemcom_fast/button.py b/custom_components/sagemcom_fast/button.py
@@ -11,7 +11,7 @@
 from sagemcom_api.models import DeviceInfo as GatewayDeviceInfo
 
 from . import HomeAssistantSagemcomFastData
-from .const import DOMAIN
+from .const import DOMAIN, LOGGER
 
 
 async def async_setup_entry(
@@ -43,7 +43,11 @@ def __init__(self, gateway: GatewayDeviceInfo, client: SagemcomClient) -> None:
 
     async def async_press(self) -> None:
         """Handle the button press."""
-        await self.client.reboot()
+        try:
+            await self.client.login()
+            await self.client.reboot()
+        except Exception as exception:  # pylint: disable=broad-except
+            LOGGER.exception(exception)
 
     @property
     def device_info(self) -> DeviceInfo: