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

cleanup

Browse Source
This commit is contained in:
checktheroads
2020-04-12 02:17:16 -07:00
parent bb8cc55dc7
commit c02b8d359e
6 changed files with 37 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
hyperglass/configuration/models/routers.py
View File

@@ -11,7 +11,7 @@ from pydantic import StrictInt, StrictStr, validator
# Project
from hyperglass.util import log, clean_name
from hyperglass.constants import Supported
from hyperglass.constants import SCRAPE_HELPERS, TRANSPORT_REST, TRANSPORT_SCRAPE
from hyperglass.exceptions import ConfigError, UnsupportedDevice
from hyperglass.configuration.models.ssl import Ssl
from hyperglass.configuration.models.vrfs import Vrf, Info
@@ -65,8 +65,12 @@ class Router(HyperglassModel):
Returns:
{str} -- Valid NOS
"""
if not Supported.is_supported(value):
raise UnsupportedDevice(f'"{value}" device type is not supported.')
if value in SCRAPE_HELPERS.keys():
value = SCRAPE_HELPERS[value]
if value not in (*TRANSPORT_REST, *TRANSPORT_SCRAPE):
raise UnsupportedDevice('NOS "{n}" is not supported.', n=value)
return value
@validator("name")