Skip to content

Instantly share code, notes, and snippets.

@kubsoo

kubsoo/upgrade_cisco.py

Created April 4, 2018 12:14
Show Gist options
  • Save kubsoo/39d9e72e035a131099a82bf3f577f985 to your computer and use it in GitHub Desktop.
Save kubsoo/39d9e72e035a131099a82bf3f577f985 to your computer and use it in GitHub Desktop.
Download ZIP
Python script to upgrade ios on cisco devices. Script checks md5 checksum, upload ios, insert boot system command, backup run config and reloads switch.
Raw
upgrade_cisco.py
#!/usr/bin/python
import netmiko
from netmiko import SCPConn
import os,sys,commands,re
def check_md5(filename):
command = 'md5sum '+filename
o = commands.getoutput(command)
output = o.split(' ')
return output[0]
def set_boot(net_connect,file):
net_connect.config_mode()
net_connect.send_command('boot system flash:{}'.format(file))
net_connect.exit_config_mode()
net_connect.send_command('wr')
def reload(net_connect):
net_connect.send_command('reload',expect_string='')
net_connect.send_command('\n')
def archive_run(net_connect,filename):
result = net_connect.send_command("show run")
# close SSH connection
net_connect.disconnect()
file = open(filename,"w")
file.write(result)
file.close()
def verify_md5(net_connect,file,md5):
result = net_connect.send_command("verify /md5 flash:{} {}".format(file,md5))
# close SSH connection
net_connect.disconnect()
reg = re.compile(r'Verified')
verify = reg.findall(result)
if verify:
result = True
else:
result = False
return result
def verify_space(net_connect,file):
result = net_connect.send_command("show flash:")
# close SSH connection
net_connect.disconnect()
reg = re.compile(r'(\d+)\sbytes\sfree')
space = reg.findall(result)
reg = re.compile(r'.*-rwx.*({})'.format(file))
exist = reg.findall(result)
f_size = os.path.getsize(file)
if space >= f_size:
result = 'True'
if space < f_size:
result = 'False'
if exist:
exist = 'True'
else:
exist = 'False'
return result,exist
def transfer_file(net_connect,file):
net_connect.config_mode()
net_connect.send_command('ip scp server enable')
scp_conn = SCPConn(net_connect)
s_file = file
d_file = file
scp_conn.scp_transfer_file(s_file, d_file)
if __name__ == "__main__":
if len(sys.argv) != 5:
print("\nplease provide the following arguments:")
print("\tupgrade_cisco.py <ip> <username> <password> <file>\n\n")
print("\tsample usage 1: upgrade_cisco.py 192.168.10.100 admin password cisco_ios_file.bin\n\n")
sys.exit(0)
ip = sys.argv[1]
username = sys.argv[2]
password = sys.argv[3]
file_s = sys.argv[4]
#verify if there is enough free space on device to upload ios file
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
ver = verify_space(net_connect,file_s)
print "\n\nVerifying sufficient space available on the file system ... %s\n\n" % ip
if ver[0] == 'True' and ver[1] == 'False':
upload = raw_input("\n\nSuccess! - proceed with image upload ? (y/n) ... ")
if upload == 'y':
print "\n\nUploading file : %s ...\n\n" % file_s
#transferring file to device
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
transfer_file(net_connect,file_s)
print "\n\nSuccess! - upload file: %s to device: %s was successfull ... \n\n" % (file_s,ip)
#veryfing md5
md5 = check_md5(file_s)
print "\n\nVerifying md5 checksum on device ... %s\n\n" % ip
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
v_md5 = verify_md5(net_connect, file_s,md5)
if v_md5 == True:
archive = raw_input("\n\nSuccess! - proceed with archive running config ? (y/n) ... ")
if archive == 'y':
file_n = ip+"-running-config.txt"
print "\n\nSaving running config into file: %s \n\n" % file_n
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
archive_run(net_connect,file_n)
boot = raw_input("\n\nSuccess! - proceed with inserting boot system command ? (y/n) ... ")
if boot == 'y':
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
set_boot(net_connect, file_s)
reload_3 = raw_input("\n\nSuccess! - proceed with reload ? (y/n) ... ")
if reload_3 == 'y':
reload_2 = raw_input("\n\nAre you sure ? (Y\N) ... ")
if reload_2 == 'Y':
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
try:
reload(net_connect)
except:
print "Reloading ... "
else:
print "\n\nAbort !!!\n\n"
else:
print "\n\nAbort !!!\n\n"
else:
print "\n\nError veryfiing md5 checksum on device, quitting !!!\n\n"
elif ver[0] == 'False' and ver[1] == 'False':
print "\n\nNot enough free space on device ... %s \n\n" % ip
elif ver[1] == 'True':
print "\n\nFile already uploaded on device ... %s \n\n" % ip
@telus-beep
Copy link

