mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
"""Validate command configuration variables."""
|
|
|
|
# Local
|
|
from .frr import FRRCommands
|
|
from .bird import BIRDCommands
|
|
from .tnsr import TNSRCommands
|
|
from .vyos import VyosCommands
|
|
from ..main import HyperglassModel
|
|
from .common import CommandGroup
|
|
from .huawei import HuaweiCommands
|
|
from .juniper import JuniperCommands
|
|
from .cisco_xr import CiscoXRCommands
|
|
from .cisco_ios import CiscoIOSCommands
|
|
from .arista_eos import AristaEOSCommands
|
|
from .cisco_nxos import CiscoNXOSCommands
|
|
from .nokia_sros import NokiaSROSCommands
|
|
from .mikrotik_routeros import MikrotikRouterOS
|
|
from .mikrotik_switchos import MikrotikSwitchOS
|
|
|
|
_DEVICE_TYPE_MAP = {
|
|
"arista_eos": AristaEOSCommands,
|
|
"bird": BIRDCommands,
|
|
"cisco_ios": CiscoIOSCommands,
|
|
"cisco_nxos": CiscoNXOSCommands,
|
|
"cisco_xr": CiscoXRCommands,
|
|
"frr": FRRCommands,
|
|
"huawei": HuaweiCommands,
|
|
"juniper": JuniperCommands,
|
|
"mikrotik_routeros": MikrotikRouterOS,
|
|
"mikrotik_switchos": MikrotikSwitchOS,
|
|
"nokia_sros": NokiaSROSCommands,
|
|
"tnsr": TNSRCommands,
|
|
"vyos": VyosCommands,
|
|
}
|
|
|
|
|
|
class Commands(HyperglassModel, extra="allow", validate_all=False):
|
|
"""Base class for command definitions."""
|
|
|
|
arista_eos: CommandGroup = AristaEOSCommands()
|
|
bird: CommandGroup = BIRDCommands()
|
|
cisco_ios: CommandGroup = CiscoIOSCommands()
|
|
cisco_nxos: CommandGroup = CiscoNXOSCommands()
|
|
cisco_xr: CommandGroup = CiscoXRCommands()
|
|
frr: CommandGroup = FRRCommands()
|
|
huawei: CommandGroup = HuaweiCommands()
|
|
juniper: CommandGroup = JuniperCommands()
|
|
mikrotik_routeros: CommandGroup = MikrotikRouterOS()
|
|
mikrotik_switchos: CommandGroup = MikrotikSwitchOS()
|
|
nokia_sros: CommandGroup = NokiaSROSCommands()
|
|
tnsr: CommandGroup = TNSRCommands()
|
|
vyos: CommandGroup = VyosCommands()
|
|
|
|
@classmethod
|
|
def import_params(cls, **input_params):
|
|
"""Import loaded YAML, initialize per-command definitions."""
|
|
obj = Commands()
|
|
for platform, cmds in input_params.items():
|
|
cmd_set = _DEVICE_TYPE_MAP.get(platform, CommandGroup)
|
|
cmds = cmd_set(**cmds)
|
|
setattr(obj, platform, cmds)
|
|
return obj
|