Bulk encryption utility for .netmiko.yml (#3506)

Author: ktbyers

netmiko/cli_tools/netmiko_bulk_encrypt.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+import argparse
+import sys
+from pathlib import Path
+from ruamel.yaml import YAML
+
+from netmiko.encryption_handling import encrypt_value, get_encryption_key
+
+yaml = YAML()
+yaml.preserve_quotes = True
+yaml.indent(mapping=2, sequence=4, offset=2)
+
+
+def encrypt_netmiko_yml(
+    input_file: str, output_file: str | None, encryption_type: str
+) -> None:
+    # Read the input YAML file
+    input_path = Path(input_file).expanduser()
+    with input_path.open("r") as f:
+        config = yaml.load(f)
+
+    # Get the encryption key
+    key = get_encryption_key()
+
+    # Encrypt password and secret for each device
+    for device, params in config.items():
+        if isinstance(params, dict):
+            if "password" in params:
+                encrypted_value = encrypt_value(
+                    params["password"], key, encryption_type
+                )
+                params["password"] = encrypted_value
+            if "secret" in params:
+                encrypted_value = encrypt_value(params["secret"], key, encryption_type)
+                params["secret"] = encrypted_value
+
+    # Write the updated config to the output file or stdout
+    if output_file:
+        output_path = Path(output_file)
+        with output_path.open("w") as f:
+            yaml.dump(config, f)
+    else:
+        yaml.dump(config, sys.stdout)
+
+
+def main_ep():
+    sys.exit(main())
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Encrypt passwords in .netmiko.yml file"
+    )
+    parser.add_argument(
+        "--input_file",
+        default="~/.netmiko.yml",
+        help="Input .netmiko.yml file (default: ~/.netmiko.yml)",
+    )
+    parser.add_argument(
+        "--output_file",
+        help="Output .netmiko.yml file with encrypted passwords (default: stdout)",
+    )
+    parser.add_argument(
+        "--encryption-type",
+        choices=["fernet", "aes128"],
+        default="fernet",
+        help="Encryption type to use (default: fernet)",
+    )
+
+    args = parser.parse_args()
+
+    encrypt_netmiko_yml(args.input_file, args.output_file, args.encryption_type)
+
+    if args.output_file:
+        print(
+            f"Encrypted .netmiko.yml file has been written to {Path(args.output_file).resolve()}",
+            file=sys.stderr,
+        )
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

netmiko/cli_tools/netmiko_encrypt.py

@@ -1,12 +1,17 @@
 #!/usr/bin/env python3
 import os
+import sys
 import argparse
 from getpass import getpass
 
 from netmiko.utilities import load_netmiko_yml
 from netmiko.encryption_handling import encrypt_value
 
 
+def main_ep():
+    sys.exit(main())
+
+
 def main():
     parser = argparse.ArgumentParser(
         description="Encrypt data using Netmiko's encryption."
@@ -56,6 +61,8 @@ def main():
     encrypted_data = encrypt_value(data, key, encryption_type)
     print(f"\nEncrypted data: {encrypted_data}\n")
 
+    return 0
+
 
 if __name__ == "__main__":
-    main()
+    sys.exit(main())