upgrade library

bj00rn · Feb 13, 2023 · 51eb3cb · 51eb3cb
1 parent d7a9f3c
commit 51eb3cb
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 12 deletions.
diff --git a/custom_components/saleryd_hrv/__init__.py b/custom_components/saleryd_hrv/__init__.py
@@ -37,13 +37,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     port = entry.data.get(CONF_WEBSOCKET_PORT)
 
     session = async_get_clientsession(hass)
-    client = Client(url, port, session)
-    await client.connect()
-    coordinator = SalerydLokeDataUpdateCoordinator(
-        hass,
-        update_method=client.async_get_data,
-        update_interval=SCAN_INTERVAL,
-    )
+    async with Client(url, port, session) as client:
+        coordinator = SalerydLokeDataUpdateCoordinator(
+            hass,
+            update_method=client.async_get_data,
+            update_interval=SCAN_INTERVAL,
+        )
 
     await coordinator.async_config_entry_first_refresh()
 

diff --git a/custom_components/saleryd_hrv/config_flow.py b/custom_components/saleryd_hrv/config_flow.py
@@ -85,10 +85,8 @@ async def _test_connection(self, ip, port):
         """Return true if connection is working"""
 
         try:
-            session = async_create_clientsession(self.hass)
-            client = Client(ip, port, session)
-            asyncio.wait_for(client.connect(), 10)
-            return True
+            async with Client(ip, port, async_create_clientsession(self.hass)):
+                return True
         except Exception as e:  # pylint: disable=broad-except
             _LOGGER.error("Could not connect", exc_info=True)
             pass

diff --git a/custom_components/saleryd_hrv/manifest.json b/custom_components/saleryd_hrv/manifest.json
@@ -10,6 +10,6 @@
     "@bj00rn"
   ],
   "requirements": [
-    "pysaleryd>=2.0.2"
+    "pysaleryd>=3.0.1"
   ]
 }