diff --git a/nornir_nautobot/plugins/tasks/dispatcher/__init__.py b/nornir_nautobot/plugins/tasks/dispatcher/__init__.py index b52f206..fd1799c 100644 --- a/nornir_nautobot/plugins/tasks/dispatcher/__init__.py +++ b/nornir_nautobot/plugins/tasks/dispatcher/__init__.py @@ -1,4 +1,5 @@ """Used to intialize the dispatcher.""" + # pylint: disable=raise-missing-from import importlib @@ -23,6 +24,7 @@ "mikrotik_routeros": "nornir_nautobot.plugins.tasks.dispatcher.mikrotik_routeros.NautobotNornirDriver", "ruckus_smartzone_api": "nornir_nautobot.plugins.tasks.dispatcher.ruckus_smartzone_api.NautobotNornirDriver", "ruckus_access_point": "nornir_nautobot.plugins.tasks.dispatcher.ruckus_smartzone_api.NautobotNornirDriver", + "aruba_os": "nornir_nautobot.plugins.tasks.dispatcher.aruba_os.NautobotNornirDriver", } diff --git a/nornir_nautobot/plugins/tasks/dispatcher/aruba_os.py b/nornir_nautobot/plugins/tasks/dispatcher/aruba_os.py index dcc8c32..de1e6e9 100644 --- a/nornir_nautobot/plugins/tasks/dispatcher/aruba_os.py +++ b/nornir_nautobot/plugins/tasks/dispatcher/aruba_os.py @@ -1,13 +1,9 @@ """nornir dispatcher for Aruba OS.""" -from nornir_nautobot.plugins.tasks.dispatcher.default import NapalmDefault, NetmikoDefault +from .default import NetmikoNautobotNornirDriver as DefaultNautobotNornirDriver -class NapalmArubaOs(NapalmDefault): - """Collection of Napalm Nornir Tasks specific to Aruba OS devices.""" - - -class NetmikoArubaOs(NetmikoDefault): - """Collection of Netmiko Nornir Tasks specific to Aruba OS devices.""" +class NautobotNornirDriver(DefaultNautobotNornirDriver): + """Driver for Aruba OS.""" config_command = "show run"