what i have change in this code pls help

this code was written in older version. have to do some changes. What are you trying to accomplish?

@abhishekoo7
Copy link

abhishekoo7 commented Nov 7, 2019 via email

@telus-beep
Copy link

I have done some changes in this code . Code is written in python version 2 and i have change this code into version 3 I am trying upgrade IOS of multiple cisco devices . but not able to understand this code please help.
…
On Wed, Nov 6, 2019 at 7:49 PM telus-beep @.***> wrote: what i have change in this code pls help this code was written in older version. have to do some changes. What are you trying to accomplish? — You are receiving this because you commented. Reply to this email directly, view it on GitHub https://gist.github.com/39d9e72e035a131099a82bf3f577f985?email_source=notifications&email_token=AEAABHYGP7I5MTUAWEUVYCDQSLG7RA5CNFSM4I753432YY3PNVWWK3TUL52HS4DFVNDWS43UINXW23LFNZ2KUY3PNVWWK3TUL5UWJTQAF3XKA#gistcomment-3075744, or unsubscribe https://github.com/notifications/unsubscribe-auth/AEAABH64EHMZ4DI55RJUUZLQSLG7RANCNFSM4I75343Q .

what help you want? what part do u want to understand. if you get any errors post them and post your code here

@abhishekoo7
Copy link

what parameter i have to change ?
username ?
password ?

#!/usr/bin/python

import netmiko
from netmiko import SCPConn
import os,sys,re,subprocess

def check_md5(filename):
command = 'md5sum '+filename
o = subprocess.getoutput(command)
output = o.split(' ')
return output[0]

def set_boot(net_connect,file):
net_connect.config_mode()
net_connect.send_command('boot system flash:{}'.format(file))
net_connect.exit_config_mode()
net_connect.send_command('wr')

def reload(net_connect):
net_connect.send_command('reload',expect_string='')
net_connect.send_command('\n')

def archive_run(net_connect,filename):
result = net_connect.send_command("show run")
# close SSH connection
net_connect.disconnect()
file = open(filename,"w")
file.write(result)
file.close()

def verify_md5(net_connect,file,md5):
result = net_connect.send_command("verify /md5 flash:{} {}".format(file,md5))
# close SSH connection
net_connect.disconnect()
reg = re.compile(r'Verified')
verify = reg.findall(result)
if verify:
result = True
else:
result = False
return result

def verify_space(net_connect,file):
result = net_connect.send_command("show flash:")
# close SSH connection
net_connect.disconnect()
reg = re.compile(r'(\d+)\sbytes\sfree')
space = reg.findall(result)
reg = re.compile(r'.-rwx.({})'.format(file))
exist = reg.findall(result)
f_size = os.path.getsize(file)
if space >= f_size:
result = 'True'
if space < f_size:
result = 'False'
if exist:
exist = 'True'
else:
exist = 'False'
return result,exist

def transfer_file(net_connect,file):
net_connect.config_mode()
net_connect.send_command('ip scp server enable')
scp_conn = SCPConn(net_connect)
s_file = file
d_file = file
scp_conn.scp_transfer_file(s_file, d_file)

if name == "main":
if len(sys.argv) != 5:
print("\nplease provide the following arguments:")
print("\tupgrade_cisco.py \n\n")
print("\tsample usage 1: upgrade_cisco.py 192.168.10.100 admin password cisco_ios_file.bin\n\n")
sys.exit(0)

ip = sys.argv[1]
username = sys.argv[2]
password = sys.argv[3]
file_s = sys.argv[4]

#verify if there is enough free space on device to upload ios file
net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
ver = verify_space(net_connect,file_s)

print ("\n\nVerifying sufficient space available on the file system ... %s\n\n" % ip)

