Add aruba os driver mapping

nautobot · Feb 7, 2024 · 41b9b13 · 41b9b13
1 parent 2afc0e5
commit 41b9b13
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git 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
@@ -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"