witine/lookingglass
1
0
Fork 0
forked from mirrors/thatmattlove-hyperglass
Code Issues Pull requests Releases Activity

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

Browse source
This commit is contained in:
thatmattlove 2021-12-08 12:00:02 -07:00
parent 6dea685eeb
commit b49b6ea58e
3 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
hyperglass/constants.py
View file

@ -76,8 +76,6 @@ SCRAPE_HELPERS = {
} }
DRIVER_MAP = { DRIVER_MAP = {
"frr_legacy": "hyperglass_agent",
"bird_legacy": "hyperglass_agent",
"bird": "netmiko", "bird": "netmiko",
"frr": "netmiko", "frr": "netmiko",
"http": "hyperglass_http_client", "http": "hyperglass_http_client",

1
hyperglass/execution/drivers/ssh_netmiko.py
View file

@ -70,6 +70,7 @@ class NetmikoConnection(SSHConnection):
"timeout": math.floor(params.request_timeout * 1.25), "timeout": math.floor(params.request_timeout * 1.25),
"session_timeout": math.ceil(params.request_timeout - 1), "session_timeout": math.ceil(params.request_timeout - 1),
**global_args, **global_args,
**self.device.driver_config,
} }
if "_telnet" in self.device.platform: if "_telnet" in self.device.platform:

1
hyperglass/models/config/devices.py
View file

@ -54,6 +54,7 @@ class Device(HyperglassModelWithId, extra="allow"):
structured_output: Optional[StrictBool] structured_output: Optional[StrictBool]
directives: Directives = Directives() directives: Directives = Directives()
driver: Optional[SupportedDriver] driver: Optional[SupportedDriver]
driver_config: Dict[str, Any] = {}
attrs: Dict[str, str] = {} attrs: Dict[str, str] = {}
def __init__(self, **kw) -> None: def __init__(self, **kw) -> None: