Merge pull request #293 from RobBie1221/improve_init

Add additional try except block around init
ksya · Jan 12, 2024 · ad8aae6 · ad8aae6
2 parents 444776d + f503ed9
commit ad8aae6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/custom_components/nefiteasy/__init__.py b/custom_components/nefiteasy/__init__.py
@@ -13,6 +13,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+import slixmpp
 
 from .const import (
     CONF_ACCESSKEY,
@@ -145,7 +146,11 @@ async def connect(self) -> None:
                 self.connected_state = STATE_CONNECTED
 
             if self.connected_state == STATE_CONNECTED:
-                self.nefit.get("/gateway/brandID")
+                try:
+                    self.nefit.get("/gateway/brandID")
+                except slixmpp.xmlstream.xmlstream.NotConnectedError:
+                    self.connected_state == STATE_INIT
+                    return
                 try:
                     await asyncio.wait_for(
                         self.nefit.xmppclient.message_event.wait(), timeout=29.0