1
0
mirror of https://github.com/checktheroads/hyperglass synced 2024-05-11 05:55:08 +00:00

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

This commit is contained in:
checktheroads
2020-05-29 17:51:41 -07:00
parent 19e3cc0ab1
commit f6d971a20d

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