refactor(asynceapi): Use http head request to check_connection

aristanetworks · Oct 4, 2024 · 049202b · 049202b
1 parent fc644a1
commit 049202b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 65 deletions.
diff --git a/asynceapi/aio_portcheck.py b/asynceapi/aio_portcheck.py
diff --git a/asynceapi/device.py b/asynceapi/device.py
@@ -20,7 +20,6 @@
 # -----------------------------------------------------------------------------
 # Private Imports
 # -----------------------------------------------------------------------------
-from .aio_portcheck import port_check_url
 from .config_session import SessionConfig
 from .errors import EapiCommandError
 
@@ -122,7 +121,11 @@ async def check_connection(self) -> bool:
         bool
             True when the device eAPI is accessible, False otherwise.
         """
-        return await port_check_url(self.base_url)
+        try:
+            await self.head(self.base_url, timeout=5)
+        except httpx.RequestError:
+            return False
+        return True
 
     async def cli(  # noqa: PLR0913
         self,