if ver[0] == 'True' and ver[1] == 'False':
    upload = input("\n\nSuccess! - proceed with image upload ? (y/n) ... ")
    if upload == 'y':
       print ("\n\nUploading file : %s ...\n\n" % file_s)

       #transferring file to device
       net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
       transfer_file(net_connect,file_s)
       print ("\n\nSuccess! - upload file: %s to device: %s was successfull ... \n\n" % (file_s,ip))

       #veryfing md5
       md5 = check_md5(file_s)
       print ("\n\nVerifying md5 checksum on device ... %s\n\n" % ip)
       net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
       v_md5 = verify_md5(net_connect, file_s,md5)
       if v_md5 == True:
            archive = raw_input("\n\nSuccess! - proceed with archive running config ? (y/n) ... ")
            if archive == 'y':
                    file_n = ip+"-running-config.txt"
                    print ("\n\nSaving running config into file: %s \n\n" % file_n)
                    net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
                    archive_run(net_connect,file_n)

            boot = input("\n\nSuccess! - proceed with inserting boot system command ? (y/n) ... ")
            if boot == 'y':
                    net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
                    set_boot(net_connect, file_s)

                    reload_3 =input("\n\nSuccess! - proceed with reload ? (y/n) ... ")
                    if reload_3 == 'y':
                       reload_2 =input("\n\nAre you sure ? (Y/N) ... ")
                       if reload_2 == 'Y':
                          net_connect = netmiko.ConnectHandler(device_type='cisco_ios', ip=ip, username=username, password=password)
                          try:
                             reload(net_connect)
                          except:
                             print ("Reloading ... ")
                       else:
                          print ("\n\nAbort !!!\n\n")
                    else:
                        print ("\n\nAbort !!!\n\n")

       else:
            print ("\n\nError veryfiing md5 checksum on device, quitting !!!\n\n")

elif ver[0] == 'False' and ver[1] == 'False':
   print ("\n\nNot enough free space on device ... %s \n\n" % ip)

elif ver[1] == 'True':
    print ("\n\nFile already uploaded on device ... %s \n\n" % ip)

@telus-beep
Copy link

you have to run the script for eg if running in cmd, it has to be like this:-

python yourscriptname.py 100.100.100.100 cisco cisco c1900.bin

@abhishekoo7
Copy link

i am getting errror help

C:\Users\abhishek.kumar2\Desktop\demo>python upgrade_cisco.py 192.168.1.2 cisco cisco isr4300-universalk9.03.16.04b.S.155-3.S4b-ext.SPA.bin
Traceback (most recent call last):
File "upgrade_cisco.py", line 86, in
ver = verify_space(net_connect,file_s)
File "upgrade_cisco.py", line 53, in verify_space
if space >= f_size:
TypeError: '>=' not supported between instances of 'list' and 'int'

@telus-beep
Copy link

i am getting errror help

C:\Users\abhishek.kumar2\Desktop\demo>python upgrade_cisco.py 192.168.1.2 cisco cisco isr4300-universalk9.03.16.04b.S.155-3.S4b-ext.SPA.bin
Traceback (most recent call last):
File "upgrade_cisco.py", line 86, in
ver = verify_space(net_connect,file_s)
File "upgrade_cisco.py", line 53, in verify_space
if space >= f_size:
TypeError: '>=' not supported between instances of 'list' and 'int'

This is problem of different versions 2.6 and 3. i am not sure how to fix this

@kubsoo
Copy link
Author

kubsoo commented Nov 11, 2019

space format is list and f_size format is int. You cannot use <>= operators between two types. In order to fix that try to modify line #49
space = reg.findall(result) to space = reg.findall(result)[0]. This will give you first element of that list, which should be in int.

@telus-beep
Copy link

space format is list and f_size format is int. You cannot use <>= operators between two types. In order to fix that try to modify line #49
space = reg.findall(result) to space = reg.findall(result)[0]. This will give you first element of that list, which should be in int.

what abt this eroor, any idea?

Traceback (most recent call last):
File "final.py", line 2, in
from netmiko import SCPConn
File "/usr/lib/python2.6/site-packages/netmiko/init.py", line 6, in
log.addHandler(logging.NullHandler()) # noqa
AttributeError: 'module' object has no attribute 'NullHandler'

@abhishekoo7
Copy link

thanks, i am going to try this tomorrow

@athirupati
Copy link

athirupati commented Feb 16, 2024 •
edited
Loading

Hi @abhishekoo7 Would you mind commenting back to this thread regarding the successful completion. Helps me to understand what you have made changes. Thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment