added exception handling on httpx.ReadTimeout (#46)

jeremyschulman · Sep 26, 2023 · 26c4ef3 · 26c4ef3
1 parent 65f0f48
commit 26c4ef3
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/netcam_aioeos/bgp_peering/eos_checks_bgp_peering.py b/netcam_aioeos/bgp_peering/eos_checks_bgp_peering.py
@@ -33,7 +33,9 @@ async def check_neeighbors(
     results: CheckResultsCollection = list()
     checks = check_collection.checks
 
-    dev_data = await dut.api_cache_get(key="bgp-summary", command="show ip bgp summary vrf all")
+    dev_data = await dut.api_cache_get(
+        key="bgp-summary", command="show ip bgp summary vrf all"
+    )
 
     for nei_check in checks:
         _check_bgp_neighbor(

diff --git a/netcam_aioeos/bgp_peering/eos_checks_bgp_routers.py b/netcam_aioeos/bgp_peering/eos_checks_bgp_routers.py
@@ -48,7 +48,9 @@ async def check_bgp_neighbors(
     checks = check_bgp_routers.checks
     dut: EOSDeviceUnderTest = self
 
-    dev_data = await dut.api_cache_get(key="bgp-summary", command="show ip bgp summary vrf all")
+    dev_data = await dut.api_cache_get(
+        key="bgp-summary", command="show ip bgp summary vrf all"
+    )
 
     for rtr_chk in checks:
         _check_router_vrf(dut=dut, check=rtr_chk, dev_data=dev_data, results=results)

diff --git a/netcam_aioeos/config/eos_dcfg.py b/netcam_aioeos/config/eos_dcfg.py
@@ -18,6 +18,8 @@
 
 from typing import Optional
 
+import httpx
+
 # -----------------------------------------------------------------------------
 # Public Imports
 # -----------------------------------------------------------------------------
@@ -122,6 +124,8 @@ async def config_check(self, replace: Optional[bool | None] = None) -> str | Non
             )
         except RuntimeError as exc:
             errors = exc.args[0]
+        except httpx.ReadTimeout:
+            errors = f"Read timeout after {self.eapi.timeout}. Increase config.timeout"
         else:
             errors = None
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "netcam-aioeos"
-version = "0.6.6"
+version = "0.6.7"
 description = "NetCAD/CAM - Arista EOS/eAPI asyncio driver"
 readme = "README.md"
 authors = ["Jeremy Schulman"]