checktheroads-hyperglass/hyperglass/models/commands/__init__.py

"""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