这是群里大佬Lindh08分享的实用脚本,支持多厂商的设备配置文件备份，也支持自定义配置，修改密码模块有点完善，感兴趣的小伙伴可以运行测试，也可以在此基础进行完善。

    代码如下：

    1. import os
    2. import time
    3. from datetime import datetime
    4. import getpass
    5. import re
    6. import sys
    7. from netmiko import ConnectHandler
    8. from netmiko.ssh_exception import NetMikoTimeoutException
    9. from netmiko.ssh_exception import AuthenticationException
    10. from paramiko.ssh_exception import SSHException
    11. import gevent
    12. from gevent import spawn
    13. from gevent import monkey;monkey.patch_all()
    14. from gevent.pool import Pool
    15. from openpyxl import Workbook
    16. from openpyxl import load_workbook
    18. #####################################################################################################
    19. # v1.2.0 2021.4.29 新增进度条显示功能
    20. # v1.3.0 2021.4.30 新增华为设备模块
    21. # v1.4.0 2021.5.6  新增锐捷AC模块
    22. # v1.5.0 2021.5.7  1.新增"点击回车键"退出程序功能
    23. #           2.打包封装程序并增加图标
    24. # v2.0.0 2021.6.2  重构代码框架，后续可模块化添加设备和任务类型
    25. # v2.0.1 2021.6.3  细微调整结构框架
    26. # v3.0.0 2021.6.5  1.将原先多线程方式运行改为用gevent协程方式运行
    27. #                  2.修改flag信息
    28. #                  3.任务99名称修改为"自定义操作"
    29. #                  4.取消进度条显示，改为实时任务进度显示
    30. # v3.1.0 2021.6.6  1.重构函数，增加报错输出。
    31. #                  2.修复get_config中锐捷的output变量bug
    32. # v3.2.0 2021.6.8  1.读取文件形式由txt改成excel
    33. # v3.2.1 2021.6.16 增加特权密码
    34. # v3.2.2 2021.6.17 1.修复进入特权bug
    35. #                  2.增加ASA设备
    36. #                  3.修复ASA有failover创建文件bug
    37. # v3.2.3 2021.6.18 修改H3C登录bug
    38. #####################################################################################################
    40. print('****************************************************')
    41. print('*Welcome!This program designed by Lindh08.         *')
    42. print('*Please contact me by E-mail.Add:511768312@qq.com. *')
    43. print('****************************************************')
    45. dev_type = input("请选择设备类型:\n\
    46. 1.Cisco\n\
    47. 2.Ruijie\n\
    48. 3.Ruijie_AC\n\
    49. 4.H3C\n\
    50. 5.Huawei\n\
    51. 6.Cisco-ASA\n\
    52. 请输入序号: ")
    53. print('******************************************')
    54. task = input("请选择想要运行的脚本类型:\n\
    55. 1.备份配置\n\
    56. 2.修改密码\n\
    57. 99.自定义操作\n\
    58. 请输入序号: ")
    59. print('******************************************')
    61. username = input('Username: ')
    62. password = getpass.getpass('Password: ')
    63. secret = getpass.getpass('Secret: ')
    65. #####################################################################################################
    66. # 1.1判断模块
    67. #####################################################################################################
    68. # 1.1.1判断设备类型
    69. #####################################################################################################
    70. if int(dev_type)==1: # Cisco设备
    71.   dev_type_brand = "cisco"
    72.   dev_type_str = "cisco_ios"
    73. elif int(dev_type)==2: #锐捷路由器或者交换机设备
    74.   dev_type_brand = "ruijie"
    75.   dev_type_str = "ruijie_os"
    76. elif int(dev_type)==3: #锐捷AC
    77.   dev_type_brand = "ruijie_AC"
    78.   dev_type_str = "ruijie_os"
    79. elif int(dev_type)==4: #H3C设备
    80.   dev_type_brand = "H3C"
    81.   dev_type_str = "hp_comware"
    82. elif int(dev_type)==5: #华为设备
    83.   dev_type_brand = "Huawei"
    84.   dev_type_str = "huawei"
    85. elif int(dev_type)==6: #Cisco-ASA
    86.   dev_type_brand = "ASA"
    87.   dev_type_str = "cisco_asa_ssh"
    88. #elif int(dev_type)==x: #后续添加设备
    89. #  dev_type_brand = "xxxx"
    90. #  dev_type_str = "xxxx"
    91. #####################################################################################################
    92. # 1.1.2判断任务类型
    93. #####################################################################################################
    94. if int(task)==1: #备份配置
    95.   task_type = "bak"
    96. elif int(task)!=1:
    97.   if int(task)==2: #修改密码
    98.     task_type = "pwd"
    99.   #elif int(task)==x: #后续可添加模块
    100.   elif int(task)==99: #其他操作
    101.     task_type = "other"
    102. #####################################################################################################
    103. # 1.2函数模块
    104. #####################################################################################################
    105. # 1.2.1会话连接
    106. #####################################################################################################
    107. def get_config(ip):
    109.   dev = {'device_type':dev_type_str,
    110.       'host':ip,
    111.       'username':username,
    112.       'password':password,
    113.       'secret':secret,
    114.       }
    116.   ssh_session = ConnectHandler(**dev)
    117.   get_hostname = ssh_session.find_prompt().replace("#","").replace("<","").replace(">","").replace("/","-")
    118.   print(f"---- 正在连接： {get_hostname}({ip.strip()}).")
    119.   # 判断任务类型输入命令
    120.   if int(task)==1: #备份配置
    121.     if int(dev_type)==1 or int(dev_type)==2 or int(dev_type)==3 or int(dev_type)==6: # Cisco设备
    122.       ssh_session.enable()
    123.       command = ("sh run")
    124.       output = ssh_session.send_command(command)
    125.     elif int(dev_type)==4 or int(dev_type)==5 : # 华为设备
    126.       command = ("dis cur")
    127.       output = ssh_session.send_command(command)
    128.   elif int(task)!=1:
    129.     cmdlist = open(f'cmd\\{int(task)}.{task_type}\\{task_type}_cmd_{dev_type_brand}.txt','r')
    130.     cmdlist.seek(0)  
    131.     output = ssh_session.send_config_set(cmdlist.readlines())
    132.     output += ssh_session.save_config()
    134.   return output,get_hostname
    136. #####################################################################################################
    137. # 1.2.2输出配置信息
    138. #####################################################################################################
    139. def output_log(output,get_hostname):
    141.   now = datetime.now()
    142.   date= "%s-%s-%s"%(now.year,now.month,now.day)
    143.   config_path = f'log\\{int(task)}.{task_type}\\{task_type}'+date
    144.   verify_path = os.path.exists(config_path)
    145.   if not verify_path:
    146.     os.makedirs(config_path)
    148.   config_filename = f'{config_path}\\{get_hostname}_{date}.txt'
    149.   print ('---- 正在写入输出文件: ', config_filename)
    150.   with open( config_filename, "w",encoding='utf-8' ) as config_out:
    151.     config_out.write(output)
    153. #####################################################################################################
    154. # 1.2.3检查问题设备
    155. #####################################################################################################
    156. def output_issue_device(issue_device):
    158.   now = datetime.now()
    159.   date= "%s-%s-%s"%(now.year,now.month,now.day)
    160.   time_now = "%s-%s"%(now.hour,now.minute)
    161.   config_path = 'log\\issue_device\\'+'issue_'+date
    162.   verify_path = os.path.exists(config_path)
    164.   if not verify_path:
    165.     os.makedirs(config_path)
    167.   config_filename = f'{config_path}\\issue_{date}_{time_now}.txt'
    169.   print ('---- 正在写入问题设备: ', config_filename)
    170.   with open (config_filename, "a", encoding='utf-8') as issue_facts:
    171.     issue_facts.write('\n'.join(issue_device)+'\n')
    173. #####################################################################################################
    174. # 1.2.4运行读取ip
    175. #####################################################################################################
    176. def read_device():
    177.   ip_list = []
    179.   wb = load_workbook(f'lists\\ip_list.xlsx')
    180.   ws = wb[dev_type_brand]
    182.   for cow_num in range(2,ws.max_row+1):
    183.     ip = ws["a"+str(cow_num)].value
    184.     ip_list.append(ip)
    186.   return ip_list
    188. #####################################################################################################
    189. # 1.2.5运行gevent
    190. #####################################################################################################
    191. def run_gevent(ip):
    192.   issue_device = []
    194.   try:
    195.     device_config = get_config(ip)
    196.     output = device_config[0]
    197.     get_hostname = device_config[1]
    198.     output_log(output,get_hostname)
    200.   except (AuthenticationException):
    201.     issue_message = (ip + ': 认证错误 ')
    202.     issue_device.append(issue_message)
    203.     output_issue_device(issue_device)
    204.   except NetMikoTimeoutException:
    205.     issue_message = (ip + ': 网络不可达 ')
    206.     issue_device.append(issue_message)
    207.     output_issue_device(issue_device)
    208.   except (SSHException):
    209.     issue_message = (ip +': SSH端口异常 ')
    210.     issue_device.append(issue_message)
    211.     output_issue_device(issue_device)
    212.   except Exception as unknown_error:
    213.     issue_message = (ip +': 发生未知错误: ')
    214.     output_issue_device(issue_device)
    215.     issue_device.append(issue_message+str(unknown_error))
    218. #####################################################################################################
    219. # 1.2.6主函数
    220. #####################################################################################################
    221. def main():
    222.   start_time = time.time()
    223.   print(f"******************************************")
    224.   print(f"程序于{time.strftime('%X')}开始执行\n")
    225.   ip_list = read_device()
    226.   pool = Pool(100)
    227.   pool.map(run_gevent,ip_list)
    228.   pool.join()
    229.   print(f"\n程序于{time.strftime('%X')}执行结束")
    230.   print(f"******************************************")
    231.   quit_program = input("请按回车键退出: ")
    233. #####################################################################################################
    234. # 1.3运行程序
    235. #####################################################################################################
    236. if __name__ == '__main__':
    237.   main()
    238. #####################################################################################################

    演示如下：

    这里小编把py脚本打包成exe可执行文件来操作，在此以思科设备为例。
    image.png
    image.png
    https://mp.weixin.qq.com/s/AwVf_wmCFoQZsiNl1k501g （公众号文章中有视频演示）
    ，时长05:34

    脚本文件获取：公众号后台回复“ py网络工具 ”
    5、python网络运维工具包 - 图3