Logout after testing credentials

pypolestar · Dec 18, 2024 · a498311 · a498311
1 parent 7acd9f8
commit a498311
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/custom_components/polestar_api/config_flow.py b/custom_components/polestar_api/config_flow.py
@@ -85,14 +85,18 @@ async def _test_credentials(
             password=password,
             client_session=get_async_client(self.hass),
         )
-        await api_client.async_init()
 
-        if found_vins := api_client.get_available_vins():
-            _LOGGER.debug("Found %d VINs for %s", len(found_vins), username)
-        else:
-            _LOGGER.warning("No VINs found for %s", username)
-            raise NoCarsFoundException
+        try:
+            await api_client.async_init()
 
-        if vin and vin not in found_vins:
-            _LOGGER.warning("VIN %s not found for %s", vin, username)
-            raise VinNotFoundException
+            if found_vins := api_client.get_available_vins():
+                _LOGGER.debug("Found %d VINs for %s", len(found_vins), username)
+            else:
+                _LOGGER.warning("No VINs found for %s", username)
+                raise NoCarsFoundException
+
+            if vin and vin not in found_vins:
+                _LOGGER.warning("VIN %s not found for %s", vin, username)
+                raise VinNotFoundException
+        finally:
+            await api_client.async_logout()