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

validate netmiko supported NOS by importing from netmiko directly, closes #41

Browse source
This commit is contained in:
checktheroads 2020-05-29 17:51:41 -07:00
parent 19e3cc0ab1
commit f6d971a20d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
hyperglass/configuration/models/routers.py
View file

@ -8,6 +8,7 @@ from pathlib import Path
# Third Party
from pydantic import StrictInt, StrictStr, StrictBool, validator
from netmiko.ssh_dispatcher import CLASS_MAPPER_BASE as NETMIKO_SUPPORTED
# Project
from hyperglass.log import log
@ -16,7 +17,6 @@ from hyperglass.models import HyperglassModel, HyperglassModelExtra
from hyperglass.constants import (
SCRAPE_HELPERS,
TRANSPORT_REST,
TRANSPORT_SCRAPE,
SUPPORTED_STRUCTURED_OUTPUT,
)
from hyperglass.exceptions import ConfigError, UnsupportedDevice
@ -100,7 +100,7 @@ class Router(HyperglassModel):
if value in SCRAPE_HELPERS.keys():
value = SCRAPE_HELPERS[value]
if value not in (*TRANSPORT_REST, *TRANSPORT_SCRAPE):
if value not in (*TRANSPORT_REST, *NETMIKO_SUPPORTED.keys()):
raise UnsupportedDevice('NOS "{n}" is not supported.', n=value)
return value