Hi,
I am new to python and have written a script based on my experience so far with python to automate the process of an IOS upgrade. I have over 500 devices within my organization to upgrade their images to the latest ios. I have written my code in two stages: a pre-upgrade stage and an upgrade + post-upgrade stage.
The pre-upgrade stage is scripted to log in to the devices one after the other to collect data such as - hostname, current version, model, flash space availaibility etc.
Code worked well within a lab environment but testing it out on the live devices throughs the error below:
"Socket exception: An existing connection was forcibly closed by the remote host
(10054)"
Logs on the cisco router had this : %SSH-4-SSH2_UNEXPECTED_MSG: Unexpected message type has arrived. Terminating the connection from x.x.x.x.
Code: (Truncated to improve readability)
import os
import subprocess, re, netmiko
from netmiko import ConnectHandler
from netmiko.ssh_exception import NetmikoTimeoutException
from netmiko.ssh_exception import SSHException
from netmiko.ssh_exception import AuthenticationException
from netmiko import SCPConn
from datetime import datetime, time
import time
import csv
#Cisco ISR Data
new_ios_ISR_size = "702197190"
#Cisco C800 Data
new_ios_c800_size = "97199776"
#Cisco C1900 Data
new_ios_c1900_size = "41735808"
###########################################################################
#Creating the CSV files for pre upgrade
#clearing the old data from the pre-upgrade CSV file and writing the headers
f = open("data/pre_upgrade.csv", "w+")
f.write("IP Address, Hostname, Uptime, Current_Version, Current_Image, Serial_Number, Device_Model, Device_Memory, Available_space, space")
f.write("\n")
f.close()
#clearing the old data from the logs file and writing the headers
f = open("data/logs.txt", "w+")
f.close()
now = datetime.now()
logs_time = now.strftime("%H:%M:%S")
#############################################################################################################################
def preupgrade():
username = 'xxxxxxxxxx'
password = 'xxxxxxxxx'
with open('data/ip_address.csv', 'r') as file:
reader = csv.reader(file)
num = 0
for row in reader:
ip = row[0]
num += 1
print(ip, "Device " + str(num))
device = {
'device_type': 'cisco_ios',
'host': ip,
'username': username,
'password': password
}
now = datetime.now()
logs_time = now.strftime("%H:%M:%S")
print ("" + logs_time + ": " + ip + " Logging in to device")
f = open("data/logs.txt", "a")
f.write("" + logs_time + ": " + ip + " Logging in to device " + "\n" )
f.close()
#handling exceptions errors
try:
net_connect = ConnectHandler(**device)
except (NetmikoTimeoutException, AuthenticationException, SSHException, ValueError, TimeoutError, ConnectionError, OSError):
now = datetime.now()
logs_time = now.strftime("%H:%M:%S")
print ("" + logs_time + ": " + ip + " device login issue ")
f = open("data/logs.txt", "a")
f.write("" + logs_time + ": " + ip + " device login issue " + "\n" + "\n" )
f.close()
continue
#list where informations will be stored
pre_upgrade_devices = []
now = datetime.now()
logs_time = now.strftime("%H:%M:%S")
print ("" + logs_time + ": " + ip + " Log in Successfuul, Collecting device pre-upgrade report")
f = open("data/logs.txt", "a")
f.write("" + logs_time + ": " + ip + " Log in Successfuul, Collecting device pre-upgrade report " + "\n" )
f.close()
# execute show version on router and save output to output object
sh_ver_output = net_connect.send_command('show version')
#now = datetime.now()
#logs_time = now.strftime("%H:%M:%S")
#print ("" + logs_time + ": " + ip + " Checking the version ")
#finding hostname in output using regular expressions
regex_hostname = re.compile(r'(\S+)\suptime')
hostname = regex_hostname.findall(sh_ver_output
Notes:
- It works sometimes and then throws the above error and I have to wait till almost the next day and then it works unaided.
- The devices are a mix of 1SR4331/K9 and C891F-K9 routers
@Seb Rupik @Alexander Stevenson @dekwan
