Log out of Sagemcom client after every query (#30)

* Log out of Sagemcom client after every query

* Fix black

* Update __init__.py

* Revert change

Co-authored-by: Mick Vleeshouwer <mick@imick.nl>

custom_components/sagemcom_fast/__init__.py

@@ -101,6 +101,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
         _LOGGER.exception(exception)
         return False
 
+    try:
+        gateway = await client.get_device_info()
+    finally:
+        await client.logout()
+
     update_interval = entry.options.get(CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL)
 
     coordinator = SagemcomDataUpdateCoordinator(
@@ -119,7 +124,6 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     }
 
     # Create gateway device in Home Assistant
-    gateway = await client.get_device_info()
     device_registry = await hass.helpers.device_registry.async_get_registry()
 
     device_registry.async_get_or_create(

custom_components/sagemcom_fast/config_flow.py

@@ -67,6 +67,7 @@ async def async_validate_input(self, user_input):
         )
 
         await client.login()
+        await client.logout()
 
         return self.async_create_entry(
             title=host,

custom_components/sagemcom_fast/device_tracker.py

@@ -60,7 +60,11 @@ async def _async_update_data(self) -> Dict[str, Device]:
         """Update hosts data."""
         try:
             async with async_timeout.timeout(10):
-                hosts = await self._client.get_hosts(only_active=True)
+                try:
+                    await self._client.login()
+                    hosts = await self._client.get_hosts(only_active=True)
+                finally:
+                    await self._client.logout()
                 """Mark all device as non-active."""
                 for idx, host in self.hosts.items():
                     host.active = False