Add cisco_ios support.

get_config.py

@@ -6,7 +6,9 @@
 import db
 
 
-def get_config(device_username, device_password, device_ip, device_type):
+def get_config(
+    device_username, device_password, device_ip, device_type, enable_password
+):
     """
     Get the running configuration of a device.
 
@@ -15,14 +17,19 @@ def get_config(device_username, device_password, device_ip, device_type):
         device_password: The password to authenticate with.
         device_ip: The IP address of the device.
         device_type: The device parameters.
+        enable_password: The enable password to authenticate with.
 
     Returns:
         The configuration of the device.
     """
 
     # Import module named device_type, pass device type to get_running_config, return to string
     config = __import__(f"vendors.{device_type}", fromlist=[""]).get_running_config(
-        device_username, device_password, device_ip, device_type
+        device_username,
+        device_password,
+        device_ip,
+        device_type,
+        enable_password,
     )
 
     return config
@@ -41,7 +48,11 @@ def fetch_all_configs():
 
     for device in db.list_all_ips_with_type():
         config = get_config(
-            device.username, device.password, device.ip, device.device_type
+            device.username,
+            device.password,
+            device.ip,
+            device.device_type,
+            device.enable_password,
         )
 
         print(f"Fetching config for {device.ip} ...")

get_hostname.py

@@ -20,7 +20,7 @@ def get_hostname(device_ip, device_type, username, password, enable_password):
 
     # Import module named device_type, pass device type to get_running_config, return to string
     hostname = __import__(f"vendors.{device_type}", fromlist=[""]).get_hostname(
-        my_secrets.USERNAME, my_secrets.PASSWORD, device_ip, device_type
+        username, password, device_ip, device_type, enable_password
     )
 
     # Return hostname

vendors/__init__.py

@@ -1 +1 @@
-__all__ = ["hp_procurve"]
+__all__ = ["hp_procurve", "cisco_ios"]

vendors/cisco_ios.py

@@ -0,0 +1,89 @@
+import netmiko
+import get_config
+import re
+
+
+def get_running_config(
+    device_username, device_password, device_ip, device_type, enable_password
+):
+    """
+    Get the command to get the running config.
+
+    Args:
+        device_username: The username to authenticate with.
+        device_password: The password to authenticate with.
+        device_ip: The IP address of the device.
+        device_type: The device parameters.
+        enable_password: The enable password to authenticate with.
+
+    Returns:
+        String: The running configuration of the device.
+        String: The error message if the device could not be connected to.
+    """
+
+    try:
+        device = {
+            "device_type": device_type,
+            "host": device_ip,
+            "username": device_username,
+            "password": device_password,
+            # wait a little longer for the device to respond
+            "fast_cli": False,
+            "global_delay_factor": 2,
+            "secret": enable_password,
+        }
+
+        with netmiko.ConnectHandler(**device) as connection:
+            # send enable command
+            output = connection.send_command("enable", expect_string="Password:")
+
+            if "Password:" in output:
+                connection.send_command_timing(device["secret"])
+
+            # Get the running configuration
+            config = connection.send_command("show running-config")
+
+        # Close the connection
+        connection.disconnect()
+
+        # Return the configuration
+        return config
+    except Exception as e:
+        return (
+            f"Error: Could not connect to {device_ip} with the following error: \n{e}"
+        )
+
+
+def get_hostname(
+    device_username, device_password, device_ip, device_type, enable_password
+):
+    """
+    Get the hostname of a device.
+
+    Args:
+        device_username: The username to authenticate with.
+        device_password: The password to authenticate with.
+        device_ip: The IP address of the device.
+        device_type: The device parameters.
+        enable_password: The enable password to authenticate with.
+
+    Returns:
+        String: The hostname of the device.
+    """
+    # Get the running configuration
+
+    device = {
+        "device_type": device_type,
+        "host": device_ip,
+        "username": device_username,
+        "password": device_password,
+        # wait a little longer for the device to respond
+        "fast_cli": False,
+        "global_delay_factor": 2,
+        "secret": enable_password,
+    }
+    # Get the hostname
+    with netmiko.ConnectHandler(**device) as connection:
+        prompt = connection.find_prompt()
+        prompt_index = prompt.find(">")  # get index of > in prompt
+        return prompt[:prompt_index]  # return hostname from prompt

vendors/hp_procurve.py

@@ -3,7 +3,9 @@
 import re
 
 
-def get_running_config(device_username, device_password, device_ip, device_type):
+def get_running_config(
+    device_username, device_password, device_ip, device_type, enable_password
+):
     """
     Get the command to get the running config.
 
@@ -12,6 +14,7 @@ def get_running_config(device_username, device_password, device_ip, device_type)
         device_password: The password to authenticate with.
         device_ip: The IP address of the device.
         device_type: The device parameters.
+        enable_password: The enable password to authenticate with.
 
     Returns:
         String: The running configuration of the device.
@@ -27,6 +30,7 @@ def get_running_config(device_username, device_password, device_ip, device_type)
             # wait a little longer for the device to respond
             "fast_cli": False,
             "global_delay_factor": 2,
+            "secret": enable_password,
         }
 
         with netmiko.ConnectHandler(**device) as connection: