代码有问题可以看下这两篇文章
Python使用openpyxl库操作Excel之（二）访问已有的Excel文件_Gsdxiaohei的博客-CSDN博客_openpyxl打开指定文件
Python netmiko.ConnectHandler方法代码示例 - 纯净天空 (vimsky.com)
文件目录：
import getpass
import os
import time
import netmiko
from datetime import datetime
from gevent import monkey
from netmiko import ConnectHandler
from paramiko.ssh_exception import SSHException
#monkey.patch_all()
from gevent.pool import Pool
from openpyxl import load_workbook
########################################################################
#############################
# v1.2.0 2021.4.29 新增进度条显示功能
# v1.3.0 2021.4.30 新增华为设备模块
# v1.4.0 2021.5.6 新增锐捷AC模块
# v1.5.0 2021.5.7 1.新增"点击回⻋键"退出程序功能
# 2.打包封装程序并增加图标
# v2.0.0 2021.6.2 重构代码框架，后续可模块化添加设备和任务类型
# v2.0.1 2021.6.3 细微调整结构框架
# v3.0.0 2021.6.5 1.将原先多线程⽅式运⾏改为⽤gevent协程⽅式运⾏
# 2.修改flag信息
# 3.任务99名称修改为"⾃定义操作"
# 4.取消进度条显示，改为实时任务进度显示
# v3.1.0 2021.6.6 1.重构函数，增加报错输出。
# 2.修复get_config中锐捷的output变量bug
# v3.2.0 2021.6.8 1.读取⽂件形式由txt改成excel
# v3.2.1 2021.6.16 增加特权密码
# v3.2.2 2021.6.17 1.修复进⼊特权bug
# 2.增加ASA设备
# 3.修复ASA有failover创建⽂件bug
# v3.2.3 2021.6.18 修改H3C登录bug
########################################################################
#############################
print('****************************************************')
print('*Welcome!This program designed by Lindh08. *')
print('*Please contact me by E-mail.Add:511768312@qq.com. *')
print('****************************************************')
dev_type = input("请选择设备类型:\n\
1.Cisco\n\
2.Ruijie\n\
3.Ruijie_AC\n\
4.H3C\n\
5.Huawei\n\
6.Cisco-ASA\n\
请输⼊序号: ")
print('******************************************')
task = input("请选择想要运⾏的脚本类型:\n\
1.备份配置\n\
2.修改密码\n\
99.⾃定义操作\n\
请输⼊序号: ")
print('******************************************')
username = 'admin'
password = 'Sjzx@201909!'
#secret = getpass.getpass('Secret: ')
########################################################################
#############################
# 1.1判断模块
########################################################################
#############################
# 1.1.1判断设备类型
########################################################################
#############################
if int(dev_type) == 1: # Cisco设备
dev_type_brand = "cisco"
dev_type_str = "cisco_ios"
elif int(dev_type) == 2: # 锐捷路由器或者交换机设备
dev_type_brand = "ruijie"
dev_type_str = "ruijie_os"
elif int(dev_type) == 3: # 锐捷AC
dev_type_brand = "ruijie_AC"
dev_type_str = "ruijie_os"
elif int(dev_type) == 4: # H3C设备
dev_type_brand = "H3C"
dev_type_str = "hp_comware"
elif int(dev_type) == 5: # 华为设备
dev_type_brand = "Huawei"
dev_type_str = "huawei"
elif int(dev_type) == 6: # Cisco-ASA
dev_type_brand = "ASA"
dev_type_str = "cisco_asa_ssh"
# elif int(dev_type)==x: #后续添加设备
# dev_type_brand = "xxxx"
# dev_type_str = "xxxx"
########################################################################
#############################
# 1.1.2判断任务类型
########################################################################
#############################
if int(task) == 1: # 备份配置
task_type = "bak"
elif int(task) != 1:
if int(task) == 2: # 修改密码
task_type = "pwd"
# elif int(task)==x: #后续可添加模块
elif int(task) == 99: # 其他操作
task_type = "other"
########################################################################
#############################
# 1.2函数模块
########################################################################
#############################
# 1.2.1会话连接
########################################################################
#############################
def get_config(ip):
dev = {'device_type': dev_type_str,
'host': ip,
'username': username,
'password': password,
#'secret': secret,
}
ssh_session = ConnectHandler(**dev)
get_hostname = ssh_session.find_prompt().replace("#", "").replace("<", "").replace(">", "").replace("/", "-")
print(f"---- 正在连接： {get_hostname}({ip.strip()}).")
# 判断任务类型输⼊命令
if int(task) == 1: # 备份配置
if int(dev_type) == 1 or int(dev_type) == 2 or int(dev_type) == 3 or int(dev_type) == 6: # Cisco系设备
ssh_session.enable()
command = ("sh run")
output = ssh_session.send_command(command)
elif int(dev_type) == 4 or int(dev_type) == 5: # 华为系设备
command = ("dis cur")
output = ssh_session.send_command(command)
elif int(task) != 1:
cmdlist = open(f'cmd\\{int(task)}.{task_type}\\{task_type}_cmd_{dev_type_brand}.txt', 'r')
cmdlist.seek(0)
output = ssh_session.send_config_set(cmdlist.readlines())
output += ssh_session.save_config()
return output, get_hostname, ip
########################################################################
#############################
# 1.2.2输出配置信息
########################################################################
#############################
def output_log(output, get_hostname, ip):
now = datetime.now()
date = "%s-%s-%s" % (now.year, now.month, now.day)
config_path = f'log\\{int(task)}.{task_type}\\{task_type}' + date
verify_path = os.path.exists(config_path)
if not verify_path:
os.makedirs(config_path)
config_filename = f'{config_path}\\{get_hostname}_{ip}_{date}.txt'
print('---- 正在写⼊输出⽂件: ', config_filename)
with open(config_filename, "w", encoding='utf-8') as config_out:
config_out.write(output)
########################################################################
#############################
# 1.2.3检查问题设备
########################################################################
#############################
def output_issue_device(issue_device):
now = datetime.now()
date = "%s-%s-%s" % (now.year, now.month, now.day)
time_now = "%s-%s" % (now.hour, now.minute)
config_path = 'log\\issue_device\\' + 'issue_' + date
verify_path = os.path.exists(config_path)
if not verify_path:
os.makedirs(config_path)
config_filename = f'{config_path}\\issue_{date}_{time_now}.txt'
print('---- 正在写⼊问题设备: ', config_filename)
with open(config_filename, "a", encoding='utf-8') as issue_facts:
issue_facts.write('\n'.join(issue_device) + '\n')
########################################################################
#############################
# 1.2.4运⾏读取ip
########################################################################
#############################
def read_device():
ip_list = []
wb = load_workbook(f'检测中心设备管理IP信息.xlsx')
ws = wb['DLAN']
for cow_num in range(2, ws.max_row + 1):
ip = ws["a" + str(cow_num)].value
ip_list.append(ip)
return ip_list
########################################################################
#############################
# 1.2.5运⾏gevent
########################################################################
#############################
def run_gevent(ip):
issue_device = []
try:
device_config = get_config(ip)
output = device_config[0]
get_hostname = device_config[1]
output_log(output, get_hostname, ip)
except netmiko.NetmikoAuthenticationException:
issue_message = (ip + ': 认证错误 ')
issue_device.append(issue_message)
output_issue_device(issue_device)
except netmiko.exceptions.NetmikoTimeoutException:
issue_message = (ip + ': ⽹络不可达 ')
issue_device.append(issue_message)
output_issue_device(issue_device)
except SSHException:
issue_message = (ip + ': SSH端⼝异常 ')
issue_device.append(issue_message)
output_issue_device(issue_device)
except Exception as unknown_error:
issue_message = (ip + ': 发⽣未知错误: ')
output_issue_device(issue_device)
issue_device.append(issue_message + str(unknown_error))
# 演示如下：
########################################################################
#############################
# 1.2.6主函数
def main():
start_time = time.time()
print(f"******************************************")
print(f"程序于{time.strftime('%X')}开始执⾏\n")
ip_list = read_device()
pool = Pool(100)
pool.map(run_gevent, ip_list)
pool.join()
print(f"\n程序于{time.strftime('%X')}执⾏结束")
print(f"******************************************")
quit_program = input("请按回⻋键退出: ")
# 1.3运⾏程序
if __name__ == '__main__':
main()
C:\Users\82127\.conda\envs\pythonProject\python.exe E:/log/python/pythonProject/Demo10/Demo10.py
****************************************************
*Welcome!This program designed by Lindh08. *
*Please contact me by E-mail.Add:511768312@qq.com. *
****************************************************
请选择设备类型:
1.Cisco
2.Ruijie
3.Ruijie_AC
4.H3C
5.Huawei
6.Cisco-ASA
请输⼊序号: 5
******************************************
请选择想要运⾏的脚本类型:
1.备份配置
2.修改密码
99.⾃定义操作
请输⼊序号: 1
******************************************
******************************************
程序于23:48:21开始执⾏
---- 正在连接： Main Menu(172.22.1.1).
---- 正在写⼊问题设备: log\issue_device\issue_2022-6-4\issue_2022-6-4_23-48.txt
---- 正在写⼊问题设备: log\issue_device\issue_2022-6-4\issue_2022-6-4_23-48.txt
---- 正在写⼊问题设备: log\issue_device\issue_2022-6-4\issue_2022-6-4_23-48.txt
---- 正在写⼊问题设备: log\issue_device\issue_2022-6-4\issue_2022-6-4_23-48.txt
---- 正在连接： SMVIC-HQ-BAGL-Core(172.22.1.254).
---- 正在写⼊输出⽂件: log\1.bak\bak2022-6-4\SMVIC-HQ-BAGL-Core_172.22.1.254_2022-6-4.txt
程序于23:48:56执⾏结束
******************************************
请按回⻋键退出:
进程已结束,退出代码0
扫一扫
1216
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
