netmiko+threading+os实现批量网络设备配置

1、确保存在要登录的iplist以及交换机的配置命令，netmiko直接进入交换机的配置模式下，不需要输入sys

2、iplist文本文档和交换机的配置文档h3conf都要和脚本在一个目录下

3、此代码针对于H3C交换机，华为也适用，只需更改设备类型为‘huawei’即可

4、iplist的比较少的话建议自己书写，如果较多，可以查看通过‘Python获取网络中在线设备’的脚本快速获取。

import concurrent.futures
import os
from netmiko import ConnectHandler

#只有验证不通过的ip地址

# 设备基础信息
device_info = {
    'device_type': 'hp_comware',  # 替换为你的设备类型
    'ip': '',  # IP地址将在循环中设置
    'username': '按实际写',
    'password': '按照实际写',

    # 其他可能需要的参数，如port, secret等
}


# 读取IP地址列表
def read_ip_list(file_path):
    with open(file_path, 'r') as file:
        return [line.strip() for line in file.readlines()]


# 发送配置到设备的函数
def send_config_to_device(ip):
    device_info['ip'] = ip  # 设置设备IP地址
    try:
        # 尝试连接到设备
        connection = ConnectHandler(**device_info)

        # 读取并发送配置文件
        with open('h3conf.txt', 'r') as config_file:
            config_commands = config_file.read().splitlines()
        output = connection.send_config_set(config_commands)

        # 打印配置输出
        print(f"设备 {ip} 的配置输出:")
        print(output)

        # 关闭连接
        connection.disconnect()
    except Exception as e:
        # 如果出现异常，记录失败的IP地址
        write_authentication_failed(ip)
        print(f"连接到设备 {ip} 失败: {str(e)}")


# 将失败的IP地址写入文件的函数
def write_authentication_failed(ip):
    log_dir = 'logs'
    os.makedirs(log_dir, exist_ok=True)  # 确保日志目录存在
    failed_file_path = os.path.join(log_dir, 'Authentication failed.txt')
    with open(failed_file_path, 'a') as failed_file:
        # 只写入IP地址，每个IP地址占一行
        failed_file.write(ip + '\n')


# 主函数
def main():
    ip_list_file = 'iplist.txt'
    ip_list = read_ip_list(ip_list_file)

    # 创建一个最大线程数为50的线程池
    with concurrent.futures.ThreadPoolExecutor(max_workers=50) as executor:
        # 提交任务到线程池
        future_to_ip = {executor.submit(send_config_to_device, ip): ip for ip in ip_list}
        for future in concurrent.futures.as_completed(future_to_ip):
            ip = future_to_ip[future]
            try:
                # 等待任务完成并获取结果
                future.result()
            except Exception as exc:
                print(f'生成结果为IP地址 {ip} 的任务时抛出异常: {exc}')

    print("所有设备配置完成。")


# 如果这是直接运行的脚本，则调用main函数
if __name__ == "__main__":
    main()