Allow customization of driver (Netmiko) configuration on a per-device…

… basis
thatmattlove · Dec 8, 2021 · b49b6ea · b49b6ea
1 parent 6dea685
commit b49b6ea
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/hyperglass/constants.py b/hyperglass/constants.py
@@ -76,8 +76,6 @@
 }
 
 DRIVER_MAP = {
-    "frr_legacy": "hyperglass_agent",
-    "bird_legacy": "hyperglass_agent",
     "bird": "netmiko",
     "frr": "netmiko",
     "http": "hyperglass_http_client",

diff --git a/hyperglass/execution/drivers/ssh_netmiko.py b/hyperglass/execution/drivers/ssh_netmiko.py
@@ -70,6 +70,7 @@ async def collect(self, host: str = None, port: int = None) -> Iterable:
             "timeout": math.floor(params.request_timeout * 1.25),
             "session_timeout": math.ceil(params.request_timeout - 1),
             **global_args,
+            **self.device.driver_config,
         }
 
         if "_telnet" in self.device.platform:

diff --git a/hyperglass/models/config/devices.py b/hyperglass/models/config/devices.py
@@ -54,6 +54,7 @@ class Device(HyperglassModelWithId, extra="allow"):
     structured_output: Optional[StrictBool]
     directives: Directives = Directives()
     driver: Optional[SupportedDriver]
+    driver_config: Dict[str, Any] = {}
     attrs: Dict[str, str] = {}
 
     def __init__(self, **kw) -> None: