Partner demo

demo_scripts/README.md

@@ -0,0 +1,20 @@
+Nexus 9000 Partner Demo Scripts
+===============================
+Welcome to the OneCloud Consulting Nexus 9000 demo script repository.
+
+These scripts were developed and tested on a Nexus 9000 C9372PX chassis running 7.0(3)I1(2).
+
+Summary
+========
+1. __find_freeip.py__ : Pings a range of IP addresses then prints which are UP followed by a summary of addresses that are unused.
+2. __sh_switch_details.py__ : Displays specific details from neighbor based on its IP address. Demonstrates use of a single programmatic Python instruction versus a series of CLI commands to obtain the same information.
+3. __cdp.py__ : Configures interface descriptions based on CDP information.
+4. __lldp.py__ : Configured interface descriptions based on LLDP information.
+5. __ospf.py__ : Provides menu driven options to configure OSPF interface attributes (area/cost).
+6. __pim.py__ : Provides menu driven options to configure PIM  interface attributes.
+7. __cleanup.py__ : Provides menu driven options to cleanup demo configurations.
+8. __sh_version.py__ : Uses NX-API to compare expected NX-OS version against actual NX-OS versions running in the network.
+9. __sh_proc_cpu_sort.py__ : Checks CPU health of switch at regular intervals with option to log output to file for TAC.
+10. __sh_proc_mem.py__ : Checks memory utilization of switch at regular intervals with option to log to file for TAC.
+11. __sh_int_count.py__ : Checks interface packet count at regular intervals with option to log to file for TAC.
+

demo_scripts/cdp.py

@@ -0,0 +1,120 @@
+"""Script Cataloging Information
+:Product Info:Nexus::9000::9516::NX-OS Release 6.2
+:Category:Config-Mgmt
+:Box Type:On-Box
+:Title:Interface Description configuration
+:Short Description:To dynamically configure interface descriptions
+:Long Description:Check the CDP state and modify the interface description accordingly.
+:Input:command to check the CDP state and based on the command output,
+       modify the description of the interface
+:Output:interface description should be updated
+"""
+
+import os
+from cli import *
+import sys
+
+"""
+
+Class to update the interface description based on the
+CDP state
+"""
+
+class Interface_Desc:
+
+    interface_message = {}
+
+    #get the nexus switch version and chassis details
+    def nexus_version(self):
+        versioncmd = "show version"
+        out = json.loads(clid(versioncmd))
+        chassis_id = out['chassis_id']
+        osversion = out['rr_sys_ver']
+        cpu_name = out['cpu_name']
+        memory =  out['memory']
+        processor_board =  out['proc_board_id']
+        device = out['host_name']
+        bootflash = out['bootflash_size']
+
+        print "Nexus Switch OS version is :" , osversion
+        print "Chassis ID is :", chassis_id
+        print  cpu_name + "with" + str(memory) + "KB of memory"
+        print "Processor Board ID is " + processor_board
+
+        print "Host Name : " + device
+        print "Bootflash : " + str(bootflash) + ' ' + "KB"
+        print "\n"
+
+    def cdp_status(self):
+        intob = Interface_Desc()
+
+        #check CDP is enabled or not
+        cdp_stat = "show cdp global"
+        stat = json.loads(clid(cdp_stat))
+
+        if (stat['cdp_global_enabled'] == 'enabled'):
+            print "CDP is enabled on the Host Switch"
+            cdp_nei = "show cdp nei"
+            status = json.loads(clid(cdp_nei))
+            status_list = status['TABLE_cdp_neighbor_brief_info']['ROW_cdp_neighbor_brief_info']
+            cdp_dict = {}
+
+            if (isinstance(status_list, list)):
+                for i in status_list:
+                    for key,value in i.items():
+                        if (key == 'device_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'intf_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'port_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'capability'):
+                            cdp_dict.update({key:value})
+                    intob.updateinterface(cdp_dict)
+            elif (isinstance(status_list, dict)):
+                for key,value in status_list.items():
+                        if (key == 'device_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'intf_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'port_id'):
+                            cdp_dict.update({key:value})
+                        if (key == 'capability'):
+                            cdp_dict.update({key:value})
+                intob.updateinterface(cdp_dict)
+            else:
+                print "Not implemented for this response type"
+
+        else:
+            print "CDP is not enabled on the Host Switch.Please check the CDP manual to enable it. "
+            exit(1)
+
+    #update the interface description
+    def updateinterface(self, data):
+        for key,value in data.iteritems():
+            if (key == 'intf_id'):
+                cmd1 = "interface" + ' ' + value
+                desc = "description" + '  ' + "Connected to device" + ' ' + data['device_id'] + ' ' + "on" + ' ' + data['port_id']
+                msg = "Connected to device" + '  ' + data['device_id'] + '  ' + "on" + '   ' + data['port_id']
+
+                cmd = "conf t" + ' ' + " ;" + ' ' + cmd1 + ' ' + ";" + ' ' + desc
+                cli(cmd)
+                print "\n"
+                print "Interface" + ' ' + data['intf_id'] + ' ' + "description is updated as : " + ' ' + msg
+                if (data['capability']):
+                    print "Neighbor device" + ' ' + data['device_id'] + ' ' + "is capable as : "
+                    for i in data['capability']:
+                        print str(i)
+
+                    #print  data['capability']
+
+
+
+if __name__ == '__main__':
+    interfaceob = Interface_Desc()
+    interfaceob.nexus_version()
+    interfaceob.cdp_status()
+
+
+
+

demo_scripts/cleanup.py

@@ -0,0 +1,225 @@
+"""
+ Cisco Nexus 9000 Switch On-Box script for cleaning
+ repetitive switch configurations such as disabling
+ features, removing IP addresses from all interfaces,
+ and so on.
+"""
+from cli import *
+import time
+
+class Cleanup(object):
+
+   def __init__(self):
+      """ No variables are required for this class. """
+
+   def print_Banner(self):
+      msg = '\n   _   _   _   _   _   _   _'
+      msg += '\n  / \\ / \\ / \\ / \\ / \\ / \\ / \\'
+      msg += '\n ( C | l | e | a | n | u | p )'
+      msg += '\n  \\_/ \\_/ \\_/ \\_/ \\_/ \\_/ \\_/'
+      msg += '\n\nCisco Nexus 9000 Switch On-Box script'
+      msg += '\nfor cleaning repetitive switch configurations.\n'
+      msg += '\nDeveloped By: OneCloud Consulting\n'	
+      msg += 'Please contact [email protected] for any queries.\n'
+      print msg
+
+   def print_Menu(self):
+      print "-"*25 + "   Menu Options   " + "-"*25
+      print "1) Remove all 'feature' CLI statements."
+      print "2) Remove 'ip pim sparse-mode' from all interfaces."
+      print "3) Remove IP addresses from all interfaces."
+      print "4) Remove all configured Loopback interfaces."
+      print "5) Remove all configured 'vrf context' except management."
+      print "6) Remove descriptions from all interfaces."
+      print "7) Run all of the above."
+      print "\nType any other key to exit menu."
+      print "-"*68
+
+   def remove_Feature(self):
+      # This function assumes all configured feature statements are comprised of 2 words.
+      # In the case of 'feature nv overlay', 3 words are used.
+      # To workaround this, explicitly remove 'feature nv overlay'.
+      cli('conf ; no feature nv overlay')
+      time.sleep(5)  # wait 5 seconds for above command to get removed
+
+      #Now parse and iterate across output of 'show run | include feature'
+      show_run_feature = []
+      show_run_feature = cli('show run | inc feature')
+
+      feature = show_run_feature.split( )
+
+      cli_commands = "conf"
+
+      for i in xrange(1, len(feature) ,2):
+         cli_commands += " ; no feature " + feature[i]
+
+      if len(feature) == 0:
+         print "There are no features to remove."
+      else:
+         print "You are about to remove all 'feature' CLI statements."
+
+         confirm = raw_input("Are you sure? [y|n]")
+         if confirm == "y":
+            cli(cli_commands)
+            result = cli('show run | inc feature')
+            print result
+            print "All 'feature' CLI statements removed!\n"
+            return
+         else:
+            print "You chose not to commit. Script aborted."
+            return
+
+   def remove_Pim(self):
+      show_int_brief = []
+      show_int_brief = cli('show ip int brief | inc Eth|Lo')
+
+      interfaces = show_int_brief.split( )
+
+      cli_commands = "conf"
+
+      for i in xrange(0, len(interfaces) ,3):
+         cli_commands += " ; interface " + str(interfaces[i])
+         cli_commands += " ; no ip pim sparse-mode"
+
+      print "You are about to remove PIM sparse-mode under all interfaces."
+
+      confirm = raw_input("Are you sure? [y|n]")
+
+      if confirm == "y":
+         cli(cli_commands)
+         print "\nPIM successfully removed from all interfaces!\n"
+         print "Type 'show ip pim interface brief' to verify.\n"
+         return
+      else:
+         print "You chose not to commit. Script aborted."
+         return
+
+   def remove_Ipaddress(self):
+         show_int_brief = []
+         show_int_brief = cli('show ip int brief | inc Eth|Lo')
+
+         interfaces = show_int_brief.split( )
+
+         cli_commands = "conf"
+
+         for i in xrange(0, len(interfaces) ,3):
+            cli_commands += " ; interface " + str(interfaces[i])
+            cli_commands += " ; no ip address"
+
+         print "You are about to remove IP addresses from all interfaces."
+
+         confirm = raw_input("Are you sure? [y|n]")
+
+         if confirm == "y":
+            cli(cli_commands)
+            print "\nIP addresses successfully removed from all interfaces!\n"
+            print "Type 'show ip interface brief' to verify.\n"
+            return
+         else:
+            print "You chose not to commit. Script aborted."
+            return
+
+   def remove_Loopbacks(self):
+            show_int_brief = []
+            show_int_brief = cli('show int brief | inc Lo')
+
+            interfaces = show_int_brief.split( )
+
+            cli_commands = "conf"
+
+            for i in xrange(0, len(interfaces) ,3):
+               cli_commands += " ; no interface " + str(interfaces[i])
+
+            print "You are about to remove all Loopback interfaces."
+
+            confirm = raw_input("Are you sure? [y|n]")
+
+            if confirm == "y":
+               cli(cli_commands)
+               print "\nAll Loopback interfaces removed!\n"
+               print "Type 'show interface brief | inc Loopback' to verify.\n"
+               return
+            else:
+               print "You chose not to commit. Script aborted."
+               return
+
+   def remove_Vrf_Contexts(self):
+            show_run = []
+            show_run = cli('show run | include context | exclude management')
+
+            contexts = show_run.split( )
+
+            cli_commands = "conf"
+
+            for i in xrange(2, len(contexts) ,3):
+               cli_commands += " ; no vrf context " + str(contexts[i])
+            print "You are about to remove all vrf contexts (except management)."
+
+            confirm = raw_input("Are you sure? [y|n]")
+
+            if confirm == "y":
+               cli(cli_commands)
+               print "\nAll vrf contexts removed!\n"
+               print "Type 'show run | inc context' to verify.\n"
+               return
+            else:
+               print "You chose not to commit. Script aborted."
+               return
+
+   def remove_Interface_Descriptions(self):
+               show_int_brief = []
+               show_int_brief = cli('show int brief | json-pretty | inc interface | exc TABLE | exc ROW')
+               interfaces = show_int_brief.replace(',', '')
+               interfaces = interfaces.replace('"', '')
+
+               interfaces = interfaces.split( )
+
+               cli_commands = "conf"
+
+               for i in xrange(1, len(interfaces) ,2):
+                  cli_commands += " ; interface " + str(interfaces[i] + " ; no description")
+
+               print "You are about to remove descriptions from all interfaces."
+
+               confirm = raw_input("Are you sure? [y|n]")
+
+               if confirm == "y":
+                  cli(cli_commands)
+                  print "\nRemoved descriptions from all interfaces!\n"
+                  print "Type 'sh run | include description' to verify.\n"
+                  return
+               else:
+                  print "You chose not to commit. Script aborted."
+                  return
+
+if __name__ == '__main__':
+   foo = Cleanup()
+   foo.print_Banner()
+   foo.print_Menu()
+   option = raw_input("\nSelect option [1-7]:")
+
+   if option == "1":
+      print "This may take some time..."
+      foo.remove_Feature()
+   elif option == "2":
+      check = cli('show feature | inc pim')
+      check_enabled = check.split()
+      if check_enabled[2] == "disabled":
+        print "There are no interfaces with 'ip pim sparse-mode' configured."
+      elif check_enabled[2] == "enabled":
+        foo.remove_Pim()
+   elif option == "3":
+      foo.remove_Ipaddress()
+   elif option == "4":
+      foo.remove_Loopbacks()
+   elif option == "5":
+      foo.remove_Vrf_Contexts()
+   elif option == "6":
+      foo.remove_Interface_Descriptions()
+   elif option == "7":
+      print "Checking configured features. This may take some time..."
+      foo.remove_Feature()
+      foo.remove_Ipaddress()
+      foo.remove_Loopbacks()
+      foo.remove_Pim()
+      foo.remove_Vrf_Contexts()