Added multi-vendor support

Author: Jérome Tissières

README.md

@@ -1,12 +1,19 @@
 # NX-OS-SCP-Bulk-Transfer
 
-A Simple Python script to upload and verify (checksum) files to multiple Cisco NX-OS devices in parallel, with the support of Netmiko.
+A Simple Python script to upload and verify (checksum) files to multiple network devices in parallel, supported by Netmiko
 
 
 ## Use Case Description
 
 If you have a large number of NX-OS devices and need to upload a file, for example a new version of NX-OS, to all these devices without the help of an external tool like Cisco DCNM or similar, it can be very time-consuming. This script takes care of everything.
 
+This script uses the strength and functionality of Netmiko to copy, in parallel, a single file to multiple network devices.
+
+Supported devices are:
+
+	Cisco Cisco IOS/IOS-XE/IOS-XR/NX-OS, Arista EOS, Juniper JunOS
+
+
 ## Installation
 
 Requirements: All you need is Python and pip. 
@@ -19,12 +26,21 @@ Installation:
 
 ## Configuration
 
-Update the file hosts.txt with the list of your NX-OS hosts, one host per line.
+Update the file hosts.csv with the list of your network hosts, one host per line, including the vendor type..
+
+Vendor types are:
+
+	cisco_ios
+	arista_eos
+	juniper_junos
+	cisco_nxos
 
 Example:
 
-	host01.example.com
-	host02.example.com
+	host01.example.com,cisco_ios
+	host02.example.com,cisco_nxos
+
+The file system is the default defined into Netmiko. Please refer to the Netmiko documentation here[https://pynet.twb-tech.com/blog/automation/netmiko-scp.html]
 
 ## Usage
 

hosts.txt

@@ -1,18 +1,17 @@
 #!/usr/bin/python
 from getpass import getpass
 from argparse import ArgumentParser
+import csv
 import os.path
+import sys
 from time import time
 from concurrent.futures import ProcessPoolExecutor, wait
 from netmiko import ConnectHandler, file_transfer
 from netmiko.ssh_exception import NetMikoAuthenticationException, NetMikoTimeoutException
 from paramiko.ssh_exception import AuthenticationException
 
-# --- Define the variables
-SW_LIST = []
+# --- Define the threads
 MAX_THREADS = 4
-future_list = []
-
 
 # --- Check file exists function
 def is_valid_file(parser, arg):
@@ -55,21 +54,21 @@ def upload_nemiko(net_device):
         transfer_dict = file_transfer(ssh_conn,
                             source_file=SOURCE_FILE,
                             dest_file=SOURCE_FILE,
-                            file_system='bootflash:',
-                            direction='put',
-                            overwrite_file=True)
-        print('Results for:', HOST)
+                            )
+        print(80*"=")
+        print('Results for', HOST+':')
         print('File exists already: ',transfer_dict['file_exists'])
         print('File transferred: ',transfer_dict['file_transferred'])
         print('MD5 verified :',transfer_dict['file_verified'])
-        print(80*"=")
     except NetMikoTimeoutException:
-        print('Skipped: SSH Timed out')
         print(80*"=")
+        print('Results for', HOST+':')
+        print('Skipped: SSH Timed out')
         #continue
     except (AuthenticationException, NetMikoAuthenticationException):
-        print('Skipped: Authentication failed')
         print(80*"=")
+        print('Results for', HOST+':')
+        print('Skipped: Authentication failed')
         #continue
 
 # --- Init argparse
@@ -80,20 +79,29 @@ def upload_nemiko(net_device):
 # --- Define the OS file to upload
 SOURCE_FILE = (args.filename)
 
-# --- Define the switch list
-with open("./hosts.txt", 'r') as FILE:
-    SW_LIST = [line.rstrip() for line in FILE]
+# --- Check the hosts.csv file and get the list of hosts
+VENDORS_TYPES = ["cisco_ios", "arista_eos", "juniper_junos", "cisco_nxos"]
+VENDOR_TYPE = ''
+HOSTS_LIST = []
+
+with open("./hosts.csv", 'r') as csvfile:
+    csv_reader = csv.reader(csvfile, delimiter=',')
+    for row in csv_reader:
+        if VENDOR_TYPE in VENDORS_TYPES not in str(row[1]):
+            print('Invalid CSV, please check the vendor types. Must be: cisco_ios, arista_eos, juniper_junos or cisco_nxos')
+            sys.exit()
+        HOSTS_LIST.append(row[0])
 
 # --- Ask confirmation
 print(80*"=")
-print('Please, confirm the upload of:',SOURCE_FILE+' on: ')
-print(*SW_LIST, sep ='\n')
+print('Please, confirm the upload of',SOURCE_FILE+' on: ')
+print(*HOSTS_LIST, sep ='\n')
 prompt = str("Proceed?")
 
 if confirm(prompt=prompt, resp=False) == True:
     # --- Get credentials
     print(80*"-")
-    USERNAME = input('Please insert your NX-OS username: ')
+    USERNAME = input('Please insert your username: ')
     print("And your password")
     PASSWORD = getpass()
     print(80*"-")
@@ -105,19 +113,26 @@ def upload_nemiko(net_device):
     pool = ProcessPoolExecutor(MAX_THREADS)
 
     # --- SCP itself, in multi-threads
-    for HOST in SW_LIST:
-        net_device = {
-        'device_type': 'cisco_nxos',
-        'host': HOST,
-        'username': USERNAME,
-        'password': PASSWORD,
-        }
-        future = pool.submit(upload_nemiko, net_device)
-        future_list.append(future)
-
-    wait(future_list)
+    SW_LIST = []
+    FUTURE_LIST = []
+    with open("./hosts.csv", 'r') as csvfile:
+        SW_LIST = csv.reader(csvfile, delimiter=',')
+        for CSV_ROW in SW_LIST:
+            HOST = CSV_ROW[0]
+            DEVICE_TYPE = CSV_ROW[1]
+            net_device = {
+            'device_type': DEVICE_TYPE,
+            'host': HOST,
+            'username': USERNAME,
+            'password': PASSWORD,
+            }
+            FUTURE = pool.submit(upload_nemiko, net_device)
+            FUTURE_LIST.append(FUTURE)
+
+        wait(FUTURE_LIST)
 
     # --- All done confirmation
+    print(80*"=")
     print("Uploads completed in {} seconds.".format(time() - start_time))
     print(80*"=")
 else: